Update main.py

main.py

@@ -31,31 +31,48 @@ HEADERS = {
     "X-Title": "My Free AI App"
 }
 
-MODEL = "google/gemini-2.0-flash-exp"  # FREE & fast
+MODEL = "meta-llama/llama-3.3-70b-instruct:free"  # FREE & fast
 
 # ---------------------------
 # Routes
 # ---------------------------
+
 @app.get("/")
 def home():
     return {"message": "Backend is working"}
 
 @app.get("/chat")
 def chat(prompt: str = Query(...)):
+    if not OPENROUTER_API_KEY:
+        return {"error": "OPENROUTER_API_KEY not set in Hugging Face Secrets"}
+
+    payload = {
+        "model": MODEL,
+        "messages": [
+            {"role": "user", "content": prompt}
+        ]
+    }
+
     try:
-        payload = {
-            "model": MODEL,
-            "messages": [
-                {"role": "user", "content": prompt}
-            ]
-        }
-
-        response = requests.post(API_URL, headers=HEADERS, json=payload, timeout=30)
-        data = response.json()
+        response = requests.post(
+            API_URL,
+            headers=HEADERS,
+            json=payload,
+            timeout=30
+        )
 
+        if response.status_code != 200:
+            return {
+                "error": "OpenRouter API error",
+                "status_code": response.status_code,
+                "details": response.text
+            }
+
+        data = response.json()
         reply = data["choices"][0]["message"]["content"]
 
         return {"reply": reply}
 
     except Exception as e:
         return {"error": str(e)}
+