Update app.py

app.py

@@ -53,81 +53,57 @@ 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')
             ),
-             angularaxis=dict(
-                 tickfont=dict(size=16, color='black', family="Microsoft JhengHei, Noto Sans CJK, Arial")
+            angularaxis=dict(
+                tickfont=dict(size=16, color='black')
             )
         ),
         showlegend=True,
         legend=dict(
-            font=dict(size=14, color='black', family="Microsoft JhengHei, Noto Sans CJK, Arial")
+            font=dict(size=14, color='black')
         ),
         title='學生成績雷達圖',
         plot_bgcolor='white',
         paper_bgcolor='white',
-        font=dict(family="Microsoft JhengHei, Noto Sans CJK, Arial")
-    )
-
-    return fig
-
-def apply_font_to_all_text(fig):
-    """強制設定圖表內所有文字元素的字型"""
-    for trace in fig.data:
-        if hasattr(trace, 'textfont'):
-            trace.textfont.family = "Microsoft JhengHei, Noto Sans CJK, Arial"
-        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")
     )
 
-    if hasattr(fig, 'layout') and hasattr(fig.layout, 'xaxis'):
-        fig.layout.xaxis.tickfont.family = "Microsoft JhengHei, Noto Sans CJK, Arial"
-    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"
-    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"
-
     return fig
 
-
 def save_all_radar_charts(df, selected_columns, comparison_rows):
     """批次生成並儲存每個學生與指定比較對象的雷達圖"""
     output_dir = "radar_charts"
     os.makedirs(output_dir, exist_ok=True)
+    font_path = "path/to/your/font/file.ttf"  # 請替換成你字體檔案的路徑
+    try:
+        font = ImageFont.truetype(font_path, size=12, encoding="utf-8")  # 設置中文字型
+    except FileNotFoundError:
+        st.error(f"找不到字型檔案: {font_path},請確保檔案路徑正確")
+        return None
 
     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)
-        
-        # 使用 to_image 取得圖表影像
         img_bytes = pio.to_image(fig, format="png", 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)
+        draw = ImageDraw.Draw(composite)
         
-        # 儲存圖片
+        #手動繪製圖表標籤文字
+        for trace in fig.data:
+            if isinstance(trace, go.Scatterpolar):
+                for i, theta in enumerate(trace.theta):
+                    x, y = fig.layout.polar.radialaxis.range[1] * (0.5 - 0.5*trace.r[i] / fig.layout.polar.radialaxis.range[1]) * np.cos(np.radians(90-theta)), fig.layout.polar.radialaxis.range[1] * (0.5 - 0.5*trace.r[i] / fig.layout.polar.radialaxis.range[1]) * np.sin(np.radians(90-theta))
+                    draw.text((img.width / 2 + x, img.height / 2 + y - 25), str(trace.name), font=font, fill=(0, 0, 0))
         file_path = os.path.join(output_dir, f"{student}_comparison.png")
         composite.save(file_path)
 
@@ -139,6 +115,7 @@ def save_all_radar_charts(df, selected_columns, comparison_rows):
 
     return zip_path
 
+
 def main():
     st.title('學生成績雷達圖產生器')
 
@@ -171,19 +148,21 @@ def main():
                     st.error("請至少選擇一個比較對象")
                 else:
                     zip_path = save_all_radar_charts(df, selected_columns, comparison_rows)
-                    with open(zip_path, "rb") as f:
-                        st.download_button(
-                            label="下載壓縮檔",
-                            data=f,
-                            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)
-                        )
-                    st.session_state.download_triggered = False
+                    if zip_path: #check the path is valid
+                         with open(zip_path, "rb") as f:
+                            st.download_button(
+                                label="下載壓縮檔",
+                                data=f,
+                                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)
+                            )
+                         st.session_state.download_triggered = False
 
 if __name__ == "__main__":
     if 'download_triggered' not in st.session_state:
         st.session_state.download_triggered = False
+    import numpy as np
     main()