Spaces:

Ethscriptions
/

Tomato-Novel-Downloader-web

Running

Nanny7 commited on Jul 5, 2025

Commit

58e3e9c

1 Parent(s): a4b072a

更新了前端界面：

在进度条下方新增了 <div id="downloadResult"></div>，用于显示下载完成后的链接。
在脚本中取消自动点击，并在下载完成时生成一个可点击的下载链接，提示 “下载完成：点击此处下载文件名”。
现在下载完成后，页面上会出现漂亮的链接，点击即可获取 TXT/ZIP 文件。

Files changed (1) hide show

server.py +52 -8

server.py CHANGED Viewed

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, HTTPException
-from fastapi.responses import FileResponse, HTMLResponse
 import os, shutil, glob
 import tempfile
 import re
@@ -15,18 +15,62 @@ app = FastAPI(
 def root():
     return """
 <!DOCTYPE html>
-<html lang="zh-CN">
 <head>
-    <meta charset="UTF-8">
     <title>番茄小说下载器</title>
 </head>
 <body>
     <h1>番茄小说下载器</h1>
-    <form action="/download" method="get">
-        <label for="book_ids">小说ID列表（逗号或空格分隔）：</label>
-        <textarea id="book_ids" name="book_ids" rows="4" cols="50" required></textarea>
-        <button type="submit">下载</button>
-    </form>
 </body>
 </html>
 """

 from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse, HTMLResponse, StreamingResponse
 import os, shutil, glob
 import tempfile
 import re
 def root():
     return """
 <!DOCTYPE html>
+<html lang=\"zh-CN\">
 <head>
+    <meta charset=\"UTF-8\">
     <title>番茄小说下载器</title>
 </head>
 <body>
     <h1>番茄小说下载器</h1>
+    <label for=\"book_ids\">小说ID列表（逗号或空格分隔）：</label><br/>
+    <textarea id=\"book_ids\" rows=\"4\" cols=\"50\"></textarea><br/>
+    <button id=\"downloadBtn\">下载</button>
+    <div id=\"progress\" style=\"display:none;margin-top:10px;\">\n        <progress id=\"progBar\" value=\"0\" max=\"100\"></progress>\n        <span id=\"progText\">0%</span>\n    </div>\n    <div id=\"downloadResult\" style=\"margin-top:10px;\"></div>
+    <script>
+    document.getElementById('downloadBtn').addEventListener('click', function(){
+        const book_ids = document.getElementById('book_ids').value;
+        if (!book_ids) return;
+        document.getElementById('progress').style.display = 'block';
+        fetch(`/download?book_ids=${encodeURIComponent(book_ids)}`).then(response=>{
+            const total = response.headers.get('Content-Length');
+            const reader = response.body.getReader();
+            let received = 0;
+            const chunks = [];
+            function read() {
+                return reader.read().then(({done, value})=>{
+                    if (done) {
+                        const blob = new Blob(chunks, {type: response.headers.get('Content-Type')});
+                        const url = URL.createObjectURL(blob);
+                        const cd = response.headers.get('Content-Disposition');
+                        let filename = 'download';
+                        if (cd) {
+                            const match = cd.match(/filename="?(.+)"?/);
+                            if (match) filename = match[1];
+                        }
+                        // 生成下载链接并展示
+                        const link = document.createElement('a');
+                        link.href = url;
+                        link.download = filename;
+                        link.innerText = `下载完成：点击此处下载 ${filename}`;
+                        link.style.display = 'block';
+                        link.style.marginTop = '10px';
+                        document.getElementById('downloadResult').appendChild(link);
+                        return;
+                    }
+                    chunks.push(value);
+                    received += value.length;
+                    if (total) {
+                        const percent = Math.round(received / total * 100);
+                        document.getElementById('progBar').value = percent;
+                        document.getElementById('progText').innerText = percent + '%';
+                    }
+                    return read();
+                });
+            }
+            read();
+        });
+    });
+    </script>
 </body>
 </html>
 """