Update files

app.py

@@ -3,6 +3,7 @@ import os
 import time
 import tempfile
 import logging
+import subprocess
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.by import By
@@ -34,6 +35,22 @@ logger = logging.getLogger(__name__)
 os.environ['MPLCONFIGDIR'] = tempfile.mkdtemp()
 logger.info(f"Matplotlib config directory set to: {os.environ['MPLCONFIGDIR']}")
 
+def verify_firefox_installation():
+    if not os.path.exists('/usr/bin/firefox-esr'):
+        logger.error("Firefox binary not found at /usr/bin/firefox-esr")
+        return False
+    return True
+
+def verify_geckodriver():
+    try:
+        result = subprocess.run(['/app/geckodriver', '--version'], 
+                              capture_output=True, text=True)
+        logger.info(f"Geckodriver version: {result.stdout}")
+        return True
+    except:
+        logger.error("Geckodriver verification failed")
+        return False
+
 def setup_selenium():
     try:
         logger.info("Initializing Selenium Firefox driver...")
@@ -65,6 +82,7 @@ def setup_selenium():
         # Configure service with log_path
         service = Service(
             executable_path=geckodriver_path,
+            service_args=['--marionette-port', '2828'],
             log_path=os.path.join(tempfile.gettempdir(), 'geckodriver.log')
         )
 
@@ -73,10 +91,27 @@ def setup_selenium():
             service=service
             # executable_path=os.path.join(os.getcwd(), 'geckodriver')
         )
+        logger.info("Selenium driver initialized successfully")
         return driver
     except Exception as e:
-        logger.error(f"Failed to initialize Selenium: {str(e)}")
-        raise
+        logger.error("Possible causes:")
+        logger.error("- Missing or incompatible geckodriver")
+        logger.error("- Firefox binary not found")
+        logger.error("- Insufficient permissions")
+        logger.error("- Missing system dependencies")
+        raise RuntimeError(f"Failed to initialize WebDriver: {str(e)}")
+
+def check_selenium_environment():
+    try:
+        driver = setup_selenium()
+        driver.quit()
+        return True
+    except:
+        return False
+
+if not check_selenium_environment():
+    logger.critical("Selenium environment check failed!")
+    exit(1)
 
 def form_input_text(word):
     # word = 'go'
@@ -103,6 +138,8 @@ def fetch_sentences(input_text):
         # Open a website
         driver.get("https://copilot.microsoft.com/chat")
         print(f"...after: driver.get('https://copilot.microsoft.com/chat')")
+
+        logger.info("Waiting for page to load...")
         # Wait for an element to load
         WebDriverWait(driver, 50).until(EC.presence_of_element_located((By.TAG_NAME, "body")))
         time.sleep(10)  # Keeps the browser open for 10 seconds
@@ -190,9 +227,18 @@ def scrape_website(url):
     # return fetch_sentences(form_input_text(current_word))
 
     try:
+        if not verify_firefox_installation():
+            return "Error: Firefox not properly installed"
+            
+        current_word = 'go'
         return fetch_sentences(form_input_text(current_word))
+        
+    except RuntimeError as e:
+        logger.error(f"Runtime error: {str(e)}")
+        return f"System error: {str(e)}"
     except Exception as e:
-        return f"Error in scrape_website: {str(e)}"
+        logger.error(f"Unexpected error: {str(e)}")
+        return f"Unexpected error occurred: {str(e)}"
 
 
     # driver = None