Update app.py

app.py

@@ -4,19 +4,41 @@ import gradio as gr
 import aisuite as ai
 import zipfile
 import time # 為了讓檔案解壓縮有時間
-# 移除 Colab 專用的 userdata 匯入
 
 # RAG 相關
 from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import HuggingFaceEmbeddings
 
+from huggingface_hub import login
+# 移除 Colab 專用的 userdata 匯入
+
+# =========================================================
+# 1. 檔案解壓縮與環境變數載入 (Hugging Face Secrets 會自動注入 OS 環境變數)
+# =========================================================
+
+# 1.0 透過 Token 登入 Hugging Face (【新增的關鍵程式碼】)
 # =========================================================
 # 1. 檔案解壓縮與環境變數載入 (Hugging Face Secrets 會自動注入 OS 環境變數)
 # =========================================================
 
-# Hugging Face 環境中，Secrets 會被注入為環境變數。
-# 我們不再需要 google.colab import userdata 來獲取 Key。
-# 程式將直接從 os.environ 讀取。
+# 1.0 透過 Token 登入 Hugging Face (【關鍵修正區塊】)
+groq_api_key = os.environ.get('GROQ')
+hf_token = os.environ.get('HuggingFace')
+
+if hf_token:
+    try:
+        login(token=hf_token)
+        print("✅ Hugging Face 登入成功，已驗證模型存取權。")
+    except Exception as e:
+        print(f"❌ 警告：Hugging Face Token 登入失敗。請檢查 Secrets 裡面的 'HuggingFace' Token 是否有讀取權限或是否過期。錯誤: {e}")
+
+if not groq_api_key:
+    print("❌ 警告：GROQ API Key 未設定在 Hugging Face Secrets 中！")
+    
+# 由於 aisuite 依賴 os.environ，我們在這裡確保 Groq Key 再次被設定
+os.environ['GROQ_API_KEY'] = groq_api_key if groq_api_key else ""
+
+# 1.1 啟動時解壓縮 FAISS 資料庫
 
 # 1.1 啟動時解壓縮 FAISS 資料庫
 if not os.path.exists("faiss_dbV2"):
@@ -30,8 +52,8 @@ if not os.path.exists("faiss_dbV2"):
         exit()
 
 # 1.2 檢查 API Key
-groq_api_key = os.environ.get('GROQ') # 注意：從 Secrets 讀取時 Key 要大寫，如 GROQ
-hf_token = os.environ.get('HuggingFace')
+#groq_api_key = os.environ.get('GROQ') # 注意：從 Secrets 讀取時 Key 要大寫，如 GROQ
+#hf_token = os.environ.get('HuggingFace')
 
 if not groq_api_key:
     print("❌ 警告：GROQ API Key 未設定在 Hugging Face Secrets 中！")