Create app.py

app.py

@@ -0,0 +1,96 @@
+import gradio as gr
+import os
+import shutil
+from jmcomic import create_option, JmcomicException, Img2pdfPlugin
+
+# 定义下载目录 (Hugging Face Spaces 推荐使用 /tmp 或应用根目录)
+BASE_DIR = "download_cache"
+
+def run_download(album_id, cookies_str):
+    # 1. 清理旧数据 (防止空间爆满)
+    if os.path.exists(BASE_DIR):
+        shutil.rmtree(BASE_DIR)
+    os.makedirs(BASE_DIR, exist_ok=True)
+
+    # 2. 构建配置 (不依赖外部 YAML，直接代码生成)
+    # 我们基于默认配置进行修改
+    cc = {
+        'dir_rule': {'base_dir': BASE_DIR},
+        'client': {
+            'domain': [], # 让库自动寻找可用域名
+            'impl': 'api', # 强制使用移动端API，兼容性更好
+        },
+        'download': {
+            'image': {'suffix': '.jpg'} # 统一转为jpg，减小体积且兼容性好
+        },
+        'plugins': {
+            'after_album': [
+                {
+                    'plugin': 'img2pdf', # 使用 PDF 插件
+                    'kwargs': {
+                        'pdf_dir': f'{BASE_DIR}/pdf/', # PDF 输出位置
+                        'filename_rule': 'TITLE',      # PDF 文件名规则：使用本子标题
+                        'delete_original_file': True   # 合并后删除原图
+                    }
+                }
+            ]
+        }
+    }
+    
+    try:
+        # 3. 创建 Option 对象
+        option = create_option(cc)
+        
+        # 4. 设置 Cookies (如果有)
+        # 很多本子需要登录才能看，或者为了避免 IP 封锁，建议传入 Cookies
+        if cookies_str and len(cookies_str.strip()) > 10:
+            # 简单处理：如果是 key=value 格式或 JSON 格式，jmcomic 通常能识别
+            # 这里我们假设用户粘贴的是类似 "key=value; key2=value2" 的浏览器 cookie 字符串
+            # jmcomic 的 header 配置支持直接覆盖
+            option.call_once() # 初始化
+            option.headers['cookie'] = cookies_str
+
+        # 5. 执行下载
+        # download_album 会返回结果列表，或者直接执行
+        print(f"开始下载 Album ID: {album_id}...")
+        jm_result = option.download_album(album_id)
+        
+        # 6. 寻找生成的 PDF
+        pdf_dir = os.path.join(BASE_DIR, 'pdf')
+        if not os.path.exists(pdf_dir):
+            return None, "下载可能成功，但 PDF 目录未找到，请检查日志。"
+            
+        files = os.listdir(pdf_dir)
+        pdf_files = [f for f in files if f.endswith('.pdf')]
+        
+        if not pdf_files:
+            return None, "下载完成，但未生成 PDF 文件 (可能是本子内容为空或权限不足)。"
+            
+        # 返回第一个找到的 PDF
+        final_pdf_path = os.path.join(pdf_dir, pdf_files[0])
+        return final_pdf_path, f"成功！文件名: {pdf_files[0]}"
+
+    except JmcomicException as e:
+        return None, f"JMComic 错误: {str(e)}"
+    except Exception as e:
+        return None, f"系统错误: {str(e)}"
+
+# --- Gradio 界面 ---
+
+with gr.Blocks(title="JMComic Downloader to PDF") as demo:
+    gr.Markdown("## 🚀 JMComic 本子下载器 (PDF 版)")
+    gr.Markdown("输入本子 ID，自动下载并合并为 PDF。")
+    
+    with gr.Row():
+        inp_id = gr.Textbox(label="本子 ID (Album ID)", placeholder="例如: 123456")
+        inp_cookies = gr.Textbox(label="Cookies (可选但推荐)", placeholder="如果在云端无法下载，请粘贴你的浏览器 Cookies 字符串", lines=1)
+    
+    btn_run = gr.Button("开始下载并转换", variant="primary")
+    
+    out_status = gr.Textbox(label="状态信息")
+    out_file = gr.File(label="下载 PDF")
+
+    btn_run.click(fn=run_download, inputs=[inp_id, inp_cookies], outputs=[out_file, out_status])
+
+if __name__ == "__main__":
+    demo.launch()