Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 os.environ["HF_HOME"] = "/tmp/.cache/huggingface"
 os.environ["MPLCONFIGDIR"] = "/tmp/.config/matplotlib"
+
 import streamlit as st
 import joblib
 import pandas as pd
@@ -10,26 +11,39 @@ import platform
 from huggingface_hub import hf_hub_download
 import matplotlib.font_manager as fm
 
-# 列出可用字型，確認是否有 Noto Sans CJK TC
-print([f.name for f in fm.fontManager.ttflist if "Noto" in f.name])
-
-# 設定字型
-plt.rcParams['font.family'] = 'Heiti TC'
-plt.rcParams['axes.unicode_minus'] = False
+# 自訂函數：嘗試找系統中可用的中文字型路徑
+def find_chinese_font():
+    for font_path in fm.findSystemFonts(fontext='ttf'):
+        # 這裡可依你系統字型名關鍵字調整
+        if ("NotoSans" in font_path and ("TC" in font_path or "TraditionalChinese" in font_path)) \
+           or "STHeiti" in font_path or "Heiti" in font_path or "LiHei" in font_path:
+            return font_path
+    return None
+
+chinese_font_path = find_chinese_font()
+if chinese_font_path:
+    chinese_font_prop = fm.FontProperties(fname=chinese_font_path)
+    print(f"使用中文字型檔: {chinese_font_path}")
+else:
+    chinese_font_prop = None
+    print("找不到適合的中文字型檔，請確認系統已安裝中文字型")
 
-# 跨平台字型設定
+# matplotlib 全局字型設定（Fallback用）
 if platform.system() == 'Windows':
     plt.rcParams['font.family'] = 'Microsoft JhengHei'
 elif platform.system() == 'Darwin':  # macOS
     plt.rcParams['font.family'] = 'AppleGothic'
 else:
-    plt.rcParams['font.family'] = 'Noto Sans CJK TC'  # Linux
+    # Linux 預設用找到的字型名稱，沒找到就用 DejaVu Sans
+    if chinese_font_prop:
+        plt.rcParams['font.family'] = chinese_font_prop.get_name()
+    else:
+        plt.rcParams['font.family'] = 'DejaVu Sans'
 
-plt.rcParams['axes.unicode_minus'] = False  # 負號使用 ASCII 減號
+plt.rcParams['axes.unicode_minus'] = False  # 負號用 ASCII 減號
 
 @st.cache_resource(show_spinner=True)
 def load_model_and_explainer():
-    # 下載模型與 LabelEncoder
     model_path = hf_hub_download(
         repo_id="jung-ming/Ocean-Meets-Forest",
         filename="rf_model_with_encoder.pkl",
@@ -38,15 +52,11 @@ def load_model_and_explainer():
     bundle = joblib.load(model_path)
     model = bundle["model"]
     le = bundle["label_encoder"]
-
-    # 建立 explainer（避免用 pickle 載入 Numba 編譯物件）
     explainer = shap.TreeExplainer(model, feature_perturbation="interventional")
-
     return model, le, explainer
 
 model, le, explainer = load_model_and_explainer()
 
-# 建立映射
 ship_type_to_code = dict(zip(le.classes_, le.transform(le.classes_)))
 
 st.title("🚢 台中港艘次預測系統")
@@ -70,18 +80,16 @@ if st.button("🔮 開始預測"):
 
     st.subheader("🧠 模型決策解釋圖（SHAP Waterfall plot）")
 
-    # 計算 SHAP 值
     shap_values = explainer(input_df)
-
-    # 取得 Axes（shap.plots.waterfall 回傳的是 Axes）
     ax = shap.plots.waterfall(shap_values[0], show=False)
 
-    # 設定中文字型
-    for text in ax.texts:
-        text.set_fontname("Noto Sans CJK TC")
-        if text.get_text().startswith('\u2212'):
-            text.set_text(text.get_text().replace('\u2212', '-'))
+    # 強制為每個文字物件設定中文字型，避免顯示方塊
+    if chinese_font_prop:
+        for text in ax.texts:
+            text.set_fontproperties(chinese_font_prop)
+            # 修正負號顯示問題（使用 ASCII 減號）
+            if text.get_text().startswith('\u2212'):
+                text.set_text(text.get_text().replace('\u2212', '-'))
 
-    # 顯示圖表
     st.pyplot(ax.figure)
     plt.close(ax.figure)