import matplotlib.pyplot as plt

# 设置中文字体
plt.rcParams["font.family"] = ["SimHei"]

def plot_views_distribution(main_views, filename="用户观点分布.png"):
    """绘制用户观点分布柱状图"""
    plt.figure(figsize=(10, 6))
    plt.bar(
        main_views.keys(), 
        main_views.values(), 
        color=['#4CAF50', '#2196F3', '#f44336', '#FFC107', '#9E9E9E']
    )
    plt.title('用户对大语言模型的主要关注点分布')
    plt.ylabel('弹幕数量')
    plt.xticks(rotation=30)
    
    # 添加数值标签
    for i, v in enumerate(main_views.values()):
        plt.text(i, v + 5, str(v), ha='center')
        
    plt.tight_layout()
    plt.savefig(filename, dpi=300)
    plt.show()