Update app.py

app.py

@@ -30,6 +30,27 @@ 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:
@@ -61,9 +82,6 @@ def close_popups() -> str:
     """
     webdriver.ActionChains(driver).send_keys(Keys.ESCAPE).perform()
 
-# Set ChromeDriver path
-os.environ['webdriver.chrome.driver'] = '/usr/local/bin/chromedriver'
-
 # Initialize Chrome driver with error handling
 try:
     chrome_options = webdriver.ChromeOptions()