Add direct streaming zip download button

app.py

@@ -3,8 +3,7 @@ import re
 import queue
 import threading
 import zipfile
-from urllib.parse import quote, urlencode
-import html
+from urllib.parse import quote
 
 import requests
 import gradio as gr
@@ -61,7 +60,11 @@ def get_repo_resolve_prefix(repo_type: str, repo: str) -> str:
 
 def list_repo_files(repo: str, revision: str, repo_type: str):
     api_path = get_repo_api_path(repo_type, repo)
-    url = f"{HF_BASE}{api_path}/tree/{quote(revision, safe='')}?recursive=1"
+
+    url = (
+        f"{HF_BASE}{api_path}"
+        f"/tree/{quote(revision, safe='')}?recursive=1"
+    )
 
     response = requests.get(
         url,
@@ -76,14 +79,17 @@ def list_repo_files(repo: str, revision: str, repo_type: str):
 
     items = response.json()
 
-    return [
-        {
-            "path": item["path"],
-            "size": item.get("size"),
-        }
-        for item in items
-        if item.get("type") == "file"
-    ]
+    files = []
+    for item in items:
+        if item.get("type") == "file":
+            files.append(
+                {
+                    "path": item.get("path"),
+                    "size": item.get("size"),
+                }
+            )
+
+    return files
 
 
 def get_file_download_url(repo: str, repo_type: str, revision: str, path: str) -> str:
@@ -96,17 +102,19 @@ def get_file_download_url(repo: str, repo_type: str, revision: str, path: str) -
     )
 
 
+def safe_filename(value: str) -> str:
+    value = re.sub(r"[^\w.-]+", "_", value)
+    return value.strip("_") or "repo"
+
+
 class QueueWriter:
     """
-    给 zipfile 使用的“假文件对象”。
+    给 zipfile 使用的流式 writer。
 
-    zipfile 往这里 write 的每一段 bytes，
-    都会立刻放进队列，然后 StreamingResponse 再把它发给浏览器。
+    zipfile 写出来的每一段 bytes 都会进入 queue，
+    FastAPI StreamingResponse 再把 queue 里的 bytes 发送给浏览器。
 
-    关键点：
-    不落盘。
-    不先生成完整 zip。
-    一边从 HF 拉文件，一边写 zip，一边传给浏览器。
+    所以这里不会先生成完整 zip 文件。
     """
 
     def __init__(self, output_queue: queue.Queue):
@@ -139,6 +147,7 @@ def stream_repo_as_zip(repo: str, revision: str, repo_type: str, files: list):
             ) as zip_file:
                 for file in files:
                     file_path = file["path"]
