Update app.py

app.py

@@ -2,7 +2,6 @@
 # https://username-spaceName.hf.space/proxy/anthropic/v1
 # Streaming is not supported
 
-
 import logging
 import json
 from flask import Flask, request, Response
@@ -19,37 +18,24 @@ def relay_request(path):
 
     relay_url = 'https://' + path
 
+    headers_ = {}
     
-    headers_ = dict()
-    
-    
-    if 'Content-Type' in request.headers:
-        headers_['Content-Type'] = request.headers['Content-Type']
-    if 'x-api-key' in request.headers:
-        headers_['Authorization'] = "Bearer " + request.headers['x-api-key']
-    if 'Authorization' in request.headers:
-        headers_['Authorization'] = request.headers['Authorization']
-
-    if request.method == 'POST':
-        response = requests.request(
-            method='POST',
-            url=relay_url,
-            headers=headers_,
-            json=json.loads(request.get_data()),
-            allow_redirects=False 
-        )
-    else:
-        response = requests.request(
-            method=request.method,
-            url=relay_url,
-            headers=headers_,
-            allow_redirects=False 
-        )
-        
-
-
-    headers = [(name, value) for name, value in response.headers.items()]
-    return Response(response.content, response.status_code, headers)
+    # Copy headers from the client request
+    for key, value in request.headers.items():
+        if key.lower() != 'host':
+            headers_[key] = value
+
+    # Proxy the request
+    response = requests.request(
+        method=request.method,
+        url=relay_url,
+        headers=headers_,
+        data=request.stream,
+        stream=True  # Stream the response back to the client
+    )
+
+    # Stream the response content back to the client
+    return Response(response.iter_content(chunk_size=1024), status=response.status_code, headers=response.headers.items())
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)