Update app.py

app.py

@@ -6,12 +6,15 @@ from urllib.parse import urlparse
 
 app = FastAPI()
 
-UA = {"User-Agent": "Mozilla/5.0"}
+# Use a session to persist cookies automatically during the extraction phase
+session = requests.Session()
+session.headers.update({"User-Agent": "Mozilla/5.0"})
 
 def get_buzz_info(buzz_url):
     try:
         # 1. Fetch page to extract filename
-        r = requests.get(buzz_url, headers=UA, timeout=10)
+        # We use the session object so cookies are stored
+        r = session.get(buzz_url, timeout=10)
         r.raise_for_status()
         
         # Regex for filename
@@ -23,21 +26,35 @@ def get_buzz_info(buzz_url):
 
         # 2. Get flashbang URL (The direct download link)
         dl_url = buzz_url.rstrip("/") + "/download"
-        h = {"HX-Request": "true", "Referer": buzz_url, **UA}
         
-        r2 = requests.get(dl_url, headers=h, timeout=10, allow_redirects=False)
+        # Add required headers for the HTMX request
+        h = {
+            "HX-Request": "true", 
+            "Referer": buzz_url
+        }
+        
+        # Perform the request to get the download link
+        r2 = session.get(dl_url, headers=h, timeout=10, allow_redirects=False)
         
-        # Buzzheavier sends the link in the 'hx-redirect' header
         link = r2.headers.get("hx-redirect")
         
         if not link:
-            # Fallback: Sometimes it might just work without hx-redirect if the status is 200
+            # Fallback if status is 200 but no redirect header
             if r2.status_code == 200:
-                print("Warning: No redirect, but got 200 OK. URL might be direct.")
+                print("Warning: No redirect, but got 200 OK.")
             else:
                 raise ValueError(f"No hx-redirect header found. Status: {r2.status_code}")
-            
-        return {"filename": filename, "download_url": link}
+
+        # 3. Extract Cookies to pass to the Worker
+        # Convert cookie jar to a standard "Cookie: name=value;" string
+        cookie_dict = session.cookies.get_dict()
+        cookie_string = "; ".join([f"{k}={v}" for k, v in cookie_dict.items()])
+
+        return {
+            "filename": filename, 
+            "download_url": link,
+            "cookies": cookie_string  # <--- NEW: Sending cookies back
+        }
 
     except Exception as e:
         print(f"Error: {e}")
@@ -49,6 +66,9 @@ def home():
 
 @app.get("/resolve")
 def resolve_url(url: str):
+    # Clear cookies between requests to ensure a fresh session for every user
+    session.cookies.clear()
+    
     data = get_buzz_info(url)
     if not data:
         raise HTTPException(status_code=500, detail="Failed to resolve URL")