Upload app.py

app.py

@@ -24,6 +24,9 @@ from models_sql import Ownership
 
 app = FastAPI(title="ComfyUI Ranking Community API")
 
+# ==========================================
+# 增加一个根目录健康检查接口，专治 HF Spaces 的 "Restarting" 误判
+# ==========================================
 @app.get("/")
 def health_check():
     return {"status": "ok", "message": "ComfyUI Ranking API is running perfectly!"}
@@ -54,8 +57,16 @@ app.include_router(proxy_router)
 def upload_file(file: UploadFile = File(...), file_type: str = Form(...)):
     # 同步读取文件内容
     content = file.file.read() 
-    if len(content) > 10 * 1024 * 1024:
-        raise HTTPException(status_code=400, detail="文件过大，请限制在 10MB 以内")
+    
+    # 🟢 动态文件大小风控
+    max_size = 10 * 1024 * 1024  # 默认兜底 10MB
+    if file_type == "avatar":
+        max_size = 2 * 1024 * 1024   # 头像严格限制为 2MB
+    elif file_type == "cover":
+        max_size = 5 * 1024 * 1024   # 封面图限制为 5MB
+        
+    if len(content) > max_size:
+        raise HTTPException(status_code=400, detail=f"文件过大，{file_type} 类型请限制在 {max_size // (1024*1024)}MB 以内")
         
     ext = file.filename.split(".")[-1].lower()
     if ext not in ["jpg", "jpeg", "png", "gif", "webp", "json", "mp4"]: