Update main.py

main.py

@@ -172,8 +172,13 @@ QUESTION:
 {user_query}
 ANSWER:"""
 
+    # Verificar si existe la API Key como secreto
+    api_key = st.secrets.get("OPENROUTER_API_KEY", None)
+    if not api_key:
+        return "❌ Error: Missing `OPENROUTER_API_KEY` in Hugging Face Space secrets."
+
     headers = {
-        "Authorization": f"Bearer {st.secrets['OPENROUTER_API_KEY']}",
+        "Authorization": f"Bearer {api_key}",
         "HTTP-Referer": "https://yourappname.streamlit.app",
         "X-Title": "PoliticBot"
     }
@@ -189,18 +194,22 @@ ANSWER:"""
         "top_p": 0.95
     }
 
-    response = httpx.post(
-        "https://openrouter.ai/api/v1/chat/completions",
-        headers=headers,
-        json=payload,
-        timeout=60
-    )
-
-    if response.status_code != 200:
-        return f"❌ Error {response.status_code}: {response.text}"
+    try:
+        response = httpx.post(
+            "https://openrouter.ai/api/v1/chat/completions",
+            headers=headers,
+            json=payload,
+            timeout=60
+        )
+        response.raise_for_status()
+    except httpx.RequestError as e:
+        return f"❌ Connection error: {e}"
+    except httpx.HTTPStatusError as e:
+        return f"❌ API error: {e.response.status_code} - {e.response.text}"
 
     return response.json()["choices"][0]["message"]["content"].strip()
 
+
 # ---------------------- STREAMLIT INTERFACE ----------------------
 
 st.image('portada3.jpg', use_container_width=True)