Update main.py

main.py

@@ -1,35 +1,62 @@
 import os
-import httpx
 from fastapi import FastAPI, Request
 import uvicorn
+from litellm import acompletion
 
-# --- CONFIG ---
-TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN", "")
-TELEGRAM_API_URL = f"https://api.telegram.org/bot{TELEGRAM_TOKEN}"
+# Strip the API key to kill any hidden spaces or newlines!
+os.environ["GEMINI_API_KEY"] = os.getenv("GEMINI_API_KEY", "").strip()
 
 app = FastAPI()
 
+# --- MEMORY BANK ---
+user_engines = {}
+
+async def run_agent(prompt: str, model_name: str) -> str:
+    try:
+        response = await acompletion(
+            model=model_name, 
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.2
+        )
+        return response.choices[0].message.content or "No response."
+    except Exception as e:
+        return f"⚠️ Engine Error: {str(e)}"
+
 @app.api_route("/webhook", methods=["GET", "POST", "PUT"])
 async def telegram_webhook(request: Request):
+    if request.method == "GET":
+        return {"status": "alive"}
+        
     try:
         data = await request.json()
-        print(f"DEBUG 1 (INCOMING): {data}") # Prints what Telegram sent
         
         if "message" in data and "text" in data["message"]:
             chat_id = data["message"]["chat"]["id"]
+            user_msg = data["message"]["text"].strip()
+            
+            # 1. GEAR SHIFTER
+            if user_msg.lower() == "/pro":
+                user_engines[chat_id] = "gemini/gemini-1.5-pro"
+                return {"method": "sendMessage", "chat_id": chat_id, "text": "🧠 Engine Switched: Gemini 1.5 Pro (50 free requests/day)."}
+            elif user_msg.lower() == "/flash":
+                user_engines[chat_id] = "gemini/gemini-1.5-flash"
+                return {"method": "sendMessage", "chat_id": chat_id, "text": "⚡ Engine Switched: Gemini 1.5 Flash (1,500 free requests/day)."}
+            
+            # 2. RUN AI
+            current_model = user_engines.get(chat_id, "gemini/gemini-1.5-flash") 
+            response_text = await run_agent(user_msg, current_model)
+            
+            # 3. DIRECT WEBHOOK REPLY
+            # We respond directly to Telegram's knock. No background tasks, no httpx, no crashing.
+            return {
+                "method": "sendMessage",
+                "chat_id": chat_id,
+                "text": response_text
+            }
             
-            # Send a hardcoded reply immediately, bypassing all AI
-            async with httpx.AsyncClient() as client:
-                res = await client.post(
-                    f"{TELEGRAM_API_URL}/sendMessage",
-                    json={"chat_id": chat_id, "text": "PONG! The server connection is 100% working."}
-                )
-                print(f"DEBUG 2 (OUTGOING STATUS): {res.status_code}") # Prints if Telegram accepted it
-                print(f"DEBUG 3 (OUTGOING MSG): {res.text}")
-                
         return {"status": "ok"}
     except Exception as e:
-        print(f"DEBUG CRASH: {e}")
+        print(f"CRASH: {repr(e)}")
         return {"status": "error"}
 
 if __name__ == "__main__":