update

app/server.py

@@ -4,11 +4,11 @@ from fastapi import FastAPI, Request, Form
 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
+from helium import start_chrome, kill_browser, go_to
 import uuid
 import logging
 import pathlib
+import time
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -51,34 +51,25 @@ 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 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
+        # Lancer Chrome avec des options simples
         logger.info("Launching Chrome browser")
-        # Utiliser les options Chrome configurées pour Selenium
-        driver = start_chrome(url, headless=True, options=chrome_options)
-        
-        logger.info("Browser launched successfully")
+        driver = start_chrome(headless=True)
         
         try:
+            # Aller à l'URL
+            logger.info(f"Navigating to URL: {url}")
+            go_to(url)
+            
+            # Attendre que la page soit chargée
+            logger.info("Waiting for page to load")
+            time.sleep(3)
+            
+            # Prendre la capture d'écran
             logger.info(f"Taking screenshot at {filepath}")
-            # Utiliser get_driver et save_screenshot pour prendre la capture d'écran
-            selenium_driver = get_driver()
-            selenium_driver.save_screenshot(filepath)
+            driver.save_screenshot(filepath)
             logger.info(f"Screenshot saved to {filepath}")
         finally:
-            # Always kill the browser to free resources
+            # Toujours fermer le navigateur
             kill_browser()
             logger.info("Browser closed")