Update main.py

main.py

@@ -6,14 +6,22 @@ import uvicorn
 from playwright.async_api import async_playwright
 from openclaw import Agent, Tool
 
+# --- CONFIG ---
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 SPACE_HOST = os.getenv("SPACE_HOST")
 STATIC_DIR = "static"
 
 app = FastAPI()
+
+# Create static directory if it doesn't exist
+if not os.path.exists(STATIC_DIR):
+    os.makedirs(STATIC_DIR)
+
 app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
 
+# --- TOOLS ---
 async def take_screenshot(url: str) -> str:
+    """Navigates to a URL and takes a screenshot. Returns the URL to view it."""
     filepath = f"{STATIC_DIR}/screenshot.png"
     async with async_playwright() as p:
         browser = await p.chromium.launch(headless=True)
@@ -21,28 +29,53 @@ async def take_screenshot(url: str) -> str:
         await page.goto(url)
         await page.screenshot(path=filepath, full_page=True)
         await browser.close()
-    return f"Screenshot saved. View here: https://{SPACE_HOST}/static/screenshot.png"
+    
+    return f"Screenshot successful. View it here: https://{SPACE_HOST}/static/screenshot.png"
 
 tools = [Tool(name="take_screenshot", func=take_screenshot, description="Takes a screenshot of a URL.")]
 
+# --- AGENT ---
 agent = Agent(
     model="gemini/gemini-2.5-pro",
     tools=tools,
-    system_prompt="Expert DevOps agent. Concise replies (<1000 chars). Explain before acting.",
+    system_prompt=(
+        "You are an expert coding/devops agent. Users interact via a WhatsApp bridge. "
+        "Keep replies concise and under 1000 chars. Explain before acting. "
+        "If you take a screenshot, provide the URL clearly."
+    ),
     model_kwargs={"temperature": 0.2, "max_tokens": 8192, "top_p": 0.95}
 )
 
+# --- ENDPOINTS ---
 @app.get("/health")
-async def health(): return {"status": "online"}
+async def health_check():
+    return {"status": "alive"}
 
-@app.post("/webhook")
+@app.api_route("/webhook", methods=["GET", "POST"])
 async def wa_webhook(request: Request):
-    data = await request.json()
-    user_msg = data.get("message", "")
+    """Universal webhook that handles both GET and POST requests."""
+    user_msg = ""
+    
+    # Try to get message from URL (GET)
+    user_msg = request.query_params.get("message", "")
+    
+    # Try to get message from JSON Body (POST)
+    if request.method == "POST":
+        try:
+            data = await request.json()
+            user_msg = data.get("message", user_msg)
+        except:
+            pass
+
+    # Process prompt if it starts with "claw "
     if user_msg.lower().startswith("claw "):
         prompt = user_msg[5:].strip()
-        response = await agent.arun(prompt)
-        return {"replies": [{"message": response}]}
+        try:
+            response_text = await agent.arun(prompt)
+            return {"replies": [{"message": response_text}]}
+        except Exception as e:
+            return {"replies": [{"message": f"Agent Error: {str(e)}"}]}
+    
     return {"replies": []}
 
 if __name__ == "__main__":