Update main.py

main.py

@@ -2,7 +2,7 @@ import requests
 import random
 import time
 import threading
-from flask import Flask, request, Response, jsonify
+from flask import Flask, request, Response, jsonify, stream_with_context
 
 PROXY_LIST_URL = "https://proxies.typegpt.net/ips.txt"
 proxies_cache = []
@@ -10,6 +10,18 @@ last_refresh = 0
 
 app = Flask(__name__)
 
+# DeepInfra required headers
+DEEPINFRA_HEADERS = {
+    "accept": "text/event-stream",
+    "content-type": "application/json",
+    "referer": "https://deepinfra.com/",
+    "sec-ch-ua": '"Chromium";v="140", "Not=A?Brand";v="24", "Google Chrome";v="140"',
+    "sec-ch-ua-mobile": "?0",
+    "sec-ch-ua-platform": '"Windows"',
+    "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",
+    "x-deepinfra-source": "web-embed",
+}
+
 def fetch_proxies():
     try:
         resp = requests.get(PROXY_LIST_URL, timeout=10)
@@ -38,11 +50,9 @@ def refresh_proxies_loop():
 def health():
     return "Healthy", 200
 
-@app.route("/<path:target_url>", methods=["GET", "POST", "PUT", "DELETE", "PATCH", "OPTIONS", "HEAD"])
-def proxy_request(target_url):
-    if not target_url.startswith("http"):
-        target_url = "http://" + target_url
-
+@app.route("/deepinfra", methods=["POST"])
+def proxy_deepinfra():
+    target_url = "https://api.deepinfra.com/v1/openai/chat/completions"
     proxy = get_random_proxy(proxies_cache)
     if not proxy:
         return jsonify({"error": "No proxies available"}), 500
@@ -50,30 +60,40 @@ def proxy_request(target_url):
     proxies = {"http": proxy, "https": proxy}
 
     try:
-        print(f"[INFO] Forwarding {request.method} to {target_url} via {proxy}")
+        print(f"[INFO] Forwarding POST to {target_url} via {proxy}")
+
+        # Start with required DeepInfra headers, merge client’s headers on top
+        forward_headers = dict(DEEPINFRA_HEADERS)
+        for k, v in request.headers:
+            if k.lower() != "host":
+                forward_headers[k] = v
 
-        # Forward original request (method, headers, body, params)
-        resp = requests.request(
-            method=request.method,
+        # Stream request
+        upstream = requests.post(
             url=target_url,
-            headers={k: v for k, v in request.headers if k.lower() != "host"},
+            headers=forward_headers,
             data=request.get_data(),
-            cookies=request.cookies,
             params=request.args,
             proxies=proxies,
-            timeout=30,
-            allow_redirects=False
+            stream=True,
+            timeout=300,
         )
 
-        # Copy headers + add X-Proxy-Used
-        headers = dict(resp.headers)
+        def generate():
+            for chunk in upstream.iter_content(chunk_size=None):
+                if chunk:
+                    yield chunk
+
+        headers = dict(upstream.headers)
         headers["X-Proxy-Used"] = proxy
 
         return Response(
-            resp.content,
-            status=resp.status_code,
+            stream_with_context(generate()),
+            status=upstream.status_code,
             headers=headers,
+            content_type=upstream.headers.get("Content-Type", "application/json")
         )
+
     except Exception as e:
         return jsonify({"error": "Proxy failed", "proxy": proxy, "details": str(e)}), 502