Update app.py

app.py

@@ -112,7 +112,6 @@ def save_radar_chart_image(fig):
   """使用 kaleido 輸出 png 的記憶體檔案"""
   img_bytes = pio.to_image(fig, format="png", engine="kaleido")
   return img_bytes
-
 def create_composite_image(fig, student):
     """使用 PIL 合成圖片，確保學生的成績在最上層"""
     img_bytes = pio.to_image(fig, format="png", engine="kaleido")
@@ -144,30 +143,41 @@ def main():
                     st.plotly_chart(fig, use_container_width=True)
                 except Exception as e:
                     st.error(f"生成雷達圖時發生錯誤：{e}")
-                    
-            st.write("### 批次繪製個別學生比較圖")
-            
-            individual_students = st.multiselect("選擇要個別比較的學生", df['姓名'].tolist(), key = "student")
-            comparison_items = st.multiselect("選擇要比較的項目", df['姓名'].tolist(), key = "item")
-          
-            if individual_students and comparison_items:
-                
-                image_options = []
-                image_bytes = {}
-                for student in individual_students:
-                    fig = create_radar_chart(df, [student] + comparison_items, selected_columns)
-                    image_bytes[student] = create_composite_image(fig,student)
-                    image_options.append(f"{student} 與 {', '.join(comparison_items)} 的比較")
-                    
-                selected_image_options = st.multiselect("選擇要顯示的圖片", options=image_options)
-                
-                cols = st.columns(3) # 排成三列
-                for i, option in enumerate(selected_image_options):
-                  student = option.split(" 與 ")[0]
-                  with cols[i%3]:
-                      st.image(image_bytes[student], use_container_width=True)
-                      st.text(option)
 
+            st.write("### 批次下載全班學生雷達圖")
+
+            comparison_item = st.selectbox("選擇要比較的項目", df['姓名'].tolist())
+
+            if comparison_item:
+                image_bytes = {}
+                for student in df['姓名']:
+                    if student == comparison_item:
+                       continue
+
+                    fig = create_radar_chart(df, [student, comparison_item], selected_columns)
+                    fig = apply_font_to_all_text(fig)
+                    image_bytes[student] = create_composite_image(fig, student)
+
+                cols = st.columns(3)  # 排成三列
+                for i, student in enumerate(image_bytes.keys()):
+                    with cols[i%3]:
+                        st.image(image_bytes[student], use_container_width=True)
+                        st.text(f"{student} 與 {comparison_item} 的比較")
+                        
+                if st.button("下載全班雷達圖"):
+                    zip_path = "all_radar_charts.zip"
+                    with ZipFile(zip_path, 'w') as zipf:
+                        for student, img in image_bytes.items():
+                             img_bytes = BytesIO()
+                             img.save(img_bytes, format="png")
+                             zipf.writestr(f"{student}_vs_{comparison_item}.png", img_bytes.getvalue())
+                    with open(zip_path, "rb") as f:
+                        st.download_button(
+                            label="下載全班雷達圖",
+                            data=f,
+                            file_name="all_radar_charts.zip",
+                            mime="application/zip"
+                        )
 if __name__ == "__main__":
     import numpy as np
     main()