Update app.py

app.py

@@ -17,18 +17,22 @@ def find_scenes(video_path, threshold=27.0):
     scene_list = scene_manager.get_scene_list()
     
     #print(scene_list)
+    shots = []
     
-    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()
+    for i, scene in enumerate(scene_list):
+        shot_in = scene[0].get_frames() / scene[0].get_framerate()
+        shot_out = (scene[1].get_frames() - 1) / scene[0].get_framerate()
+        shot = ffmpeg_extract_subclip(video_path, shot_in, shot_out, targetname=i+"_cut.mp4")
+        shots.append(shot)
+    #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()
     #print(shot_in, shot_out)
     
-    #input_video_path = video_path
-    #output_video_path = 'video_out.mp4'
-    # ffmpeg_extract_subclip("full.mp4", start_seconds, end_seconds, targetname="cut.mp4")
-    ffmpeg_extract_subclip(video_path, shot_in, shot_out, targetname="cut.mp4")
-    #print(output_video_path)
-    return scene_list, "cut.mp4"
+    
+    #ffmpeg_extract_subclip(video_path, shot_in, shot_out, targetname="cut.mp4")
+    
+    return scene_list, shots
 
 video_input=gr.Video(source="upload", format="mp4");
 
-gr.Interface(fn=find_scenes, inputs=video_input, outputs=["json", "video" * 2]).launch()
+gr.Interface(fn=find_scenes, inputs=video_input, outputs=["json", "files"]).launch()