Upload app.py

app.py

@@ -1,8 +1,11 @@
 # ⚙️ 后端逻辑/核心服务端.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 用于返回文件流
 import hashlib
-import urllib.parse  # 【核心新增】：用于处理中文文件名的 URL 编码
+import urllib.parse
+import urllib.request  # 新增用于代理请求
+import os              # 新增用于读取环境变量
 import 数据库连接 as db
 
 # 引入拆分后的四大业务模块
@@ -37,9 +40,6 @@ async def upload_file(file: UploadFile = File(...), file_type: str = Form(...)):
     file_hash = hashlib.md5(content).hexdigest()[:10]
     
     new_filename = f"{file_hash}_{file.filename}"
-    
-    # 【核心修复】：由于标准的 HTTP URL 不支持直接包含中文，
-    # 我们对文件名进行 URL 安全转码 (如把 '测试' 转为 '%E6%B5%8B%E8%AF%95')
     safe_filename = urllib.parse.quote(file.filename)
     safe_url_filename = f"{file_hash}_{safe_filename}"
     
@@ -47,9 +47,35 @@ async def upload_file(file: UploadFile = File(...), file_type: str = Form(...)):
     target_dir = dir_mapping.get(file_type, "others")
     full_path_in_repo = f"{target_dir}/{new_filename}"
     
-    # 保存文件时使用原始名称（文件系统支持中文）
     db.save_file(full_path_in_repo, content)
     
-    # 返回给前端的下载链接，必须使用 URL Encode 编码后的名称，防止客户端 ascii 报错
     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}
+    return {"status": "success", "url": url, "display_name": file.filename, "hashed_name": new_filename}
+
+# =========================================================
+# 【核心新增】：代理下载私有数据集文件的接口
+# =========================================================
+@app.post("/api/proxy_download")
+async def proxy_download(req_data: dict):
+    target_url = req_data.get("url")
+    
+    # 安全校验：只允许代理下载 huggingface.co 域名的文件
+    if not target_url or "huggingface.co" not in target_url:
+        return JSONResponse(content={"error": "无效的下载链接或拒绝访问非 HF 域名"}, 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}")
+        
+    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)