Update app.py

app.py

@@ -44,14 +44,23 @@ def paginate_segments(page=1, page_size=15):
     data = [segment_display(seg) for seg in subset]
     return headers, data, max_page
 
+# Function to return downloadable JSON file
+def download_prompts():
+    if os.path.exists("all_prompts_output.json"):
+        return "all_prompts_output.json"
+    else:
+        return None
+
 # Gradio interface
 def build_interface():
     with gr.Blocks() as demo:
         gr.Markdown("## 🎬 Keyframe Candidate Viewer")
         page_state = gr.State(1)
-        table = gr.Dataframe(headers=["Segment ID", "Description", "Candidate 1", "Candidate 2", "Candidate 3"],
-                             datatype=["str", "str", "image", "image", "image"],
-                             row_count=15)
+        table = gr.Dataframe(
+            headers=["Segment ID", "Description", "Candidate 1", "Candidate 2", "Candidate 3"],
+            datatype=["str", "str", "image", "image", "image"],
+            row_count=15
+        )
         total_pages_text = gr.Textbox(label="Page Info", interactive=False)
 
         def update(page):
@@ -69,7 +78,9 @@ def build_interface():
 
         gr.Markdown("### 📥 下载所有生成的图像提示词")
         with gr.Row():
-            download_button = gr.File(label="下载 prompts", value="all_prompts_output.json")
+            download_btn = gr.Button("📥 导出 prompts JSON")
+            download_output = gr.File(label="下载文件")
+            download_btn.click(fn=download_prompts, outputs=download_output)
 
     return demo