update

app/server.py

@@ -4,7 +4,7 @@ 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, screenshot
+from helium import start_chrome, kill_browser, get_driver
 import uuid
 import logging
 import pathlib
@@ -73,7 +73,9 @@ async def take_screenshot(url: str = Form(...)):
         
         try:
             logger.info(f"Taking screenshot at {filepath}")
-            screenshot(filepath)
+            # Utiliser get_driver et save_screenshot pour prendre la capture d'écran
+            selenium_driver = get_driver()
+            selenium_driver.save_screenshot(filepath)
             logger.info(f"Screenshot saved to {filepath}")
         finally:
             # Always kill the browser to free resources