Update app.py

app.py

@@ -9,16 +9,12 @@ app = Flask(__name__)
 TARGET_API = os.getenv("TARGET_API", "https://huggingface.co")
 
 # Path mappings from environment variable
-# Expected format: {"path1": "mapped_path1", "path2": "mapped_path2"}
 def get_path_mappings():
     mappings_str = os.getenv("PATH_MAPPINGS", '{"/": "/"}')
     try:
         return json.loads(mappings_str)
     except json.JSONDecodeError:
-        # Fallback to default mappings if JSON is invalid
-        return {
-            "/": "/",
-        }
+        return {"/": "/"}
 
 PATH_MAPPINGS = get_path_mappings()
 
@@ -37,48 +33,31 @@ def proxy(path):
     # Construct target URL
     target_url = f"{TARGET_API}{full_path}"
 
-    # Forward the request to the target API
-    headers = {key: value for key, value in request.headers if key != 'Host'}
-
-    # Handle streaming response
-    if request.method == 'POST':
-        response = requests.post(
-            target_url,
-            headers=headers,
-            json=request.get_json(silent=True),
-            params=request.args,
-            stream=True
-        )
-    elif request.method == 'GET':
-        response = requests.get(
-            target_url,
-            headers=headers,
-            params=request.args,
-            stream=True
-        )
-    elif request.method == 'PUT':
-        response = requests.put(
-            target_url,
-            headers=headers,
-            json=request.get_json(silent=True),
-            params=request.args,
-            stream=True
-        )
-    elif request.method == 'DELETE':
-        response = requests.delete(
-            target_url,
-            headers=headers,
-            params=request.args,
-            stream=True
-        )
-    elif request.method == 'PATCH':
-        response = requests.patch(
-            target_url,
-            headers=headers,
-            json=request.get_json(silent=True),
-            params=request.args,
-            stream=True
-        )
+    # ==================== HEADER 过滤逻辑 ====================
+    # 定义允许转发的必要 Header 白名单（统一小写，方便匹配）
+    ESSENTIAL_HEADERS = {'content-type', 'accept', 'user-agent'}
+    
+    headers = {}
+    for key, value in request.headers.items():
+        if key.lower() in ESSENTIAL_HEADERS:
+            headers[key] = value
+    # ========================================================
+
+    # 统一构建 requests 请求参数
+    request_kwargs = {
+        "method": request.method,
+        "url": target_url,
+        "headers": headers,
+        "params": request.args,
+        "stream": True
+    }
+
+    # 仅在有请求体的动词中带上 JSON 数据
+    if request.method in ['POST', 'PUT', 'PATCH']:
+        request_kwargs["json"] = request.get_json(silent=True)
+
+    # 发起请求
+    response = requests.request(**request_kwargs)
 
     # Create a response with the same status code, headers, and streaming content
     def generate():
@@ -105,4 +84,4 @@ def index():
 
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=False)
+    app.run(host='0.0.0.0', port=7860, debug=False)