Update app.py

app.py

@@ -1,34 +1,21 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from pydantic import BaseModel
-from seleniumbase import Driver
-import threading
-import time
+from typing import List
 import queue
 
 app = FastAPI()
+url_queue = queue.Queue()
 
 class URLRequest(BaseModel):
     url: str
 
-url_queue = queue.Queue()
-
-driver = Driver(uc=True, headless=True, incognito=True, agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 Chrome/85.0.4183.83 Safari/537.36")
-
-def browser_worker():
-    while True:
-        url = url_queue.get()
-        try:
-            print(f"[open] {url}")
-            driver.open(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.post("/submit")
+async def submit_url(req: URLRequest):
+    url_queue.put(req.url)
+    return {"status": "queued", "url": req.url}
 
-@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}
+@app.get("/next")
+async def get_next_url():
+    if url_queue.empty():
+        return {"status": "empty"}
+    return {"url": url_queue.get()}