Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import cv2 
 
 from scenedetect import open_video, SceneManager
 from scenedetect.detectors import ContentDetector
@@ -6,13 +7,20 @@ from scenedetect.detectors import ContentDetector
 
 from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 
-outputs = ["json", "file"]
+outputs = ["json", "file", "gallery"]
 data_outputs = []
 
+
 def fn(list):
     return tuple(list);
 
-
+def save_frame(video_path, start_frame):
+    vid = cv2.VideoCapture(video_path)
+    for i in range(start_frame, 1):
+        vid.set(1, i)
+        ret, still = vid.read()
+        output_image = cv2.imwrite(f'{video_path}_frame{i}.jpg', still)
+    return output_image
 
 def find_scenes(video_path, threshold=27.0):
     # Open our video, create a scene manager, and add a detector.
@@ -28,6 +36,7 @@ def find_scenes(video_path, threshold=27.0):
     #print(scene_list)
     
     shots = []
+    images_outputs = []
     
     for i, scene in enumerate(scene_list):
         shot_in = scene[0].get_frames() / scene[0].get_framerate()
@@ -36,6 +45,9 @@ def find_scenes(video_path, threshold=27.0):
         ffmpeg_extract_subclip(video_path, shot_in, shot_out, targetname=target_name)
         data_outputs.append(target_name)
         shots.append(target_name)
+        img = save_frame(video_path, scene[0].get_frames())
+        images_outputs.append(img)
+        
         #outputs.append("video")
     #shot_in = scene_list[1][0].get_frames() / scene_list[1][0].get_framerate()
     #shot_out = (scene_list[1][1].get_frames() - 1) / scene_list[1][0].get_framerate()
@@ -45,7 +57,7 @@ def find_scenes(video_path, threshold=27.0):
     print(results)
     #ffmpeg_extract_subclip(video_path, shot_in, shot_out, targetname="cut.mp4")
     
-    return scene_list, shots
+    return scene_list, shots, images_outputs
     
 video_input=gr.Video(source="upload", format="mp4");