Update app.py

app.py

@@ -8,6 +8,7 @@ import numpy as np
 import uuid
 import base64
 import torch
+import shutil
 
 # Auto-select device: GPU if available
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -195,15 +196,28 @@ def process(video_file):
     html_output_path = os.path.join(base_dir, f"{video_id}.html")
     final_html = generate_html(entries, video_id, video_path, screenshots_dir, plots_dir, html_output_path)
 
-    return final_html
+    # Create ZIP of screenshots
+    zip_path = os.path.join(base_dir, f"{video_id}_screenshots.zip")
+    shutil.make_archive(zip_path.replace(".zip", ""), "zip", screenshots_dir)
+
+    # Return HTML content (for preview), HTML file (for download), and screenshots zip
+    with open(final_html, "r", encoding="utf-8") as f:
+        html_content = f.read()
+
+    return html_content, final_html, zip_path
+
 
 # Gradio UI
 demo = gr.Interface(
     fn=process,
     inputs=[gr.File(label="Upload Video", file_types=[".mp4", ".mov", ".mkv"])],
-    outputs=gr.File(label="Download Annotated HTML"),
+    outputs=[
+        gr.HTML(label="Preview Annotated Transcript"),
+        gr.File(label="Download Annotated HTML"),
+        gr.File(label="Download Screenshots (ZIP)")
+    ],
     title="Video Annotated Transcript",
-    description="🎥 Upload a video file (mp4/mov/mkv). The tool will transcribe speech, capture screenshots, analyze sound intensity, and generate an editable HTML transcript."
+    description="🎥 Upload a video file. The tool transcribes speech, captures screenshots, analyzes sound intensity, and generates an editable HTML transcript."
 )
 
 if __name__ == "__main__":