Update src/controller/custom_controller.py

src/controller/custom_controller.py

@@ -1,27 +1,31 @@
-import pyperclip
-from browser_use.agent.views import ActionResult
-from browser_use.browser.context import BrowserContext
-from browser_use.controller.service import Controller
-
-
-class CustomController(Controller):
-    def __init__(self):
-        super().__init__()
-        self._register_custom_actions()
-
-    def _register_custom_actions(self):
-        """Register all custom browser actions"""
-
-        @self.registry.action("Copy text to clipboard")
-        def copy_to_clipboard(text: str):
-            pyperclip.copy(text)
-            return ActionResult(extracted_content=text)
-
-        @self.registry.action("Paste text from clipboard", requires_browser=True)
-        async def paste_from_clipboard(browser: BrowserContext):
-            text = pyperclip.paste()
-            # send text to browser
-            page = await browser.get_current_page()
-            await page.keyboard.type(text)
-
-            return ActionResult(extracted_content=text)
+import pyperclip
+from browser_use.agent.views import ActionResult
+from browser_use.browser.context import BrowserContext
+from browser_use.controller.service import Controller
+
+class CustomController(Controller):
+    def __init__(self):
+        super().__init__()
+        self._register_custom_actions()
+
+    def _register_custom_actions(self):
+        @self.registry.action("Copy text to clipboard")
+        def copy_to_clipboard(text: str):
+            try:
+                pyperclip.copy(text)
+                return ActionResult(extracted_content=text)
+            except Exception as e:
+                return ActionResult(error=f"Failed to copy text to clipboard: {str(e)}")
+
+        @self.registry.action("Paste text from clipboard", requires_browser=True)
+        async def paste_from_clipboard(browser: BrowserContext):
+            try:
+                text = pyperclip.paste()
+                page = await browser.get_current_page()
+                if not page:
+                    return ActionResult(error="No active page found in browser.")
+
+                await page.keyboard.type(text)
+                return ActionResult(extracted_content=text)
+            except Exception as e:
+                return ActionResult(error=f"Failed to paste text from clipboard: {str(e)}")