Merge pull request #11 from kirill-markin/feature/improve-local-page-handling

Add local page handling to LocalPlaywrightComputer

Author: katia-openai

computers/local_playwright.py

@@ -18,7 +18,32 @@ def _get_browser_and_page(self) -> tuple[Browser, Page]:
             args=launch_args,
             env={"DISPLAY": ":0"}
         )
-        page = browser.new_page()
+        
+        context = browser.new_context()
+        
+        # Add event listeners for page creation and closure
+        context.on("page", self._handle_new_page)
+        
+        page = context.new_page()
         page.set_viewport_size({"width": width, "height": height})
+        page.on("close", self._handle_page_close)
+
         page.goto("https://bing.com")
+        
         return browser, page
+        
+    def _handle_new_page(self, page: Page):
+        """Handle the creation of a new page."""
+        print("New page created")
+        self._page = page
+        page.on("close", self._handle_page_close)
+        
+    def _handle_page_close(self, page: Page):
+        """Handle the closure of a page."""
+        print("Page closed")
+        if self._page == page:
+            if self._browser.contexts[0].pages:
+                self._page = self._browser.contexts[0].pages[-1]
+            else:
+                print("Warning: All pages have been closed.")
+                self._page = None