Create browser_controller.py

browser_controller.py

@@ -3,7 +3,6 @@ from playwright.sync_api import sync_playwright
 class BrowserController:
     def __init__(self):
         self.playwright = sync_playwright().start()
-        # Headless=True for Hugging Face Spaces compatibility
         self.browser = self.playwright.chromium.launch(headless=True)
         self.page = self.browser.new_page()
 
@@ -24,12 +23,9 @@ class BrowserController:
         return self.get_state()
 
     def get_state(self):
-        # Screenshot as hex string, HTML as string
-        screenshot_bytes = self.page.screenshot()
-        html_content = self.page.content()
         return {
-            "screenshot": screenshot_bytes.hex(),
-            "html": html_content
+            "screenshot": self.page.screenshot().hex(),
+            "html": self.page.content()
         }
 
     def close(self):