Upload deepseek_client.py

deepseek_client.py

@@ -6,15 +6,22 @@ load_dotenv()
 
 # Get API key from Streamlit secrets or environment
 def get_api_key():
-    if hasattr(st, 'secrets') and "DEEPSEEK_API_KEY" in st.secrets:
-        return st.secrets["DEEPSEEK_API_KEY"]
+    try:
+        if hasattr(st, 'secrets') and "DEEPSEEK_API_KEY" in st.secrets:
+            return st.secrets["DEEPSEEK_API_KEY"]
+    except:
+        pass
     return os.getenv("DEEPSEEK_API_KEY")
 
-DEEPSEEK_API_KEY = get_api_key()
+# Don't call get_api_key() at import time - call it when needed
+DEEPSEEK_API_KEY = None
 BASE = os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com")
 
 def chat(messages, model="deepseek-chat", temperature=0.9):
-    headers = {"Authorization": f"Bearer {DEEPSEEK_API_KEY}", "Content-Type": "application/json"}
+    api_key = get_api_key()
+    if not api_key:
+        raise ValueError("DEEPSEEK_API_KEY not found. Please set it in Hugging Face Space secrets or environment variables.")
+    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
     payload = {"model": model, "messages": messages, "temperature": temperature}
     r = requests.post(f"{BASE}/chat/completions", headers=headers, data=json.dumps(payload), timeout=60)
     r.raise_for_status()