Upload folder using huggingface_hub

src/browser/custom_browser.py

@@ -12,7 +12,7 @@ from playwright.async_api import (
     async_playwright,
 )
 
-# Import browser_use components with error handling
+# Import browser_use components with comprehensive error handling
 try:
     from browser_use.browser.browser import Browser, IN_DOCKER
     from browser_use.browser.context import BrowserContext, BrowserContextConfig
@@ -29,19 +29,51 @@ except ImportError as e:
     # Fallback for different browser_use versions
     logger = logging.getLogger(__name__)
     logger.warning(f"Failed to import browser_use components: {e}")
+    
     # Set default values
     IN_DOCKER = False
-    from browser_use.browser.browser import Browser
-    from browser_use.browser.context import BrowserContext, BrowserContextConfig
-    from browser_use.browser.chrome import (
-        CHROME_ARGS,
-        CHROME_DETERMINISTIC_RENDERING_ARGS,
-        CHROME_DISABLE_SECURITY_ARGS,
-        CHROME_DOCKER_ARGS,
-        CHROME_HEADLESS_ARGS,
-    )
-    from browser_use.browser.utils.screen_resolution import get_screen_resolution, get_window_adjustments
-    from browser_use.utils import time_execution_async
+    
+    # Try to import basic components
+    try:
+        from browser_use.browser.browser import Browser
+        from browser_use.browser.context import BrowserContext, BrowserContextConfig
+    except ImportError:
+        logger.error("Could not import basic browser_use components")
+        raise
+    
+    # Try to import chrome components with fallbacks
+    try:
+        from browser_use.browser.chrome import (
+            CHROME_ARGS,
+            CHROME_DETERMINISTIC_RENDERING_ARGS,
+            CHROME_DISABLE_SECURITY_ARGS,
+            CHROME_DOCKER_ARGS,
+            CHROME_HEADLESS_ARGS,
+        )
+    except ImportError:
+        logger.warning("Could not import chrome components, using defaults")
+        CHROME_ARGS = []
+        CHROME_DETERMINISTIC_RENDERING_ARGS = []
+        CHROME_DISABLE_SECURITY_ARGS = []
+        CHROME_DOCKER_ARGS = []
+        CHROME_HEADLESS_ARGS = []
+    
+    # Try to import utility functions with fallbacks
+    try:
+        from browser_use.browser.utils.screen_resolution import get_screen_resolution, get_window_adjustments
+    except ImportError:
+        logger.warning("Could not import screen resolution utilities, using defaults")
+        def get_screen_resolution():
+            return {'width': 1920, 'height': 1080}
+        def get_window_adjustments():
+            return 0, 0
+    
+    try:
+        from browser_use.utils import time_execution_async
+    except ImportError:
+        logger.warning("Could not import time_execution_async, using dummy function")
+        def time_execution_async(func):
+            return func
 
 from .custom_context import CustomBrowserContext
 

src/browser/custom_context.py

@@ -2,7 +2,7 @@ import json
 import logging
 import os
 
-# Import browser_use components with error handling
+# Import browser_use components with comprehensive error handling
 try:
     from browser_use.browser.browser import Browser, IN_DOCKER
     from browser_use.browser.context import BrowserContext, BrowserContextConfig, BrowserContextState
@@ -10,10 +10,17 @@ except ImportError as e:
     # Fallback for different browser_use versions
     logger = logging.getLogger(__name__)
     logger.warning(f"Failed to import browser_use components: {e}")
+    
     # Set default values
     IN_DOCKER = False
-    from browser_use.browser.browser import Browser
-    from browser_use.browser.context import BrowserContext, BrowserContextConfig, BrowserContextState
+    
+    # Try to import basic components
+    try:
+        from browser_use.browser.browser import Browser
+        from browser_use.browser.context import BrowserContext, BrowserContextConfig, BrowserContextState
+    except ImportError:
+        logger.error("Could not import basic browser_use components")
+        raise
 
 from playwright.async_api import Browser as PlaywrightBrowser
 from playwright.async_api import BrowserContext as PlaywrightBrowserContext