update

app/server.py

@@ -5,6 +5,7 @@ from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 from helium import start_chrome, kill_browser, get_driver
+from selenium import webdriver
 import uuid
 import logging
 import pathlib
@@ -50,24 +51,23 @@ async def take_screenshot(url: str = Form(...)):
         logger.info(f"HOME env: {os.environ.get('HOME')}")
         logger.info(f"Current working directory: {os.getcwd()}")
         
-        # Configuration for Docker environment
-        chrome_options = {
-            "headless": True,
-            "options": [
-                '--no-sandbox',
-                '--disable-setuid-sandbox',
-                '--disable-dev-shm-usage',
-                '--disable-accelerated-2d-canvas',
-                '--no-first-run',
-                '--no-zygote',
-                '--single-process',
-                '--disable-gpu'
-            ]
-        }
+        # Configuration pour Chrome dans Helium
+        # Créer les options Chrome à la manière Selenium
+        chrome_options = webdriver.ChromeOptions()
+        chrome_options.add_argument('--headless')
+        chrome_options.add_argument('--no-sandbox')
+        chrome_options.add_argument('--disable-setuid-sandbox')
+        chrome_options.add_argument('--disable-dev-shm-usage')
+        chrome_options.add_argument('--disable-accelerated-2d-canvas')
+        chrome_options.add_argument('--no-first-run')
+        chrome_options.add_argument('--no-zygote')
+        chrome_options.add_argument('--single-process')
+        chrome_options.add_argument('--disable-gpu')
         
         # Take the screenshot with Helium
         logger.info("Launching Chrome browser")
-        driver = start_chrome(url, **chrome_options)
+        # Utiliser les options Chrome configurées pour Selenium
+        driver = start_chrome(url, headless=True, options=chrome_options)
         
         logger.info("Browser launched successfully")