Working computer use implementation!

chriscarrollsmith · chriscarrollsmith · commit 23a6e969106e · 2026-03-14T11:16:54.000-04:00
diff --git a/tests/test_computer_use.py b/tests/test_computer_use.py
@@ -217,6 +217,44 @@ async def test_execute_drag(self, session: BrowserSession):
         result = await session.execute(action)
         assert len(result) > 0
 
+    @pytest.mark.anyio
+    async def test_landing_page_has_url_input(self, session: BrowserSession):
+        page = await session._ensure_page()
+        url_input = await page.query_selector("#url")
+        assert url_input is not None
+
+    @pytest.mark.anyio
+    async def test_landing_page_has_go_button(self, session: BrowserSession):
+        page = await session._ensure_page()
+        button = await page.query_selector("button[type=submit]")
+        assert button is not None
+        text = await button.text_content()
+        assert text is not None
+        assert "Go" in text
+
+    @pytest.mark.anyio
+    async def test_landing_page_navigation(self, session: BrowserSession):
+        page = await session._ensure_page()
+        # Intercept the navigation request and serve a fake page
+        await page.route("https://example.com/", lambda route: route.fulfill(
+            status=200,
+            content_type="text/html",
+            body="<h1>Hello</h1>",
+        ))
+        await page.fill("#url", "https://example.com/")
+        await page.click("button[type=submit]")
+        await page.wait_for_load_state("load")
+        content = await page.content()
+        assert "Hello" in content
+
+    @pytest.mark.anyio
+    async def test_landing_page_not_blank(self, session: BrowserSession):
+        result = await session.screenshot()
+        img = Image.open(io.BytesIO(base64.b64decode(result)))
+        colors = img.getcolors(maxcolors=10)
+        # A blank white page would have exactly 1 color; landing page has more
+        assert colors is None or len(colors) > 1
+
     @pytest.mark.anyio
     async def test_close_is_idempotent(self, session: BrowserSession):
         await session.close()
diff --git a/utils/computer_use.py b/utils/computer_use.py
@@ -19,6 +19,32 @@
 
 logger = logging.getLogger("uvicorn.error")
 
+_LANDING_PAGE_HTML = """\
+<!DOCTYPE html>
+<html>
+<head><title>Browser</title></head>
+<body style="margin:0;display:flex;justify-content:center;align-items:center;
+             height:100vh;background:#f0f0f0;font-family:sans-serif;">
+  <div style="text-align:center;">
+    <h1 style="font-size:28px;margin-bottom:24px;">Browser &mdash; Enter a URL to get started</h1>
+    <form onsubmit="window.location=document.getElementById('url').value;return false;"
+          style="display:flex;gap:8px;justify-content:center;">
+      <input id="url" type="text"
+             placeholder="Enter a URL, e.g. https://example.com"
+             style="width:500px;padding:12px 16px;font-size:18px;border:2px solid #ccc;
+                    border-radius:6px;"
+             autofocus />
+      <button type="submit"
+              style="padding:12px 24px;font-size:18px;background:#2563eb;color:white;
+                     border:none;border-radius:6px;cursor:pointer;">
+        Go
+      </button>
+    </form>
+  </div>
+</body>
+</html>
+"""
+
 Action = Union[
     Click, DoubleClick, Drag, Keypress,
     Move, Screenshot, Scroll, Type, Wait,
@@ -100,7 +126,7 @@ async def _ensure_page(self) -> Page:
             device_scale_factor=1,
         )
         self._page = await context.new_page()
-        await self._page.goto("about:blank")
+        await self._page.set_content(_LANDING_PAGE_HTML, wait_until="load")
         return self._page
 
     async def screenshot(self) -> str:
@@ -157,11 +183,17 @@ async def execute(self, action: Action) -> str:
     async def close(self) -> None:
         """Close the browser and clean up resources."""
         if self._browser:
-            await self._browser.close()
+            try:
+                await self._browser.close()
+            except Exception:
+                logger.debug("Browser already disconnected during close")
             self._browser = None
             self._page = None
         if self._playwright:
-            await self._playwright.stop()
+            try:
+                await self._playwright.stop()
+            except Exception:
+                logger.debug("Playwright already stopped during close")
             self._playwright = None
 
 
