Update app.py

app.py

@@ -2,58 +2,88 @@ from fastapi import FastAPI, HTTPException
 import requests
 import re
 import os
+import random
 from urllib.parse import urlparse
 
 app = FastAPI()
 
-# Use a session to persist cookies automatically during the extraction phase
-session = requests.Session()
-session.headers.update({"User-Agent": "Mozilla/5.0"})
+# --- 1. RANDOMIZATION TOOLS ---
+
+# A pool of realistic User-Agents to rotate through
+USER_AGENTS = [
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36",
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:120.0) Gecko/20100101 Firefox/120.0",
+    "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36",
+    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:109.0) Gecko/20100101 Firefox/118.0"
+]
+
+def get_random_ip():
+    """Generates a random IP address to bypass simple IP counters."""
+    return f"{random.randint(1, 200)}.{random.randint(1, 255)}.{random.randint(1, 255)}.{random.randint(1, 255)}"
+
+# ------------------------------
 
 def get_buzz_info(buzz_url):
+    session = requests.Session()
+    
+    # Generate a fresh identity for this request
+    current_ua = random.choice(USER_AGENTS)
+    fake_ip = get_random_ip()
+    
+    # Headers that mimic a unique user coming from a unique IP
+    headers = {
+        "User-Agent": current_ua,
+        "X-Forwarded-For": fake_ip,
+        "X-Real-IP": fake_ip,
+        "Referer": "https://buzzheavier.com/"
+    }
+    
+    session.headers.update(headers)
+
     try:
-        # 1. Fetch page to extract filename
-        # We use the session object so cookies are stored
-        r = session.get(buzz_url, timeout=10)
+        # 1. Fetch page (View Count)
+        # We add a random query param to prevent caching on their side: ?cachebuster=123
+        url_with_prevention = f"{buzz_url}?cb={random.randint(1, 100000)}"
+        
+        r = session.get(url_with_prevention, timeout=10)
         r.raise_for_status()
         
-        # Regex for filename
         name = re.search(r'<span class="text-2xl">([^<]+)</span>', r.text)
         if name:
             filename = name.group(1).strip()
         else:
             filename = os.path.basename(urlparse(buzz_url).path) or "buzzheavier_file"
 
-        # 2. Get flashbang URL (The direct download link)
+        # 2. Trigger Download (Download Count)
         dl_url = buzz_url.rstrip("/") + "/download"
         
-        # Add required headers for the HTMX request
-        h = {
+        # Headers specifically for the download trigger
+        dl_headers = {
             "HX-Request": "true", 
-            "Referer": buzz_url
+            "Referer": buzz_url,
+            "X-Forwarded-For": fake_ip # Keep the IP consistent for this session
         }
         
-        # Perform the request to get the download link
-        r2 = session.get(dl_url, headers=h, timeout=10, allow_redirects=False)
+        r2 = session.get(dl_url, headers=dl_headers, timeout=10, allow_redirects=False)
         
         link = r2.headers.get("hx-redirect")
         
         if not link:
-            # Fallback if status is 200 but no redirect header
             if r2.status_code == 200:
                 print("Warning: No redirect, but got 200 OK.")
             else:
                 raise ValueError(f"No hx-redirect header found. Status: {r2.status_code}")
 
-        # 3. Extract Cookies to pass to the Worker
-        # Convert cookie jar to a standard "Cookie: name=value;" string
+        # 3. Extract Cookies
         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
+            "cookies": cookie_string,
+            "user_agent": current_ua # Pass this to Worker so it matches!
         }
 
     except Exception as e:
@@ -62,13 +92,10 @@ def get_buzz_info(buzz_url):
 
 @app.get("/")
 def home():
-    return {"status": "Running", "msg": "Send requests to /resolve?url=..."}
+    return {"status": "Running"}
 
 @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")