Update app.py

app.py

@@ -1,11 +1,14 @@
 import os, asyncio
 from fastapi import FastAPI, HTTPException, Header, Response
 from pydantic import BaseModel, HttpUrl
-from seleniumbase import Driver
+from selenium import webdriver
+from selenium.webdriver.chrome.service import Service
+from selenium.webdriver.chrome.options import Options
 
 API_KEY = os.getenv("OPENER_API_KEY", "")
-PROFILE = os.getenv("SB_PROFILE_DIR", "/tmp/sb-profile")
+PROFILE = os.getenv("CHROME_PROFILE_DIR", "/tmp/chrome-profile")
 UA = os.getenv("UA", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36")
+CHROMEDRIVER = os.getenv("CHROMEDRIVER", "/usr/bin/chromedriver")  # debian usually symlinks here
 
 app = FastAPI()
 drv = None
@@ -15,33 +18,31 @@ worker = None
 class OpenPayload(BaseModel):
     url: HttpUrl
 
+def _build_driver():
+    os.makedirs(PROFILE, exist_ok=True)
+    opts = Options()
+    opts.add_argument("--headless=new")
+    opts.add_argument("--no-sandbox")
+    opts.add_argument("--disable-gpu")
+    opts.add_argument("--disable-dev-shm-usage")
+    opts.add_argument(f"--user-agent={UA}")
+    opts.add_argument(f"--user-data-dir={PROFILE}")
+    svc = Service(executable_path=CHROMEDRIVER)
+    return webdriver.Chrome(service=svc, options=opts)
+
 async def _worker():
     while True:
         url = await queue.get()
         try:
             await asyncio.to_thread(drv.get, url)
             print("[open] done", url)
-        except Exception as e:
-            print("[open] err", url, e)
         finally:
             queue.task_done()
 
 @app.on_event("startup")
 async def startup():
     global drv, queue, worker
-    os.chdir("/tmp")
-    os.makedirs("downloaded_files", exist_ok=True)
-    os.makedirs(PROFILE, exist_ok=True)
-    drv = Driver(
-        uc=True,
-        headless=True,
-        incognito=False,
-        user_data_dir=PROFILE,
-        page_load_strategy="normal",
-        agent=UA,
-        no_sandbox=True,
-        disable_gpu=True,
-    )
+    drv = _build_driver()
     drv.get("data:text/html,ready")
     queue = asyncio.Queue()
     worker = asyncio.create_task(_worker())