Update main.py

main.py

@@ -11,38 +11,59 @@ TELEGRAM_API_URL = f"https://api.telegram.org/bot{TELEGRAM_TOKEN}"
 
 app = FastAPI()
 
+# --- NATIVE AGENT ENGINE ---
 async def run_agent(prompt: str) -> str:
     try:
         response = await acompletion(
-            model="gemini/1.5-pro", 
+            model="gemini/gemini-2.5-pro", # Locked onto 2.5 Pro
             messages=[{"role": "user", "content": prompt}],
             temperature=0.2
         )
-        return response.choices[0].message.content or "No response."
+        return response.choices[0].message.content or "No response generated."
     except Exception as e:
         return f"⚠️ Engine Error: {str(e)}"
 
+# --- BACKGROUND TASK: PROCESS AND REPLY ---
 async def process_and_reply(chat_id: int, text: str):
     response_text = await run_agent(text)
+    
+    # Send the final response back to Telegram
     async with httpx.AsyncClient() as client:
-        await client.post(f"{TELEGRAM_API_URL}/sendMessage", json={"chat_id": chat_id, "text": response_text})
+        await client.post(
+            f"{TELEGRAM_API_URL}/sendMessage",
+            json={
+                "chat_id": chat_id, 
+                "text": response_text
+            }
+        )
 
-# Accept EVERY method to prevent "Method Not Allowed"
+# --- ENDPOINTS ---
+@app.get("/health")
+async def health_check():
+    return {"status": "alive"}
+
+# Accept EVERY method to prevent "Method Not Allowed" from HF routing
 @app.api_route("/webhook", methods=["GET", "POST", "PUT"])
 async def telegram_webhook(request: Request, background_tasks: BackgroundTasks):
-    # If it's a GET request (like you testing in a browser), just say hi
+    # If it's a GET request (like a browser test), just confirm it's alive
     if request.method == "GET":
         return {"status": "Webhook is active. Send a POST request from Telegram!"}
     
     try:
         data = await request.json()
+        
+        # Extract the message from Telegram's JSON payload
         if "message" in data and "text" in data["message"]:
             chat_id = data["message"]["chat"]["id"]
             user_msg = data["message"]["text"]
+            
+            # Run the AI generation in the background so we can instantly reply "OK" to Telegram
             background_tasks.add_task(process_and_reply, chat_id, user_msg)
+            
+        # Telegram requires an immediate 200 OK so it doesn't get stuck in a retry loop
         return {"status": "ok"}
-    except:
-        return {"status": "error reading json"}
+    except Exception as e:
+        return {"status": f"error reading json: {str(e)}"}
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)