Update app.py

app.py

@@ -1,10 +1,7 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
-from seleniumbase import Driver
-import threading
-import time
-import queue
-import json
+import threading, queue, time, json
+from seleniumbase import SB
 
 app = FastAPI()
 
@@ -12,30 +9,31 @@ class URLRequest(BaseModel):
     url: str
 
 url_queue = queue.Queue()
+sb_ref = {"driver": None}  # เก็บไว้ใช้ภายหลัง
 
-driver = Driver(uc=True, headless=False, incognito=True)
-driver.get("https://www.google.com")
+def browser_worker():
+    with SB(uc=True, headed=True) as sb:
+        sb_ref["driver"] = sb
+        sb.get("https://www.google.com")  # โหลด google เพื่อได้ cookie
+        while True:
+            url = url_queue.get()
+            try:
+                print(f"[open] {url}")
+                sb.get(url)
+                time.sleep(0.5)
+            except Exception as e:
+                print("[error]", e)
+            url_queue.task_done()
 
 @app.get("/cookies")
 def get_cookies():
-    cookies = driver.get_cookies()
-    return cookies
-
-def browser_worker():
-    while True:
-        url = url_queue.get()
-        try:
-            print(f"[open] {url}")
-            driver.get(url)
-            time.sleep(0.5)
-        except Exception as e:
-            print("[error]", e)
-        url_queue.task_done()
-
-threading.Thread(target=browser_worker, daemon=True).start()
+    sb = sb_ref.get("driver")
+    return sb.get_cookies() if sb else {}
 
 @app.post("/open_url")
 async def open_url(request: URLRequest):
     print(f"[queue] {request.url}")
     url_queue.put(request.url)
     return {"status": "queued", "url": request.url}
+
+threading.Thread(target=browser_worker, daemon=True).start()