无需 Gradio

Dockerfile

@@ -10,7 +10,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY . /app
 
 # 暴露端口
-EXPOSE 7860
+EXPOSE 7860 8000
 
 # 启动应用
-CMD ["python", "app.py"]
+CMD ["uvicorn", "server:app", "--host", "0.0.0.0", "--port", "8000"]

requirements.txt

@@ -1,6 +1,8 @@
-gradio>=3.50.1
+# gradio>=3.50.1 (removed for FastAPI deployment)
 requests
 beautifulsoup4
 tqdm
 fake-useragent
 pycryptodome
+fastapi
+uvicorn[standard]

server.py

@@ -0,0 +1,37 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse
+import os, shutil, glob
+from main import Run
+
+app = FastAPI(
+    title="番茄小说下载器API",
+    description="通过 /download?book_id=xxx 获取小说文本文件",
+)
+
+@app.get("/")
+def root():
+    return {"message": "番茄小说下载器，使用 /download?book_id=小说ID 下载"}
+
+@app.get("/download")
+def download(book_id: str):
+    save_path = "downloads"
+    # 清理旧文件
+    if os.path.exists(save_path):
+        shutil.rmtree(save_path)
+    os.makedirs(save_path, exist_ok=True)
+
+    # 调用下载核心
+    try:
+        Run(book_id, save_path)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"下载失败: {str(e)}")
+
+    files = glob.glob(os.path.join(save_path, "*.txt"))
+    if not files:
+        raise HTTPException(status_code=404, detail="未生成txt文件，下载失败")
+    file_path = files[0]
+    return FileResponse(
+        path=file_path,
+        filename=os.path.basename(file_path),
+        media_type="text/plain"
+    )