Update app.py

app.py

@@ -1,142 +1,103 @@
-import plotly.express as px
-from plotly import graph_objects as go
-import pandas as pd
 import streamlit as st
-from io import BytesIO
-from zipfile import ZipFile
+import pandas as pd
+import plotly.express as px
 import os
-
-CHINESE_FONT = "SimHei"
+from io import BytesIO
+import zipfile
 
 def setup_chinese_font():
-    """设置中文字体，确保图表中的中文显示正常"""
-    px.defaults.template = "plotly_white"
-    go.FigureWidget._default_font = {"family": CHINESE_FONT}
-
-def create_radar_chart(df, students, subjects):
-    """
-    创建雷达图
-
-    Args:
-        df: 数据框
-        students: 学生列表
-        subjects: 科目列表
-
-    Returns:
-        plotly figure 对象
-    """
-    # 筛选指定学生和科目
-    filtered_df = df[df['姓名'].isin(students)]
-    filtered_df = filtered_df[subjects]
-
-    # 计算每个学生的平均分
-    avg_scores = filtered_df.mean()
-
-    fig = go.Figure()
-    for student in students:
-        student_data = df[df['姓名'] == student][subjects]
-        if not student_data.empty:
-            scores = student_data.values.flatten().tolist()
-            fig.add_trace(go.Scatterpolar(
-                r=scores,
-                theta=subjects,
-                name=f"{student} ({avg_scores[student]:.1f})",
-                mode='lines+markers'
-            ))
-
-    # 设置图表样式
-    fig.update_layout(
-        polar=dict(
-            radialaxis=dict(range=[0, 100], showline=True),
-            angularaxis=dict(direction="clockwise")
-        ),
-        title=f"学生成绩对比 - {', '.join(students)}",
-        font=dict(family=CHINESE_FONT)
-    )
-
+    import plotly.io as pio
+    pio.templates["custom_radar"] = {
+        "layout": {
+            "font": {"family": "SimSun-ExtB", "size": 12},
+            "title_x": 0.5,
+            "title_y": 0.98,
+            "title_font_size": 14,
+            "polar": {
+                "radialaxis": {"tickfont": {"family": "SimSun-ExtB"}},
+                "angularaxis": {"tickfont": {"family": "SimSun-ExtB"}}
+            }
+        }
+    }
+
+def create_radar_chart(student_data, avg_score, student_id):
+    fig = px.line_polar(r=student_data, theta=student_data.index,
+                        name=f"學生 {student_id}", line_close=True)
+    fig.add_scatterpolar(r=[avg_score]*len(student_data),
+                         theta=student_data.index,
+                         name="平均分數", fill=None)
+    fig.update_traces(fill='toself')
+    fig.update_layout(title_text=f"學生 {student_id} 的成績分析",
+                     title_x=0.5, showlegend=True)
     return fig
 
 def generate_radar_image(fig):
-    """
-    生成图表的图像数据
-
-    Args:
-        fig: plotly figure 对象
-
-    Returns:
-        图像数据 bytes
-    """
-    try:
-        # 使用 kaleido 转换为 PNG 格式
-        img_bytes = fig.to_image(format="png", engine="kaleido")
-        return img_bytes
-    except Exception as e:
-        st.error(f"生成图表时出错: {str(e)}")
-        return None
+    img_bytes = fig.to_image(format="png", width=800, height=600)
+    return img_bytes
 
 def main():
-    # 设置页面标题和宽屏模式
-    st.set_page_config(page_title="学生成绩分析工具", layout="wide")
+    st.title("學生成绩分析工具")
+    st.write("上傳你的成績文件，我們將為你生成詳細的分析報告。")
+
+    # 文件上传组件
+    uploaded_file = st.file_uploader(
+        "請選擇要上傳的文件",
+        accept_multiple_files=False,
+        key="fileUploader"
+    )
 
