Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 import asyncio
 import uvicorn
 from fastapi import FastAPI
@@ -7,6 +6,10 @@ from playwright.async_api import async_playwright
 from contextlib import asynccontextmanager
 
 # --- Configuration ---
+# Limit simultaneous browser instances to prevent memory crashes
+MAX_CONCURRENT_REQUESTS = 3
+semaphore = asyncio.Semaphore(MAX_CONCURRENT_REQUESTS)
+
 class GeminiBot:
     def __init__(self):
         self.playwright = None
@@ -14,50 +17,49 @@ class GeminiBot:
 
     async def start(self):
         self.playwright = await async_playwright().start()
-        # Launch browser with flags to help avoid detection
         self.browser = await self.playwright.chromium.launch(
-            headless=True, 
+            headless=True,
             args=["--no-sandbox", "--disable-blink-features=AutomationControlled"]
         )
-        print("--- Gemini Bot (Stateless) Ready ---")
+        print("--- Gemini Bot (Concurrent Stateless Mode) Ready ---")
 
     async def process_chat(self, message: str):
-        # Create context with a realistic User Agent
-        context = await self.browser.new_context(
-            user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
-        )
-        page = await context.new_page()
-        
-        try:
-            # BLOCK unnecessary assets to speed up loading
-            await page.route("**/*", lambda route: route.abort() if route.request.resource_type in 
-                             ["image", "font", "media", "stylesheet"] else route.continue_())
-            
-            # Navigate and wait for network idle to ensure JS loads
-            await page.goto("https://gemini.google.com", wait_until="networkidle")
-            
-            # Interact with the chat box
-            chat_box = page.locator("div[contenteditable='true']")
-            await chat_box.wait_for(timeout=20000)
-            await chat_box.fill(message)
-            await page.keyboard.press("Enter")
+        # The semaphore ensures only N requests run at once
+        async with semaphore:
+            context = await self.browser.new_context(
+                user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+            )
+            page = await context.new_page()
             
-            # Wait for the response container
-            await page.wait_for_selector("message-content", timeout=45000)
-            
-            # Extract Text
-            current_text = await page.evaluate("""() => {
-                const msgs = document.querySelectorAll('message-content');
-                return msgs.length > 0 ? msgs[msgs.length - 1].innerText : "Error: No message content found";
-            }""")
-            
-            return {"text": current_text}
-            
-        except Exception as e:
-            return {"text": f"Error: {str(e)}"}
-        finally:
-            # ALWAYS wipe the session
-            await context.close()
+            try:
+                # Block heavy assets for speed
+                await page.route("**/*", lambda route: route.abort() if route.request.resource_type in 
+                                 ["image", "font", "media", "stylesheet"] else route.continue_())
+                
+                await page.goto("https://gemini.google.com", wait_until="networkidle")
+                
+                # Logic: If not logged in, this will wait until timeout
+                chat_box = page.locator("div[contenteditable='true']")
+                await chat_box.wait_for(timeout=30000)
+                await chat_box.fill(message)
+                await page.keyboard.press("Enter")
+                
+                # Wait for the response container
+                await page.wait_for_selector("message-content", timeout=45000)
+                
+                # Extract the last message
+                current_text = await page.evaluate("""() => {
+                    const msgs = document.querySelectorAll('message-content');
+                    return msgs.length > 0 ? msgs[msgs.length - 1].innerText : "No response";
+                }""")
+                
+                return {"text": current_text}
+                
+            except Exception as e:
+                return {"text": f"Error: {str(e)}"}
+            finally:
+                # Session is wiped here, regardless of success or failure
+                await context.close()
 
     async def stop(self):
         if self.browser: await self.browser.close()
@@ -76,13 +78,9 @@ app = FastAPI(lifespan=lifespan)
 class ChatRequest(BaseModel):
     message: str
 
-@app.get("/")
-def health_check():
-    return {"status": "online", "message": "Gemini API is running"}
-
 @app.post("/chat")
 async def chat_api(request: ChatRequest):
-    # This now handles the Pydantic model correctly, resolving 422 errors
+    # This call is non-blocking to other concurrent requests
     return await bot.process_chat(request.message)
 
 if __name__ == "__main__":