Upload app.py

app.py

@@ -54,8 +54,10 @@ async def upload_file(file: UploadFile = File(...), file_type: str = Form(...)):
 
 
 # =========================================================
-# 【核心修复】：使用 Pydantic 定义请求体结构，避免 FastAPI 启动崩溃
+# 【核心修复】：使用 huggingface_hub 官方库替代 urllib 解决重定向 401 问题
 # =========================================================
+from huggingface_hub import hf_hub_download
+
 class ProxyDownloadRequest(BaseModel):
     url: str
 
@@ -63,22 +65,35 @@ class ProxyDownloadRequest(BaseModel):
 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:
-        return JSONResponse(content={"error": "无效的下载链接或拒绝访问非 HF 域名"}, status_code=400)
+    # 校验 URL 格式
+    if not target_url or "resolve/main/" not in target_url:
+        return JSONResponse(content={"error": "无效的 Hugging Face 下载链接"}, status_code=400)
         
     hf_token = os.environ.get("HF_TOKEN")
-    
-    # 构造请求去私有库取文件
-    req = urllib.request.Request(target_url, headers={'User-Agent': 'Mozilla/5.0'})
-    
-    # 如果环境配置了 Token，则将其塞入请求头中用于突破私有权限
-    if hf_token:
-        req.add_header("Authorization", f"Bearer {hf_token}")
+    if not hf_token:
+        return JSONResponse(content={"error": "云端环境变量未配置 HF_TOKEN，无法读取私有库"}, status_code=401)
         
     try:
-        with urllib.request.urlopen(req) as response:
-            content = response.read()
-            return Response(content=content, media_type="application/json")
+        # 1. 从前端传来的 URL 中切割出仓库内的真实相对路径
+        # 例如从 https://.../resolve/main/apps/123_%E6%B5%8B.json 提取出 apps/123_%E6%B5%8B.json
+        repo_path_encoded = target_url.split("resolve/main/")[-1]
+        
+        # 2. 对可能存在的被编码的中文字符进行解码 (恢复为 apps/123_测.json)
+        repo_path = urllib.parse.unquote(repo_path_encoded)
+        
+        # 3. 调用官方库直接从私有 Dataset 拉取文件（完美处理鉴权和重定向）
+        cached_file_path = hf_hub_download(
+            repo_id=db.DATASET_REPO_ID,
+            repo_type="dataset",
+            filename=repo_path,
+            token=hf_token
+        )
+        
+        # 4. 读取缓存在云端的本地文件流并返回给前端
+        with open(cached_file_path, "rb") as f:
+            content = f.read()
+            
+        return Response(content=content, media_type="application/json")
+        
     except Exception as e:
-        return JSONResponse(content={"error": f"云端代理下载失败: {str(e)}"}, status_code=500)
+        return JSONResponse(content={"error": f"云端官方库读取失败: {str(e)}"}, status_code=500)