Update app.py

app.py

@@ -8,20 +8,14 @@ from scenedetect.detectors import ContentDetector
 
 from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 
-outputs = ["json", "file", "gallery"]
+outputs = ["json", "file"]
 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.
@@ -37,7 +31,7 @@ def find_scenes(video_path, threshold=27.0):
     #print(scene_list)
     
     shots = []
-    images_outputs = []
+    stills = []
     
     for i, scene in enumerate(scene_list):
         shot_in = scene[0].get_frames() / scene[0].get_framerate()
@@ -47,12 +41,21 @@ def find_scenes(video_path, threshold=27.0):
         data_outputs.append(target_name)
         shots.append(target_name)
         
-        vid = cv2.VideoCapture(video_path)
-        for i in range(scene[0].get_frames(), 1):
-            vid.set(1, i)
-            ret, still = vid.read()
-            output_image = cv2.imwrite(f'{video_path}_frame{i}.jpg', still)
-            images_outputs.append(output_image)
+        video = cv2.VideoCapture(video_path)
+
+        fps = video.get(cv2.CAP_PROP_FPS)
+        print('frames per second =',fps)
+
+        frame_id = scene[0].get_frames()
+
+        video.set(cv2.CAP_PROP_POS_FRAMES, frame_id)
+        ret, frame = video.read()
+
+        # Display and save frame
+        #cv2.imshow('frame', frame); cv2.waitKey(0)
+
+        img = cv2.imwrite(str(frame_id) + '_screenshot.png',frame)
+        stills.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()
@@ -60,10 +63,10 @@ def find_scenes(video_path, threshold=27.0):
     
     results = fn(data_outputs)
     print(results)
-    print(images_outputs)
+    print(stills)
     #ffmpeg_extract_subclip(video_path, shot_in, shot_out, targetname="cut.mp4")
     
-    return scene_list, shots, images_outputs
+    return scene_list, shots
     
 video_input=gr.Video(source="upload", format="mp4");