Update main.py

main.py

@@ -4,6 +4,7 @@ from fastapi.responses import StreamingResponse
 import json
 import random
 import logging
+import ipaddress
 
 # Configure logging to output to stdout
 logging.basicConfig(
@@ -20,26 +21,42 @@ API_URLS = [
     "https://stage.api.deepinfra.com/v1/openai/chat/completions",
 ]
 
+def generate_random_ip() -> str:
+    """Generate a random IPv4 address, avoiding reserved ranges."""
+    while True:
+        ip = ipaddress.IPv4Address(random.getrandbits(32))
+        if not (ip.is_private or ip.is_multicast or ip.is_reserved or ip.is_loopback):
+            return str(ip)
+
 @app.post("/v1/openai/chat/completions")
 async def proxy_deepinfra(request: Request):
     """
     Proxies chat completion requests to the DeepInfra API.
     It randomizes the order of API URLs and uses the next as a fallback.
+    Adds spoofed random IP headers.
     """
     try:
         body = await request.json()
     except json.JSONDecodeError:
         raise HTTPException(status_code=400, detail="Invalid JSON in request body")
 
+    # Generate a random spoofed IP
+    random_ip = generate_random_ip()
+
     headers = {
         'sec-ch-ua-platform': request.headers.get('sec-ch-ua-platform', '"Windows"'),
         'Referer': request.headers.get('Referer', 'https://deepinfra.com/'),
         'sec-ch-ua': request.headers.get('sec-ch-ua', '"Chromium";v="140", "Not=A?Brand";v="24", "Google Chrome";v="140"'),
         'sec-ch-ua-mobile': request.headers.get('sec-ch-ua-mobile', '?0'),
         'User-Agent': request.headers.get('User-Agent', 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/140.0.0.0 Safari/537.36'),
-        'accept': request.headers.get('accept', 'text-event-stream'),
+        'accept': request.headers.get('accept', 'text/event-stream'),
         'X-Deepinfra-Source': request.headers.get('X-Deepinfra-Source', 'web-embed'),
         'Content-Type': request.headers.get('Content-Type', 'application/json'),
+
+        # Spoofed IP headers
+        'X-Forwarded-For': random_ip,
+        'X-Real-IP': random_ip,
+        'Forwarded': f'for={random_ip};proto=https',
     }
 
     shuffled_urls = random.sample(API_URLS, len(API_URLS))
@@ -47,7 +64,7 @@ async def proxy_deepinfra(request: Request):
     async def stream_generator():
         last_error = None
         for url in shuffled_urls:
-            logging.info(f"Attempting to connect to: {url}")
+            logging.info(f"Attempting to connect to: {url} with spoofed IP {random_ip}")
             try:
                 async with httpx.AsyncClient() as client:
                     async with client.stream("POST", url, headers=headers, json=body, timeout=None) as response:
@@ -60,12 +77,9 @@ async def proxy_deepinfra(request: Request):
                 last_error = e
                 logging.warning(f"Failed to connect to {url}: {e}. Trying next URL.")
                 continue
-
         if last_error:
             logging.error(f"All API endpoints failed. Last error: {last_error}")
             # In a streaming response, we can't easily raise an HTTPException after starting.
             # The connection will simply close, which the client will see as a failed request.
 
     return StreamingResponse(stream_generator(), media_type="text-event-stream")
-
-# The `if __name__ == "__main__":` block is removed as Gunicorn will run the app.