update

app.py

@@ -27,8 +27,36 @@ from PIL import Image
 cached_latest_posts_df = None
 last_fetched = None
 
-from moviepy.editor import VideoFileClip
+import os
+import tempfile
+from zipfile import ZipFile
+
 import numpy as np
+from moviepy.editor import VideoFileClip
+from PIL import Image
+
+
+def download_samples(video_url, num_frames):
+    frames = extract_frames_moviepy(video_url, num_frames)
+
+    # Create a temporary directory to store the images
+    with tempfile.TemporaryDirectory() as temp_dir:
+        # Save each frame as a JPEG image in the temporary directory
+        for i, frame in enumerate(frames):
+            frame_path = os.path.join(temp_dir, f"frame_{i}.jpg")
+            frame.save(
+                frame_path, format="JPEG", quality=85
+            )  # Adjust quality as needed
+
+        # Create a zip file in a persistent location
+        zip_path = "frames.zip"
+        with ZipFile(zip_path, "w") as zipf:
+            for i in range(num_frames):
+                frame_path = os.path.join(temp_dir, f"frame_{i}.jpg")
+                zipf.write(frame_path, os.path.basename(frame_path))
+
+    # Return the path of the zip file
+    return zip_path
 
 
 def extract_frames_moviepy(video_url, num_frames=10):
@@ -147,6 +175,13 @@ with gr.Blocks() as demo:
 
         sampled_frames = gr.Gallery()
 
+        download_samples_btn = gr.Button("Download Samples")
+        output_files = gr.File()
+
+        download_samples_btn.click(
+            download_samples, inputs=[video_player, num_frames], outputs=[output_files]
+        )
+
     sample_btn.click(
         extract_frames_moviepy,
         inputs=[video_player, num_frames],