Update app.py

app.py

@@ -2,7 +2,6 @@ from fastapi import FastAPI, Request, Response
 from fastapi.responses import StreamingResponse
 from fastapi.middleware.cors import CORSMiddleware
 import httpx
-import asyncio
 from typing import Optional
 
 app = FastAPI()
@@ -22,7 +21,7 @@ TARGET_URL = "http://beibeioo.top"
 # 创建一个全局的 httpx 客户端
 async def get_http_client():
     return httpx.AsyncClient(
-        timeout=30.0,  # 30秒超时
+        timeout=30.0,
         follow_redirects=True,
         http2=True
     )
@@ -30,25 +29,17 @@ async def get_http_client():
 @app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
 async def proxy(path: str, request: Request):
     try:
-        # 构建目标URL
         url = f"{TARGET_URL}/{path}"
         
         # 获取并处理请求头
         headers = dict(request.headers)
-        # 删除可能导致问题的头部
         headers.pop('host', None)
         headers.pop('connection', None)
         headers.pop('content-length', None)
         headers.pop('transfer-encoding', None)
         
-        # 如果有 referer，修改它
-        if 'referer' in headers:
-            headers['referer'] = headers['referer'].replace(str(request.base_url), TARGET_URL)
-        
-        # 获取查询参数
+        # 获取查询参数和请求体
         params = dict(request.query_params)
-        
-        # 获取请求体
         body = await request.body()
         
         async with await get_http_client() as client:
@@ -62,18 +53,18 @@ async def proxy(path: str, request: Request):
             
             # 处理响应头
             response_headers = dict(response.headers)
-            # 删除可能导致问题的响应头
             response_headers.pop('transfer-encoding', None)
             response_headers.pop('content-encoding', None)
+            response_headers.pop('content-length', None)  # 移除 content-length 头
             
-            # 修改所有涉及原始域名的响应头
+            # 修改涉及原始域名的响应头
             for key, value in response_headers.items():
                 if isinstance(value, str) and TARGET_URL in value:
                     response_headers[key] = value.replace(TARGET_URL, str(request.base_url).rstrip('/'))
             
-            # 返回响应
-            return StreamingResponse(
-                response.iter_bytes(),
+            # 使用 Response 而不是 StreamingResponse
+            return Response(
+                content=response.content,
                 status_code=response.status_code,
                 headers=response_headers,
                 media_type=response.headers.get('content-type')