Update VisionaryAgent.py

VisionaryAgent.py

@@ -13,6 +13,17 @@ import helium
 from typing import List
 from pydantic import BaseModel, Field
 from fastapi import UploadFile
+from selenium import webdriver
+import helium
+
+# Configure Chrome options for Hugging Face Spaces
+chrome_options = webdriver.ChromeOptions()
+chrome_options.add_argument("--headless")  # Runs Chrome in headless mode
+chrome_options.add_argument("--no-sandbox")  # Required for running on HF Spaces
+chrome_options.add_argument("--disable-dev-shm-usage")  # Prevents memory errors
+chrome_options.add_argument("--disable-gpu")  # Disable GPU acceleration (not needed in headless mode)
+chrome_options.add_argument("--remote-debugging-port=9222")  # Helps with debugging if needed
+
 
 # Load environment variables (API keys, etc.)
 from dotenv import load_dotenv
@@ -56,8 +67,8 @@ def scrape_website(url: str):
     return firecrawl_agent.print_response(f"Extract all relevant business information from {url}, including mission statement, services, case studies, and AI-related content. Provide structured output.")
 
 # Helium for dynamic websites
-chrome_options = webdriver.ChromeOptions()
-chrome_options.add_argument("--headless")
+# chrome_options = webdriver.ChromeOptions()
+# chrome_options.add_argument("--headless")
 driver = helium.start_chrome(headless=True, options=chrome_options)
 
 def scrape_dynamic_website(url: str):