Spaces:
Running
Running
| import gradio as gr | |
| from scenedetect import open_video, SceneManager, split_video_ffmpeg | |
| from scenedetect.detectors import ContentDetector | |
| from scenedetect.video_splitter import split_video_ffmpeg | |
| from moviepy.video.io.VideoFileClip import VideoFileClip | |
| def find_scenes(video_path, threshold=27.0): | |
| # Open our video, create a scene manager, and add a detector. | |
| video = open_video(video_path) | |
| scene_manager = SceneManager() | |
| scene_manager.add_detector( | |
| ContentDetector(threshold=threshold)) | |
| scene_manager.detect_scenes(video, show_progress=True) | |
| scene_list = scene_manager.get_scene_list() | |
| #print(scene_list) | |
| shot_in = scene_list[0][0].get_frames() | |
| shot_out = scene_list[0][1].get_frames() | |
| #print(shot_in, shot_out) | |
| input_video_path = video_path | |
| output_video_path = 'video_out.mp4' | |
| with VideoFileClip(input_video_path) as video: | |
| new = video.subclip(shot_in, shot_out) | |
| new.write_videofile(output_video_path, audio_codec='aac') | |
| return scene_list, output_video_path | |
| video_input=gr.Video(source="upload", format="mp4", mirror_webcam="False"); | |
| gr.Interface(fn=find_scenes, inputs=video_input, outputs=["json", "video"]).launch() |