-    # 设置中文字体
-    setup_chinese_font()
+    if uploaded_file is not None:
+        try:
+            # 判断文件类型
+            file_ext = os.path.splitext(uploaded_file.name)[1].lower()
+            if file_ext == ".csv":
+                df = pd.read_csv(uploaded_file)
+            elif file_ext in [".xlsx", ".xls"]:
+                df = pd.read_excel(uploaded_file, engine='openpyxl')
+            else:
+                st.error("不支援的文件格式，請上傳 CSV 或 Excel 文件。")
+                return
 
-    # 上传数据
-    with st.expander("上传成绩数据"):
-        uploaded_file = st.file_uploader("上传Excel文件", type=['xlsx', 'xls'])
-        if uploaded_file is not None:
-            try:
-                df = pd.read_excel(uploaded_file)
-                st.success("数据上传成功！")
-            except Exception as e:
-                st.error(f"数据读取失败: {str(e)}")
+            # 检查數據格式
+            if '學生ID' not in df.columns or '科目' not in df.columns:
+                st.error("文件格式不符合要求，請確保包含 '學生ID' 和 '科目' 這兩列。")
                 return
-        else:
-            st.info("请上传包含学生成绩的Excel文件。")
-            return
 
-    # 选择学生和科目
-    with st.expander("选择学生和科目"):
-        students = df['姓名'].unique().tolist()
-        selected_students = st.multiselect("选择学生", students)
-        subjects = [col for col in df.columns if col != '姓名']
-        selected_subjects = st.multiselect("选择科目", subjects, default=subjects)
+            # 獲取獨特的學生ID和科目列表
+            student_ids = df['學生ID'].unique()
+            subjects = df['科目'].unique()
 
-    # 生成图表
-    with st.expander("生成雷达图"):
-        if not selected_students or not selected_subjects:
-            st.warning("请选择至少一个学生和一个科目。")
-            return
+            # 用戶選擇
+            st.write("### 選擇分析條件")
+            selected_student = st.selectbox(
+                "請選擇要分析的学生",
+                options=student_ids,
+                key="selectedStudent"
+            )
 
-        fig = create_radar_chart(df, selected_students, selected_subjects)
-        st.plotly_chart(fig)
+            # 確定選定學生的數據
+            filtered_data = df[df['學生ID'] == selected_student]
+            student_subjects = filtered_data['科目']
+            scores = filtered_data['分數']
 
-    # 批量下载功能
-    with st.expander("批量下载图表"):
-        if st.button("生成并下载所有图表"):
-            # 创建临时文件夹存储图表
-            temp_dir = "temp_charts"
-            os.makedirs(temp_dir, exist_ok=True)
+            # 計算平均分
+            avg_score = scores.mean()
 
-            # 为每个学生生成图表并保存
-            zip_buffer = BytesIO()
-            with ZipFile(zip_buffer, mode="w") as zf:
-                for student in selected_students:
-                    fig_student = create_radar_chart(df, [student], selected_subjects)
-                    img_bytes = generate_radar_image(fig_student)
-                    if img_bytes is not None:
-                        file_path = os.path.join(temp_dir, f"{student}.png")
-                        with open(file_path, "wb") as f:
-                            f.write(img_bytes)
-                        zf.write(file_path, os.path.basename(file_path))
-            
-            # 下载Zip文件
+            # 生成雷達圖
+            fig = create_radar_chart(scores, avg_score, selected_student)
+            st.plotly_chart(fig)
+
+            # 提供下載按鈕
+            img_bytes = generate_radar_image(fig)
             st.download_button(
-                label="下载所有图表",
-                data=zip_buffer.getvalue(),
-                file_name="student_charts.zip",
-                mime="application/zip"
+                "下載圖片",
+                data=img_bytes,
+                file_name=f"student_{selected_student}_analysis.png",
+                mime="image/png"
             )
 
+        except Exception as e:
+            st.error(f"發生錯誤：{e}")
+
 if __name__ == "__main__":
     main()