First

api.py

@@ -10,7 +10,7 @@ import io
 from google import genai
 from google.genai import types
 import computer_control_helper
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel, Field
 from seleniumbase import Driver
@@ -21,6 +21,7 @@ from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException, NoSuchElementException, StaleElementReferenceException
 import random
 from streaming import StreamProcessor, create_streaming_response, StreamConfig, StreamingResponseGenerator
+import base64
 
 # Virtual display setup for Linux headless environments
 import platform
@@ -644,6 +645,21 @@ async def lifespan(app: FastAPI):
 
 app = FastAPI(lifespan=lifespan)
 
+@app.middleware("http")
+async def error_screenshot_middleware(request: Request, call_next):
+    try:
+        return await call_next(request)
+    except Exception as exc:
+        logger.error(f"Unhandled exception: {exc}", exc_info=True)
+        try:
+            driver = driver_manager.get_driver()
+            png = driver.get_screenshot_as_png()
+            b64 = base64.b64encode(png).decode('utf-8')
+            logger.error(f"Screenshot base64: {b64}")
+        except Exception as capture_err:
+            logger.error(f"Failed to capture screenshot: {capture_err}", exc_info=True)
+        raise
+
 @app.get("/health")
 async def health_check():
     logger.info("Health check endpoint called.")

test_api.py

@@ -8,7 +8,7 @@ import random
 # Configure the OpenAI client to point to the local server.
 # The API key is required by the library but not used by the local server.
 client = OpenAI(
-    base_url="http://localhost:8000",
+    base_url="https://solarumasteridion-lenith.hf.space/",
     api_key=os.environ.get("OPENAI_API_KEY", "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx") 
 )