Update app.py

app.py

@@ -1,14 +1,12 @@
 import os, asyncio
 from fastapi import FastAPI, HTTPException, Header, Response
 from pydantic import BaseModel, HttpUrl
-from selenium import webdriver
-from selenium.webdriver.chrome.service import Service
-from selenium.webdriver.chrome.options import Options
+import undetected_chromedriver as uc
 
 API_KEY = os.getenv("OPENER_API_KEY", "")
-PROFILE = os.getenv("CHROME_PROFILE_DIR", "/tmp/chrome-profile")
+PROFILE = os.getenv("UC_PROFILE_DIR", "/tmp/uc-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")
+CHROME_BIN = os.getenv("CHROME_BIN", "/usr/bin/chromium")
 
 app = FastAPI()
 drv = None
@@ -18,27 +16,7 @@ worker = None
 class OpenPayload(BaseModel):
     url: HttpUrl
 
-def clean_profile(p):
-    os.makedirs(p, exist_ok=True)
-    for n in ("SingletonLock", "SingletonCookie", "SingletonSocket", "DevToolsActivePort"):
-        f = os.path.join(p, n)
-        if os.path.exists(f): os.remove(f)
-
-def build_driver():
-    clean_profile(PROFILE)
-    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("--no-first-run")
-    opts.add_argument("--no-default-browser-check")
-    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_fn():
+async def _worker():
     while True:
         url = await queue.get()
         try:
@@ -50,11 +28,19 @@ async def worker_fn():
 @app.on_event("startup")
 async def startup():
     global drv, queue, worker
-    os.environ["XDG_RUNTIME_DIR"] = "/tmp"
-    drv = build_driver()
+    os.makedirs(PROFILE, exist_ok=True)
+    o = uc.ChromeOptions()
+    o.headless = True
+    o.binary_location = CHROME_BIN
+    o.add_argument(f"--user-data-dir={PROFILE}")
+    o.add_argument("--no-sandbox")
+    o.add_argument("--disable-dev-shm-usage")
+    o.add_argument("--disable-gpu")
+    o.add_argument(f"--user-agent={UA}")
+    drv = uc.Chrome(options=o, use_subprocess=True)
     drv.get("data:text/html,ready")
     queue = asyncio.Queue()
-    worker = asyncio.create_task(worker_fn())
+    worker = asyncio.create_task(_worker())
     print("[startup] browser ready")
 
 @app.on_event("shutdown")