Create loader.py

loader.py

@@ -0,0 +1,59 @@
+import os
+import subprocess
+import sys
+from huggingface_hub import snapshot_download
+
+# ================= 配置部分 =================
+# 填写你的私有仓库地址，格式：用户名/仓库名
+PRIVATE_REPO_ID = "HongshengY/NeuralSketch2Surf" 
+
+# 获取 Secret Token (需要在 Space Settings -> Secrets 中设置 HF_TOKEN)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
+def main():
+    if not HF_TOKEN:
+        print("❌ 错误: 未检测到 HF_TOKEN。请在 Settings -> Secrets 中添加。")
+        sys.exit(1)
+
+    print(f"🚀 正在从私有仓库 {PRIVATE_REPO_ID} 加载代码...")
+    
+    try:
+        # 1. 下载私有仓库的所有文件到当前目录 (/app)
+        snapshot_download(
+            repo_id=PRIVATE_REPO_ID,
+            repo_type="space",
+            local_dir=".",
+            token=HF_TOKEN,
+            # 排除不必要的文件以加快下载
+            ignore_patterns=[".gitattributes", "README.md", ".git/*"]
+        )
+        print("✅ 代码与模型下载完成！")
+
+        # 2. 安装依赖
+        # 虽然 Dockerfile 会预装一些，但为了确保版本匹配，还是跑一下 requirements.txt
+        if os.path.exists("backend/requirements.txt"):
+            print("📦 正在检查并安装 backend/requirements.txt...")
+            subprocess.run([sys.executable, "-m", "pip", "install", "-r", "backend/requirements.txt"], check=True)
+        
+        # 3. 启动 FastAPI 应用
+        # 对应你 README 中的启动命令：uvicorn backend.server:app --host 0.0.0.0 --port 7860
+        print("🔥 正在启动 NeuralSketch2Surf Web...")
+        
+        # 构造启动命令
+        # 注意：Hugging Face Spaces 强制要求监听 7860 端口
+        cmd = [
+            "uvicorn", 
+            "backend.server:app", 
+            "--host", "0.0.0.0", 
+            "--port", "7860"
+        ]
+        
+        # 使用 exec 替换当前进程，这样 uvicorn 就会成为 PID 1，接收所有信号
+        os.execvp("uvicorn", cmd)
+
+    except Exception as e:
+        print(f"❌ 发生错误: {e}")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()