update

app.py

@@ -27,6 +27,31 @@ from PIL import Image
 cached_latest_posts_df = None
 last_fetched = None
 
+from moviepy.editor import VideoFileClip
+import numpy as np
+
+
+def extract_frames_moviepy(video_url, num_frames=10):
+    try:
+        print(f"Extracting {num_frames} frames from {video_url}")
+        # Load the video
+        clip = VideoFileClip(video_url)
+
+        # Calculate the times to extract frames
+        duration = clip.duration
+        frame_times = np.linspace(0, duration, num_frames, endpoint=False)
+
+        frameas = []
+        # Extract and save frames
+        for i, time in enumerate(frame_times):
+            frame = clip.get_frame(time)
+            frame_image = Image.fromarray(frame)
+            frameas.append(frame_image)
+
+        return frameas
+    except e as Exception:
+        raise gr.Error(f"Error extracting frames from video: {e}")
+
 
 def get_latest_pots():
     global cached_latest_posts_df
@@ -107,15 +132,28 @@ with gr.Blocks() as demo:
                 lines=1, placeholder="Post url or id here", label="URL or Post ID"
             )
             load_btn = gr.Button("Load")
-            video_player = gr.Video()
+            video_player = gr.Video(interactive=False)
 
         with gr.Column():
             gr.Markdown("## Latest Posts")
             latest_post_dataframe = gr.Dataframe()
             get_latest_pots_btn = gr.Button("Refresh Latest Posts")
 
-    load_btn.click(load_video, inputs=[reddit_id], outputs=[video_player])
+    with gr.Column():
+        gr.Markdown("## Sampled Frames from Video")
+        with gr.Row():
+            num_frames = gr.Slider(minimum=1, maximum=20, step=1, value=10)
+            sample_btn = gr.Button("Sample")
+
+        sampled_frames = gr.Gallery()
 
+    sample_btn.click(
+        extract_frames_moviepy,
+        inputs=[video_player, num_frames],
+        outputs=[sampled_frames],
+    )
+
+    load_btn.click(load_video, inputs=[reddit_id], outputs=[video_player])
     get_latest_pots_btn.click(get_latest_pots, outputs=[latest_post_dataframe])
     demo.load(get_latest_pots, outputs=[latest_post_dataframe])
 

packages.txt

@@ -0,0 +1 @@
+ffmpeg

requirements.txt

@@ -0,0 +1,2 @@
+ffmpy
+moviepy