|
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): |
|
|
|
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() |
|
|
|
|
|
|
|
shot_in = scene_list[0][0].get_frames() |
|
shot_out = scene_list[0][1].get_frames() |
|
|
|
|
|
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() |