Update app.py

app.py

@@ -4,6 +4,7 @@ import math
 import gradio as gr
 from PIL import Image, ImageDraw
 import os
+import shutil
 from utils.keyframe_utils import generate_keyframe_prompt, generate_all_keyframe_images
 
 # Load segments JSON
@@ -51,11 +52,12 @@ def paginate_segments(page=1, page_size=15):
     data = [segment_display(seg) for seg in subset]
     return headers, data, max_page
 
-def download_prompts():
-    if os.path.exists("all_prompts_output.json"):
-        return "all_prompts_output.json"
-    else:
-        return None
+def zip_keyframe_folder():
+    zip_name = "keyframes.zip"
+    if os.path.exists(zip_name):
+        os.remove(zip_name)
+    shutil.make_archive("keyframes", 'zip', "keyframes")
+    return zip_name
 
 def build_interface():
     with gr.Blocks() as demo:
@@ -72,15 +74,14 @@ def build_interface():
             headers, rows, max_page = paginate_segments(page)
             return gr.update(headers=headers, value=rows), f"Page {page} of {max_page}", page
 
-        def generate_and_refresh():
+        def generate_and_refresh(_):
             segments = load_segments()
             generate_all_keyframe_images(segments)
             return update(1)
 
         with gr.Row():
-            with gr.Row(elem_id="page-controls"):
-                prev_btn = gr.Button("⬅", size="sm")
-                next_btn = gr.Button("➡", size="sm")
+            prev_btn = gr.Button("⬅", size="sm")
+            next_btn = gr.Button("➡", size="sm")
 
         prev_btn.click(fn=lambda p: max(1, p - 1), inputs=page_state, outputs=page_state).then(update, inputs=page_state, outputs=[table, total_pages_text, page_state])
         next_btn.click(fn=lambda p: p + 1, inputs=page_state, outputs=page_state).then(update, inputs=page_state, outputs=[table, total_pages_text, page_state])
@@ -89,13 +90,12 @@ def build_interface():
 
         gr.Markdown("### 🖼️ 生成所有分镜的关键帧图像")
         generate_btn = gr.Button("🛠️ 一键生成图像")
-        generate_btn.click(fn=generate_and_refresh, outputs=[table, total_pages_text, page_state])
+        generate_btn.click(fn=generate_and_refresh, inputs=[page_state], outputs=[table, total_pages_text, page_state])
 
-        gr.Markdown("### 📥 下载所有生成的图像提示词")
-        with gr.Row():
-            download_btn = gr.Button("📥 导出 prompts JSON")
-            download_output = gr.File(label="下载文件")
-            download_btn.click(fn=download_prompts, outputs=download_output)
+        gr.Markdown("### 📦 打包下载 keyframes 文件夹（包含图片和 prompts.json）")
+        zip_btn = gr.Button("📁 生成 ZIP")
+        zip_file_output = gr.File(label="下载 ZIP")
+        zip_btn.click(fn=zip_keyframe_folder, outputs=zip_file_output)
 
     return demo