| 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（無法列出模型）。"
|
|
|
|
|
| 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)
|
|
|
|
|
| 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")
|
|
|
