Update app.py

app.py

@@ -30,27 +30,6 @@ if not gemini_api_key:
 
 login(hf_token, add_to_git_credential=False)
 
-# Patch Helium to use system-installed ChromeDriver
-from selenium.webdriver.chrome.options import Options as ChromeOptions
-from selenium.webdriver.chrome.webdriver import WebDriver as Chrome
-from helium._impl import _APIImpl
-
-def _start_chrome_driver(self, headless, options):
-    chrome_options = options or ChromeOptions()
-    if headless:
-        chrome_options.add_argument('--headless')
-    # Do not set executable_path; let Selenium use the environment variable
-    result = Chrome(options=chrome_options)
-    return result
-
-# Save the original method
-original_start_chrome_driver = _APIImpl._start_chrome_driver
-# Override it with our custom function
-_APIImpl._start_chrome_driver = _start_chrome_driver
-
-# Set the environment variable for Selenium to use
-os.environ['webdriver.chrome.driver'] = '/usr/local/bin/chromedriver'
-
 # Define tools
 @tool
 def search_item_ctrl_f(text: str, nth_result: int = 1) -> str:
@@ -82,7 +61,7 @@ def close_popups() -> str:
     """
     webdriver.ActionChains(driver).send_keys(Keys.ESCAPE).perform()
 
-# Initialize Chrome driver with error handling
+# Initialize Chrome driver with Selenium directly
 try:
     chrome_options = webdriver.ChromeOptions()
     chrome_options.add_argument("--force-device-scale-factor=1")
@@ -92,7 +71,11 @@ try:
     chrome_options.add_argument("--disable-dev-shm-usage")
     chrome_options.add_argument("--window-position=0,0")
     chrome_options.add_argument("--headless=new")
-    driver = helium.start_chrome(headless=True, options=chrome_options)
+    driver = webdriver.Chrome(
+        executable_path='/usr/local/bin/chromedriver',
+        options=chrome_options
+    )
+    helium.set_driver(driver)
     logger.info("Chrome driver initialized successfully.")
 except Exception as e:
     logger.error(f"Failed to initialize Chrome driver: {str(e)}")