import pandas as pd
import numpy as np
from pathlib import Path
from loguru import logger

def generate(output_dir):
    logger.info("开始生成GDP趋势图表")
    
    quarters = ['2024Q1', '2024Q2', '2024Q3', '2024Q4', '2025Q1', '2025Q2']
    gdp_growth = [5.2, 4.8, 5.0, 5.3, np.random.uniform(4.5, 5.5), np.random.uniform(4.5, 5.5)]
    
    data = pd.DataFrame({
        '季度': quarters,
        'GDP增长率(%)': [round(x, 2) for x in gdp_growth]
    })
    data = data.set_index('季度')
    
    from dynamic_generator import DynamicContentGenerator
    generator = DynamicContentGenerator()
    output_path = generator.generate_chart_matplotlib(
        data,
        title='中国GDP增长趋势',
        x_label='季度',
        y_label='增长率 (%)',
        filename='gdp_chart.png',
        kind='line'
    )
    
    return output_path

if __name__ == "__main__":
    generate(Path.cwd())