Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image, ImageDraw
 import cv2
 import numpy as np
 from selenium import webdriver
+from selenium.webdriver.chrome.service import Service
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
@@ -26,10 +27,22 @@ ROBOFLOW_API_KEY = "KUP9w62eUcD5PrrRMJsV"  # Replace with your API key
 PROJECT_NAME = "model_verification_project"
 VERSION_NUMBER = 2
 
+# Paths to Chrome and ChromeDriver downloaded by Selenium Manager
+CHROME_PATH = "/home/user/.cache/selenium/chrome/linux64/132.0.6834.83/chrome"
+CHROMEDRIVER_PATH = "/home/user/.cache/selenium/chromedriver/linux64/132.0.6834.83/chromedriver"
+
 # Selenium configuration for Calligrapher
 def get_calligrapher():
     calli_url = "https://www.calligrapher.ai"
-    driver = webdriver.Chrome()
+
+    options = webdriver.ChromeOptions()
+    options.add_argument("--headless")  # Run Chrome in headless mode
+    options.add_argument("--no-sandbox")
+    options.add_argument("--disable-dev-shm-usage")
+    options.binary_location = CHROME_PATH
+
+    service = Service(CHROMEDRIVER_PATH)
+    driver = webdriver.Chrome(service=service, options=options)
     driver.maximize_window()
     driver.get(calli_url)
 
@@ -179,3 +192,4 @@ if __name__ == "__main__":
     logging.debug("Launching Gradio interface.")
     interface.launch(share=True)
 
+