components/visualizations.py

import streamlit as st
import plotly.graph_objects as go
import plotly.express as px

def plot_emotion_distribution(emotion_dict):
    """

    绘制情绪概率分布图（雷达图）

    

    Args:

        emotion_dict: 包含情绪标签和对应概率的字典

    """
    emotions = list(emotion_dict.keys())
    probabilities = [float(emotion_dict[emotion].strip('%')) / 100 for emotion in emotions]  # 转换为浮点数

    # 创建雷达图
    fig = go.Figure()

    fig.add_trace(go.Scatterpolar(
        r=probabilities + [probabilities[0]],  # 闭合图形
        theta=emotions + [emotions[0]],  # 闭合图形
        fill='toself',
        name='Emotion Distribution'
    ))

    fig.update_layout(
        title="Emotion Distribution",
        polar=dict(
            radialaxis=dict(
                visible=True,
                range=[0, 1]  # 设置范围
            )),
        showlegend=False
    )

    st.plotly_chart(fig, use_container_width=True)