Update app.py

app.py

@@ -4,12 +4,14 @@ import plotly.graph_objects as go
 import os
 from zipfile import ZipFile
 import plotly.io as pio
+from PIL import Image, ImageDraw, ImageFont
+from io import BytesIO
+
 try:
     import kaleido
 except ImportError:
     st.error("請安裝 'kaleido' 套件以啟用圖像導出功能：\n\n    $ pip install kaleido")
 
-
 def load_data(uploaded_file):
     """載入並處理CSV檔案"""
     try:
@@ -29,7 +31,6 @@ def load_data(uploaded_file):
         st.error(f"載入檔案時發生錯誤：{e}")
         return None
 
-
 def create_radar_chart(df, selected_rows, selected_columns):
     """使用Plotly建立雷達圖"""
     line_styles = ['solid', 'dot', 'dash', 'longdash', 'dashdot']
@@ -52,26 +53,26 @@ def create_radar_chart(df, selected_rows, selected_columns):
             ),
             marker=dict(opacity=0.5)
         ))
-
+        
     fig.update_layout(
-        polar=dict(
+         polar=dict(
             radialaxis=dict(
                 visible=True,
                 range=[0, df[selected_columns].values.max() * 1.1],
-                tickfont=dict(size=12, color='black', family="Microsoft JhengHei, Noto Sans CJK, Arial")  # 調整刻度文字顏色為黑色並設定字體
+                tickfont=dict(size=12, color='black', family="Microsoft JhengHei, Noto Sans CJK, Arial")
             ),
-            angularaxis=dict(
-                tickfont=dict(size=16, color='black', family="Microsoft JhengHei, Noto Sans CJK, Arial")  # 放大字體，顏色為黑色並設定字體
+             angularaxis=dict(
+                 tickfont=dict(size=16, color='black', family="Microsoft JhengHei, Noto Sans CJK, Arial")
             )
         ),
         showlegend=True,
         legend=dict(
-            font=dict(size=14, color='black', family="Microsoft JhengHei, Noto Sans CJK, Arial")  # 圖例文字加大並改為黑色並設定字體
+            font=dict(size=14, color='black', family="Microsoft JhengHei, Noto Sans CJK, Arial")
         ),
         title='學生成績雷達圖',
-        plot_bgcolor='white',  # 背景為白色
-        paper_bgcolor='white',  # 紙張背景為白色
-        font=dict(family="Microsoft JhengHei, Noto Sans CJK, Arial")  # 字型設定
+        plot_bgcolor='white',
+        paper_bgcolor='white',
+        font=dict(family="Microsoft JhengHei, Noto Sans CJK, Arial")
     )
 
     return fig
@@ -84,8 +85,8 @@ def apply_font_to_all_text(fig):
         if hasattr(trace, 'marker') and hasattr(trace.marker, 'textfont'):
             trace.marker.textfont.family = "Microsoft JhengHei, Noto Sans CJK, Arial"
     fig.update_layout(
-       font=dict(
-           family="Microsoft JhengHei, Noto Sans CJK, Arial"
+        font=dict(
+            family="Microsoft JhengHei, Noto Sans CJK, Arial"
         )
     )
 
@@ -94,12 +95,13 @@ def apply_font_to_all_text(fig):
     if hasattr(fig, 'layout') and hasattr(fig.layout, 'yaxis'):
         fig.layout.yaxis.tickfont.family = "Microsoft JhengHei, Noto Sans CJK, Arial"
     if hasattr(fig, 'layout') and hasattr(fig.layout, 'polar') and hasattr(fig.layout.polar, 'radialaxis'):
-        fig.layout.polar.radialaxis.tickfont.family = "Microsoft JhengHei, Noto Sans CJK, Arial"
+            fig.layout.polar.radialaxis.tickfont.family = "Microsoft JhengHei, Noto Sans CJK, Arial"
     if hasattr(fig, 'layout') and hasattr(fig.layout, 'polar') and hasattr(fig.layout.polar, 'angularaxis'):
-        fig.layout.polar.angularaxis.tickfont.family = "Microsoft JhengHei, Noto Sans CJK, Arial"
+            fig.layout.polar.angularaxis.tickfont.family = "Microsoft JhengHei, Noto Sans CJK, Arial"
 
     return fig
 
+
 def save_all_radar_charts(df, selected_columns, comparison_rows):
     """批次生成並儲存每個學生與指定比較對象的雷達圖"""
     output_dir = "radar_charts"
@@ -108,17 +110,26 @@ def save_all_radar_charts(df, selected_columns, comparison_rows):
     for student in df['姓名']:
         if student in comparison_rows:
             continue
-        
+
+        # 建立圖表
         fig = create_radar_chart(df, [student] + comparison_rows, selected_columns)
         fig = apply_font_to_all_text(fig)
         
-        # 在生成圖片前重新設定 kaleido 的 default_font
-        pio.kaleido.scope.default_font = "Microsoft JhengHei, Noto Sans CJK, Arial"
+        # 使用 to_image 取得圖表影像
+        img_bytes = pio.to_image(fig, format="png", engine="kaleido")
 
-        file_path = os.path.join(output_dir, f"{student}_comparison.png")
-        
-        fig.write_image(file_path, engine="kaleido")
+        # 使用 PIL 打開影像
+        img = Image.open(BytesIO(img_bytes)).convert("RGBA")
 
+        # 建立空白背景
+        background = Image.new('RGBA', img.size, (255, 255, 255, 255))
+
+        # 合成圖片
+        composite = Image.alpha_composite(background, img)
+        
+        # 儲存圖片
+        file_path = os.path.join(output_dir, f"{student}_comparison.png")
+        composite.save(file_path)
 
     # 壓縮所有圖表
     zip_path = "radar_charts.zip"
@@ -128,7 +139,6 @@ def save_all_radar_charts(df, selected_columns, comparison_rows):
 
     return zip_path
 
-
 def main():
     st.title('學生成績雷達圖產生器')
 
@@ -168,13 +178,12 @@ def main():
                             file_name="radar_charts.zip",
                             mime="application/zip",
                             key="download_button",
-                            on_click=lambda: st.session_state.update({'download_triggered':True}),  # 设置下载触发标记
-                            disabled=st.session_state.get("download_triggered",False) # 如果已下载则禁用下载按钮
+                            on_click=lambda: st.session_state.update({'download_triggered':True}),
+                            disabled=st.session_state.get("download_triggered",False)
                         )
                     st.session_state.download_triggered = False
 
 if __name__ == "__main__":
     if 'download_triggered' not in st.session_state:
         st.session_state.download_triggered = False
-
     main()