+
                     print(f"正在写入 ZIP：{file_path}", flush=True)
 
                     url = get_file_download_url(
@@ -197,17 +206,12 @@ def stream_repo_as_zip(repo: str, revision: str, repo_type: str, files: list):
         yield item
 
 
-def safe_filename(value: str) -> str:
-    value = re.sub(r"[^\w.-]+", "_", value)
-    return value.strip("_") or "repo"
-
-
 api_app = FastAPI()
 
 
 @api_app.get("/api/hf-model-zip")
 def hf_model_zip(
-    repo: str = Query(..., description="例如 sshleifer/tiny-gpt2"),
+    repo: str = Query(..., description="例如 Qwen/Qwen3.6-35B-A3B"),
     revision: str = Query("main"),
     repoType: str = Query("model"),
 ):
@@ -224,7 +228,7 @@ def hf_model_zip(
     if not repo or "/" not in repo:
         raise HTTPException(
             status_code=400,
-            detail="repo 参数格式错误，应为 owner/name，例如 sshleifer/tiny-gpt2",
+            detail="repo 参数格式错误，应为 owner/name，例如 Qwen/Qwen3.6-35B-A3B",
         )
 
     print(f"准备下载仓库：{repo}", flush=True)
@@ -258,51 +262,54 @@ def hf_model_zip(
     )
 
 
-def build_download_link(repo: str, revision: str, repo_type: str):
-    repo = repo.strip()
-    revision = revision.strip() or "main"
+download_js = """
+(repo, revision, repoType) => {
+    repo = (repo || "").trim();
+    revision = (revision || "main").trim();
+    repoType = (repoType || "model").trim();
 
-    if not repo or "/" not in repo:
-        return "<p style='color:red'>repo 参数格式错误，应为 owner/name，例如 sshleifer/tiny-gpt2</p>"
-
-    params = urlencode(
-        {
-            "repo": repo,
-            "revision": revision,
-            "repoType": repo_type,
-        }
-    )
+    if (!repo || !repo.includes("/")) {
+        alert("repo 参数格式错误，应为 owner/name，例如 Qwen/Qwen3.6-35B-A3B");
+        return [repo, revision, repoType];
+    }
 
-    url = f"/api/hf-model-zip?{params}"
-
-    return f"""
-    <div style="padding: 12px; border: 1px solid #ddd; border-radius: 8px;">
-      <p><b>下载链接已生成：</b></p>
-      <p>
-        <a href="{html.escape(url)}" target="_blank" download>
-          点击这里开始流式下载 ZIP
-        </a>
-      </p>
-      <p style="font-size: 13px; color: #666;">
-        说明：这个链接不会先把完整 ZIP 保存到 Space。
-        它会将 Hugging Face 文件流实时写入 ZIP 流，并直接传给浏览器。
-      </p>
-      <code>{html.escape(url)}</code>
-    </div>
-    """
+    const params = new URLSearchParams({
+        repo: repo,
+        revision: revision,
+        repoType: repoType
+    });
+
+    const url = `/api/hf-model-zip?${params.toString()}`;
+
+    const a = document.createElement("a");
+    a.href = url;
+    a.download = "";
+    a.style.display = "none";
+
+    document.body.appendChild(a);
+    a.click();
+    document.body.removeChild(a);
+
+    return [repo, revision, repoType];
+}
+"""
+
+
+def noop(repo, revision, repo_type):
+    return None
 
 
 with gr.Blocks(title="Hugging Face ZIP 流式下载器") as demo:
     gr.Markdown("# Hugging Face ZIP 流式下载器")
     gr.Markdown(
-        "输入 Hugging Face 模型、数据集或 Space 仓库名，生成一个流式 ZIP 下载链接。"
+        "输入 Hugging Face 模型、数据集或 Space 仓库名，点击按钮后直接开始流式下载 ZIP。"
     )
 
     with gr.Row():
         repo_input = gr.Textbox(
             label="仓库名",
             value="sshleifer/tiny-gpt2",
-            placeholder="例如：Qwen/Qwen2.5-0.5B-Instruct",
+            placeholder="例如：Qwen/Qwen3.6-35B-A3B",
         )
 
         revision_input = gr.Textbox(
@@ -316,13 +323,25 @@ with gr.Blocks(title="Hugging Face ZIP 流式下载器") as demo:
             value="model",
         )
 
-    button = gr.Button("生成下载链接")
-    output = gr.HTML(label="下载链接")
+    download_button = gr.Button("下载 ZIP", variant="primary")
 
-    button.click(
-        fn=build_download_link,
+    gr.Markdown(
+        """
+说明：
+
+- 点击按钮后会直接触发浏览器下载。
+- 不会先把完整 ZIP 保存到 Space。
+- 流量路径是：Hugging Face → Space → 客户浏览器。
+- 私有模型或 gated 模型需要在 Space Secrets 里配置 `HF_TOKEN`。
+        """
+    )
+
+    download_button.click(
+        fn=noop,
         inputs=[repo_input, revision_input, repo_type_input],
-        outputs=output,
+        outputs=[],
+        js=download_js,
+        queue=False,
     )
 
 
@@ -333,10 +352,15 @@ app = gr.mount_gradio_app(
     ssr_mode=False,
 )
 
+
 if __name__ == "__main__":
     import uvicorn
 
-    port = int(os.environ.get("PORT", os.environ.get("GRADIO_SERVER_PORT", "7860")))
+    port = int(
+        os.environ.get("PORT")
+        or os.environ.get("GRADIO_SERVER_PORT")
+        or "7860"
+    )
 
     uvicorn.run(
         app,