Update app.py

app.py

@@ -139,6 +139,7 @@ with gr.Blocks() as demo:
         output_files = gr.Files(label="分割后的文件")
     with gr.Row():
         submit_btn = gr.Button("开始分割")
+        download_all_btn = gr.Button("批量下载")
 
     # 动态显示/隐藏输入框
     def toggle_inputs(mode):
@@ -147,6 +148,24 @@ with gr.Blocks() as demo:
         else:
             return gr.Textbox(visible=False), gr.Number(visible=True)
 
+    # 批量下载功能
+    def download_all_files(file_list):
+        if not file_list:
+            return None
+        # 创建一个临时zip文件
+        import tempfile
+        import zipfile
+        import shutil
+
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.zip') as temp_zip:
+            with zipfile.ZipFile(temp_zip.name, 'w') as zf:
+                for file_path in file_list:
+                    # 获取文件名
+                    file_name = os.path.basename(file_path)
+                    # 将文件添加到zip中
+                    zf.write(file_path, file_name)
+            return temp_zip.name
+
     mode.change(toggle_inputs, inputs=mode, outputs=[split_pages, max_size_mb])
 
     # 绑定处理函数
@@ -155,7 +174,15 @@ with gr.Blocks() as demo:
         inputs=[input_pdf, mode, split_pages, max_size_mb],
         outputs=output_files
     )
+
+    # 绑定批量下载函数
+    download_all_btn.click(
+        download_all_files,
+        inputs=[output_files],
+        outputs=gr.File(label="下载所有文件")
+    )
+
     theme=gr.themes.Soft()
 
 # 启动应用
-demo.launch()
+demo.launch()