api to get frames from video

app.py

@@ -1,4 +1,4 @@
-from main import run_app, run_train, run_inference, run_generate_frame, run_interpolate_frames, resize_images
+from main import run_app, run_train, run_inference, run_generate_frame, run_interpolate_frames, resize_images, get_frames
 from evaluate import run_evaluate, get_score
 
 import spaces
@@ -34,6 +34,7 @@ with gr.Blocks() as demo:
             eval_btn = gr.Button(value="Evaluate All")
             eval_btn2 = gr.Button(value="Evaluate")
             resize_btn = gr.Button(value="Resize")
+            frames_btn = gr.Button(value="Get Frames")
         with gr.Column():
             animation = gr.Video(label="Result")
             frames = gr.Gallery(type="pil", label="Frames", format="png")
@@ -73,6 +74,10 @@ with gr.Blocks() as demo:
     resize_btn.click(
         resize_images, inputs=[frame_imgs, img_width, img_height], outputs=[frames]
     )
+
+    frames_btn.click(
+        resize_images, inputs=[mocap, fps], outputs=[frames]
+    )
     
 
 demo.launch(share=True, show_error=True)

main.py

@@ -1342,7 +1342,11 @@ def resize_images(images, width, height):
     images = [img[0] for img in images]
     return [resize_pad(img, width, height, True) for img in images]
     
-
+def get_frames(video_path, fps):
+    frames = extract_frames(video_path, fps)
+    for f in frames:
+        f.thumbnail((512,512))
+    return frames