Update main.py

main.py

@@ -62,7 +62,7 @@ def proxy_deepinfra():
     try:
         print(f"[INFO] Forwarding POST to {target_url} via {proxy}")
 
-        # 🚫 Ignore user headers completely, always use fixed DeepInfra headers
+        # Always use fixed headers (ignore client-supplied headers)
         forward_headers = dict(DEEPINFRA_HEADERS)
 
         upstream = requests.post(
@@ -83,6 +83,7 @@ def proxy_deepinfra():
         headers = dict(upstream.headers)
         headers["X-Proxy-Used"] = proxy
 
+        # 🔑 Return exactly what DeepInfra sent (status, body, headers)
         return Response(
             stream_with_context(generate()),
             status=upstream.status_code,
@@ -90,7 +91,8 @@ def proxy_deepinfra():
             content_type=upstream.headers.get("Content-Type", "application/json")
         )
 
-    except Exception as e:
+    except requests.exceptions.RequestException as e:
+        # Only if the proxy itself fails (not DeepInfra)
         return jsonify({"error": "Proxy failed", "proxy": proxy, "details": str(e)}), 502
 
 def main():