import os
import glob
import shutil
import gradio as gr
from main import Run

def download_novel(book_id: str):
    """
    下载指定小说 ID 并返回生成的 txt 文件路径
    """
    save_path = "downloads"
    # 清理旧文件
    if os.path.exists(save_path):
        shutil.rmtree(save_path)
    os.makedirs(save_path, exist_ok=True)

    # 调用下载逻辑
    Run(book_id, save_path)

    # 查找生成的 txt 文件
    files = glob.glob(os.path.join(save_path, "*.txt"))
    if files:
        return files[0]
    else:
        return None

# Gradio 界面
iface = gr.Interface(
    fn=download_novel,
    inputs=gr.Textbox(label="小说 ID"),
    outputs=gr.File(label="下载的小说文件"),
    title="番茄小说在线下载器",
    description="请输入小说 ID，点击下载后可获得 txt 文件。",
)

if __name__ == "__main__":
    iface.launch(server_name="0.0.0.0", server_port=7860)