Update app.py

app.py

@@ -1,21 +1,41 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
 from pydantic import BaseModel
-from typing import List
+from seleniumbase import Driver
+import threading
+import time
 import queue
+import json
 
 app = FastAPI()
-url_queue = queue.Queue()
 
 class URLRequest(BaseModel):
     url: str
 
-@app.post("/submit")
-async def submit_url(req: URLRequest):
-    url_queue.put(req.url)
-    return {"status": "queued", "url": req.url}
+url_queue = queue.Queue()
+
+driver = Driver(uc=True, headless=False, incognito=True)
+driver.get("https://www.google.com")
+
+@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()
 
-@app.get("/next")
-async def get_next_url():
-    if url_queue.empty():
-        return {"status": "empty"}
-    return {"url": url_queue.get()}
+@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}