/llm-viz week_XX              # 显示本周应该生成的图表类型
/llm-viz week_XX token_usage  # 生成特定类型的图表

# ❌ 错误：截断 Y 轴夸大差异
plt.ylim(95, 100)  # 从 95 开始，5% 差距看起来很大

# ✅ 正确：从 0 开始（柱状图必须）
plt.ylim(0, None)  # 或不设置，让 matplotlib 自动选择

# ❌ 错误：一张图展示所有东西
sns.scatterplot(data=df, x='a', y='b', hue='c', size='d', style='e')

# ✅ 正确：分面或分图
g = sns.relplot(data=df, x='a', y='b', hue='c', col='e')

import matplotlib.pyplot as plt
import matplotlib.font_manager as fm

def setup_chinese_font() -> str:
    """配置中文字体，返回使用的字体名称"""
    chinese_fonts = ['SimHei', 'Noto Sans CJK SC', 'Arial Unicode MS',
                     'PingFang SC', 'Microsoft YaHei']
    available = [f.name for f in fm.fontManager.ttflist]
    for font in chinese_fonts:
        if font in available:
            plt.rcParams['font.sans-serif'] = [font]
            plt.rcParams['axes.unicode_minus'] = False
            return font
    plt.rcParams['font.sans-serif'] = ['DejaVu Sans']
    return 'DejaVu Sans'

"""
示例：生成不同 LLM 任务的 Token 使用对比图。

运行方式：python3 chapters/week_XX/examples/01_token_chart.py
预期输出：生成 images/token_usage.png
"""
from __future__ import annotations

import matplotlib.pyplot as plt
import matplotlib.font_manager as fm
from pathlib import Path


def setup_chinese_font() -> str:
    chinese_fonts = ['SimHei', 'Noto Sans CJK SC', 'Arial Unicode MS']
    available = [f.name for f in fm.fontManager.ttflist]
    for font in chinese_fonts:
        if font in available:
            plt.rcParams['font.sans-serif'] = [font]
            plt.rcParams['axes.unicode_minus'] = False
            return font
    return 'DejaVu Sans'


def main() -> None:
    font = setup_chinese_font()

    tasks = ['分类', '摘要', '抽取', '问答']
    input_tokens = [150, 500, 200, 300]
    output_tokens = [20, 100, 50, 150]

    fig, ax = plt.subplots(figsize=(10, 6))

    x = range(len(tasks))
    width = 0.35

    bars1 = ax.bar([i - width/2 for i in x], input_tokens, width,
                   label='输入 Token', color='steelblue')
    bars2 = ax.bar([i + width/2 for i in x], output_tokens, width,
                   label='输出 Token', color='coral')

    ax.set_xlabel('任务类型', fontsize=12)
    ax.set_ylabel('Token 数量', fontsize=12)
    ax.set_title('不同 LLM 任务的 Token 使用对比', fontsize=14)
    ax.set_xticks(x)
    ax.set_xticklabels(tasks)
    ax.legend()

    # 添加数值标签
    for bar in bars1 + bars2:
        height = bar.get_height()
        ax.annotate(f'{int(height)}',
                    xy=(bar.get_x() + bar.get_width() / 2, height),
                    ha='center', va='bottom', fontsize=10)

    output_dir = Path(__file__).parent.parent / 'images'
    output_dir.mkdir(exist_ok=True)
    plt.savefig(output_dir / 'token_usage.png', dpi=150, bbox_inches='tight')
    plt.close()
    print(f"图片已保存: {output_dir / 'token_usage.png'}")


if __name__ == '__main__':
    main()