Create main.py

main.py

@@ -0,0 +1,49 @@
+import os
+import asyncio
+from fastapi import FastAPI, Request
+from fastapi.staticfiles import StaticFiles
+import uvicorn
+from playwright.async_api import async_playwright
+from openclaw import Agent, Tool
+
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+SPACE_HOST = os.getenv("SPACE_HOST")
+STATIC_DIR = "static"
+
+app = FastAPI()
+app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
+
+async def take_screenshot(url: str) -> str:
+    filepath = f"{STATIC_DIR}/screenshot.png"
+    async with async_playwright() as p:
+        browser = await p.chromium.launch(headless=True)
+        page = await browser.new_page()
+        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"
+
+tools = [Tool(name="take_screenshot", func=take_screenshot, description="Takes a screenshot of a URL.")]
+
+agent = Agent(
+    model="gemini/gemini-2.5-pro",
+    tools=tools,
+    system_prompt="Expert DevOps agent. Concise replies (<1000 chars). Explain before acting.",
+    model_kwargs={"temperature": 0.2, "max_tokens": 8192, "top_p": 0.95}
+)
+
+@app.get("/health")
+async def health(): return {"status": "online"}
+
+@app.post("/webhook")
+async def wa_webhook(request: Request):
+    data = await request.json()
+    user_msg = data.get("message", "")
+    if user_msg.lower().startswith("claw "):
+        prompt = user_msg[5:].strip()
+        response = await agent.arun(prompt)
+        return {"replies": [{"message": response}]}
+    return {"replies": []}
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)