Auto-commit: app.py updated

app.py

@@ -532,7 +532,187 @@ async def get_preview(preview_id: str):
 @app.get("/health")
 async def health():
     """Health check endpoint"""
-    return {"status": "healthy", "sandbox": "landrun", "security": "kernel-level"}
+    return {"status": "healthy", "sandbox": "landrun", "security": "kernel-level", "browser": "playwright-chromium"}
+
+
+@app.post("/browser/test")
+async def test_browser_automation(request: Request):
+    """
+    Test browser automation on executed code preview
+    AI agents can use this to automatically test UIs
+    """
+    data = await request.json()
+    preview_url = data.get("preview_url", "")  # e.g., "/preview/uuid"
+    test_actions = data.get("actions", [])  # List of actions to perform
+    
+    if not preview_url:
+        return JSONResponse({"error": "No preview_url provided"})
+    
+    # Build full URL
+    full_url = f"http://localhost:7860{preview_url}"
+    
+    try:
+        async with async_playwright() as p:
+            browser = await p.chromium.launch(headless=True)
+            page = await browser.new_page()
+            
+            # Navigate to preview
+            await page.goto(full_url, wait_until="networkidle", timeout=10000)
+            
+            # Take initial screenshot
+            screenshot_initial = await page.screenshot()
+            
+            # Perform test actions
+            test_results = []
+            for action in test_actions:
+                action_type = action.get("type")
+                selector = action.get("selector")
+                value = action.get("value")
+                
+                try:
+                    if action_type == "click":
+                        await page.click(selector, timeout=5000)
+                        test_results.append({"action": "click", "selector": selector, "status": "success"})
+                    
+                    elif action_type == "type":
+                        await page.fill(selector, value, timeout=5000)
+                        test_results.append({"action": "type", "selector": selector, "status": "success"})
+                    
+                    elif action_type == "wait":
+                        await page.wait_for_selector(selector, timeout=5000)
+                        test_results.append({"action": "wait", "selector": selector, "status": "success"})
+                    
+                    elif action_type == "screenshot":
+                        screenshot = await page.screenshot()
+                        test_results.append({
+                            "action": "screenshot",
+                            "status": "success",
+                            "data": base64.b64encode(screenshot).decode()
+                        })
+                    
+                    elif action_type == "get_text":
+                        text = await page.inner_text(selector, timeout=5000)
+                        test_results.append({
+                            "action": "get_text",
+                            "selector": selector,
+                            "status": "success",
+                            "text": text
+                        })
+                    
+                except Exception as e:
+                    test_results.append({
+                        "action": action_type,
+                        "selector": selector,
+                        "status": "error",
+                        "error": str(e)
+                    })
+            
+            # Take final screenshot
+            screenshot_final = await page.screenshot()
+            
+            await browser.close()
+            
+            return JSONResponse({
+                "status": "success",
+                "url_tested": full_url,
+                "test_results": test_results,
+                "screenshot_initial": base64.b64encode(screenshot_initial).decode(),
+                "screenshot_final": base64.b64encode(screenshot_final).decode()
+            })
+            
+    except Exception as e:
+        return JSONResponse({
+            "status": "error",
+            "error": str(e),
+            "url_tested": full_url
+        })
+
+
+@app.post("/browser/execute_and_test")
+async def execute_and_test(request: Request):
+    """
+    Execute code AND automatically test it with browser automation
+    One-shot API for AI agents: execute → preview → test
+    """
+    data = await request.json()
+    language = data.get("language", "react")
+    code = data.get("code", "")
+    test_actions = data.get("actions", [])
+    
+    if not code:
+        return JSONResponse({"error": "No code provided"})
+    
+    # Step 1: Execute code
+    exec_result = execute_with_landrun(language, code)
+    
+    if exec_result.get("error"):
+        return JSONResponse({
+            "status": "execution_failed",
+            "execution": exec_result
+        })
+    
+    preview_url = exec_result.get("preview_url")
+    if not preview_url:
+        return JSONResponse({
+            "status": "no_preview",
+            "execution": exec_result,
+            "message": "Code executed but no preview available"
+        })
+    
+    # Step 2: Test with browser automation
+    full_url = f"http://localhost:7860{preview_url}"
+    
+    try:
+        async with async_playwright() as p:
+            browser = await p.chromium.launch(headless=True)
+            page = await browser.new_page()
+            
+            await page.goto(full_url, wait_until="networkidle", timeout=10000)
+            
+            # Perform automated tests
+            test_results = []
+            for action in test_actions:
+                action_type = action.get("type")
+                selector = action.get("selector")
+                value = action.get("value")
+                
+                try:
+                    if action_type == "click":
+                        await page.click(selector, timeout=5000)
+                        test_results.append({"action": "click", "selector": selector, "status": "success"})
+                    elif action_type == "type":
+                        await page.fill(selector, value, timeout=5000)
+                        test_results.append({"action": "type", "selector": selector, "status": "success"})
+                    elif action_type == "get_text":
+                        text = await page.inner_text(selector, timeout=5000)
+                        test_results.append({"action": "get_text", "selector": selector, "status": "success", "text": text})
+                    elif action_type == "screenshot":
+                        screenshot = await page.screenshot()
+                        test_results.append({"action": "screenshot", "status": "success", "data": base64.b64encode(screenshot).decode()})
+                except Exception as e:
+                    test_results.append({"action": action_type, "selector": selector, "status": "error", "error": str(e)})
+            
+            # Final screenshot
+            final_screenshot = await page.screenshot()
+            
+            await browser.close()
+            
+            return JSONResponse({
+                "status": "success",
+                "execution": exec_result,
+                "browser_tests": {
+                    "url_tested": full_url,
+                    "test_results": test_results,
+                    "screenshot": base64.b64encode(final_screenshot).decode()
+                }
+            })
+            
+    except Exception as e:
+        return JSONResponse({
+            "status": "browser_error",
+            "execution": exec_result,
+            "browser_error": str(e)
+        })
 
 
 if __name__ == "__main__":