Spaces:
Sleeping
Sleeping
Update VisionaryAgent.py
Browse files- VisionaryAgent.py +11 -6
VisionaryAgent.py
CHANGED
|
@@ -16,18 +16,23 @@ from fastapi import UploadFile
|
|
| 16 |
from selenium import webdriver
|
| 17 |
import helium
|
| 18 |
|
| 19 |
-
# Configure Chrome options for Hugging Face Spaces
|
| 20 |
# Ensure the system uses the installed Chromium and Chromedriver
|
| 21 |
chrome_options = webdriver.ChromeOptions()
|
| 22 |
-
chrome_options.binary_location = "/usr/bin/chromium-browser" #
|
| 23 |
-
chrome_options.add_argument("--headless") # Run Chrome in headless mode
|
| 24 |
chrome_options.add_argument("--no-sandbox") # Required for Hugging Face Spaces
|
| 25 |
chrome_options.add_argument("--disable-dev-shm-usage") # Prevents memory issues
|
| 26 |
chrome_options.add_argument("--disable-gpu") # Disable GPU acceleration
|
| 27 |
-
chrome_options.add_argument("--remote-debugging-port=9222") # Helps
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
-
# Manually specify Chromedriver path
|
| 30 |
-
driver = webdriver.Chrome(executable_path="/usr/bin/chromedriver", options=chrome_options)
|
| 31 |
|
| 32 |
|
| 33 |
|
|
|
|
| 16 |
from selenium import webdriver
|
| 17 |
import helium
|
| 18 |
|
|
|
|
| 19 |
# Ensure the system uses the installed Chromium and Chromedriver
|
| 20 |
chrome_options = webdriver.ChromeOptions()
|
| 21 |
+
chrome_options.binary_location = "/usr/bin/chromium-browser" # Specify Chromium path
|
| 22 |
+
chrome_options.add_argument("--headless") # Run Chrome in headless mode
|
| 23 |
chrome_options.add_argument("--no-sandbox") # Required for Hugging Face Spaces
|
| 24 |
chrome_options.add_argument("--disable-dev-shm-usage") # Prevents memory issues
|
| 25 |
chrome_options.add_argument("--disable-gpu") # Disable GPU acceleration
|
| 26 |
+
chrome_options.add_argument("--remote-debugging-port=9222") # Helps debugging
|
| 27 |
+
|
| 28 |
+
# Correct Chromedriver setup using Selenium 4
|
| 29 |
+
chromedriver_path = "/usr/bin/chromedriver"
|
| 30 |
+
service = Service(chromedriver_path)
|
| 31 |
+
driver = webdriver.Chrome(service=service, options=chrome_options)
|
| 32 |
+
|
| 33 |
+
# Start Helium using the modified driver
|
| 34 |
+
helium.set_driver(driver)
|
| 35 |
|
|
|
|
|
|
|
| 36 |
|
| 37 |
|
| 38 |
|