Update app.py

app.py

@@ -1,6 +1,6 @@
 """
-Ultimate Web Agent – Production-Ready with Fast Startup
-Fixes: forced headless, Playwright health check, semaphore, env vars for keys
+Ultimate Web Agent – Production-Ready for Hugging Face Spaces
+Fixed: forced headless, semaphore, env vars, no startup crash
 """
 import os
 import json
@@ -204,7 +204,7 @@ async def run_agent(sid, task, headless):
         browser = None
         try:
             playwright = await async_playwright().start()
-            # Force headless=True regardless of input (HF Spaces have no display)
+            # Force headless=True regardless of input
             browser = await playwright.chromium.launch(headless=True)
             context = await browser.new_context(viewport={"width": 1280, "height": 720})
             page = await context.new_page()
@@ -214,7 +214,6 @@ async def run_agent(sid, task, headless):
             agent = Agent(tools, state)
             result = await agent.run(task)
             status[sid] = {"status": "completed", "result": result, "progress": 100}
-            # Notify WebSocket clients
             if sid in ws_connections:
                 for ws in ws_connections[sid]:
                     try:
@@ -420,24 +419,9 @@ async def health():
     return {"status": "healthy", "running": True}
 
 # =====================================================================
-# STARTUP: Verify Playwright works before uvicorn starts
+# STARTUP: NO health check – browser will be tested on first task
+# This prevents early crashes due to Playwright environment issues.
 # =====================================================================
-async def test_playwright():
-    """Test that Playwright can launch a headless browser."""
-    try:
-        async with async_playwright() as p:
-            browser = await p.chromium.launch(headless=True)
-            page = await browser.new_page()
-            await page.goto("about:blank")
-            await browser.close()
-        logger.info("✅ Playwright health check passed.")
-    except Exception as e:
-        logger.critical(f"❌ Playwright startup test failed: {e}")
-        raise RuntimeError("Playwright is not functional") from e
-
-# Only run the test when executed directly (not when imported)
 if __name__ == "__main__":
-    # Run the health check
-    asyncio.run(test_playwright())
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)