Upload app.py

app.py

@@ -0,0 +1,98 @@
+import os
+import pkg_resources
+import gradio as gr
+
+# ============ 工具函式 ============
+def detect_sdk():
+    """偵測目前環境中與 Google 相關套件的版本與 SDK 狀態"""
+    pkgs = []
+    has_new = has_old = False
+    for dist in pkg_resources.working_set:
+        name = dist.project_name
+        lower = name.lower()
+        if "google" in lower:
+            pkgs.append(f"{name} {dist.version}")
+        if lower == "google-genai":
+            has_new = True
+        if lower == "google-generativeai":
+            has_old = True
+
+    if has_new and not has_old:
+        sdk = "google-genai（新 SDK, v1）"
+        sdk_tag = "new"
+    elif has_old and not has_new:
+        sdk = "google-generativeai（舊 SDK, v1beta）"
+        sdk_tag = "old"
+    elif has_new and has_old:
+        sdk = "同時裝了新舊 SDK（建議只留一個以免衝突）"
+        sdk_tag = "both"
+    else:
+        sdk = "無法判定（可能沒安裝任何 Google GenAI SDK）"
+        sdk_tag = "none"
+
+    return sdk, sdk_tag, "\n".join(sorted(pkgs))
+
+def list_models_if_possible():
+    """若有新 SDK + API Key，嘗試列出可用模型；否則回傳提示訊息"""
+    api_key = os.getenv("GEMINI_API_KEY")
+    if not api_key:
+        return "⚠️ 未設定 GEMINI_API_KEY（無法列出模型）。"
+
+    # 盡量用新 SDK；若不存在就回訊息
+    try:
+        from google import genai
+    except Exception as e:
+        return f"⚠️ 未安裝新 SDK（google-genai）。無法 list models。\n{e}"
+
+    try:
+        client = genai.Client(api_key=api_key)
+        names = [m.name for m in client.models.list()]
+        if not names:
+            return "（沒有列到任何模型）"
+        return "可用模型：\n" + "\n".join(names)
+    except Exception as e:
+        return f"⚠️ 列模型失敗：{e}"
+
+def quick_health_check():
+    """快速健康檢查：API Key 是否存在、推薦模型名是否設定"""
+    api_key_ok = bool(os.getenv("GEMINI_API_KEY"))
+    model = os.getenv("GEMINI_MODEL", "（未設定，預設建議 gemini-2.5-flash）")
+    lines = []
+    lines.append(f"API Key 設定：{'✅ 已設定' if api_key_ok else '❌ 未設定'}（環境變數 GEMINI_API_KEY）")
+    lines.append(f"模型名稱：{model}")
+    lines.append("建議：gemini-2.5-flash 或 gemini-2.5-pro")
+    return "\n".join(lines)
+
+# ============ Gradio 介面邏輯 ============
+def refresh_status():
+    sdk, tag, pkgs = detect_sdk()
+    health = quick_health_check()
+    return sdk, pkgs, health
+
+def do_list_models():
+    return list_models_if_possible()
+
+with gr.Blocks(title="Gemini SDK 偵測面板 (HF Space)") as demo:
+    gr.Markdown("## 🔎 Gemini SDK 偵測面板（Hugging Face Space）\n"
+                "啟動後自動檢查你目前是跑 **新 SDK** 還是 **舊 SDK**，以及常見設定。")
+
+    with gr.Row():
+        with gr.Column():
+            sdk_label = gr.Textbox(label="目前偵測到的 SDK", interactive=False)
+            pkgs_box = gr.Textbox(label="與 Google 相關的已安裝套件（含版本）", lines=10, interactive=False)
+            health_box = gr.Textbox(label="快速健康檢查", lines=4, interactive=False)
+
+            btn_refresh = gr.Button("🔄 重新偵測")
+            btn_refresh.click(fn=refresh_status, outputs=[sdk_label, pkgs_box, health_box])
+
+        with gr.Column():
+            gr.Markdown("### 📜 列出可用模型（需要新 SDK + API Key）")
+            models_box = gr.Textbox(label="可用模型清單", lines=12, interactive=False)
+            btn_models = gr.Button("📋 List Models")
+            btn_models.click(fn=do_list_models, outputs=models_box)
+
+    # 啟動時自動偵測一次
+    demo.load(fn=refresh_status, outputs=[sdk_label, pkgs_box, health_box])
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0")