Update app.py

app.py

@@ -61,7 +61,7 @@ def proxy(path):
         random_key = random.choice(api_keys)
         headers['Authorization'] = f"Bearer {random_key}"
 
-    # Handle streaming response
+    # Handle POST requests with streaming
     if request.method == 'POST':
         response = requests.post(
             target_url,
@@ -70,31 +70,38 @@ def proxy(path):
             params=request.args,
             stream=True
         )
+
+        # Create a streaming response for POST
+        def generate():
+            for chunk in response.iter_content(chunk_size=8192):
+                yield chunk
+
+        # Create flask streaming response
+        proxy_response = Response(
+            stream_with_context(generate()),
+            status=response.status_code
+        )
+        # Forward response headers
+        for key, value in response.headers.items():
+            if key.lower() not in ('content-length', 'transfer-encoding', 'connection'):
+                proxy_response.headers[key] = value
+    
+        return proxy_response
+
+    # Handle GET requests without streaming
     elif request.method == 'GET':
         response = requests.get(
             target_url,
             headers=headers,
-            params=request.args,
-            stream=True
+            params=request.args
         )
 
-    # Create a response with the same status code, headers, and streaming content
-    def generate():
-        for chunk in response.iter_content(chunk_size=8192):
-            yield chunk
-
-    # Create flask response
-    proxy_response = Response(
-        stream_with_context(generate()),
-        status=response.status_code
-    )
-
-    # Forward response headers
-    for key, value in response.headers.items():
-        if key.lower() not in ('content-length', 'transfer-encoding', 'connection'):
-            proxy_response.headers[key] = value
-
-    return proxy_response
+        # Create a regular response for GET
+        proxy_response = Response(
+            response.content,
+            status=response.status_code
+        )
+        return proxy_response
 
 @app.route('/', methods=['GET'])
 def index():