Upload app.py

app.py

@@ -1,11 +1,12 @@
 # ⚙️ 后端逻辑/核心服务端.py (Hugging Face Spaces app.py)
 from fastapi import FastAPI, File, UploadFile, Form
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import Response, JSONResponse  # 新增 Response 用于返回文件流
+from fastapi.responses import Response, JSONResponse
+from pydantic import BaseModel  # 【新增】：引入标准数据模型
 import hashlib
 import urllib.parse
-import urllib.request  # 新增用于代理请求
-import os              # 新增用于读取环境变量
+import urllib.request
+import os
 import 数据库连接 as db
 
 # 引入拆分后的四大业务模块
@@ -24,7 +25,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# 挂载并聚合所有业务子路由
 app.include_router(users_router)
 app.include_router(items_router)
 app.include_router(comments_router)
@@ -52,12 +52,16 @@ async def upload_file(file: UploadFile = File(...), file_type: str = Form(...)):
     url = f"https://huggingface.co/datasets/{db.DATASET_REPO_ID}/resolve/main/{target_dir}/{safe_url_filename}"
     return {"status": "success", "url": url, "display_name": file.filename, "hashed_name": new_filename}
 
+
 # =========================================================
-# 【核心新增】：代理下载私有数据集文件的接口
+# 【核心修复】：使用 Pydantic 定义请求体结构，避免 FastAPI 启动崩溃
 # =========================================================
+class ProxyDownloadRequest(BaseModel):
+    url: str
+
 @app.post("/api/proxy_download")
-async def proxy_download(req_data: dict):
-    target_url = req_data.get("url")
+async def proxy_download(req_data: ProxyDownloadRequest):
+    target_url = req_data.url
     
     # 安全校验：只允许代理下载 huggingface.co 域名的文件
     if not target_url or "huggingface.co" not in target_url:
@@ -75,7 +79,6 @@ async def proxy_download(req_data: dict):
     try:
         with urllib.request.urlopen(req) as response:
             content = response.read()
-            # 直接将读取到的二进制文件流返回给前端
             return Response(content=content, media_type="application/json")
     except Exception as e:
         return JSONResponse(content={"error": f"云端代理下载失败: {str(e)}"}, status_code=500)