Spaces:
Running
Running
| import gradio as gr | |
| from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip | |
| from moviepy.video.fx.all import speedx, rotate, multiply_brightness, colorx | |
| from moviepy.video.fx.all import lum_contrast | |
| def video_editor(input_video, start_time, end_time, playback_speed, create_gif, | |
| rotation_degrees, watermark_text, brightness_factor, | |
| contrast, saturation): | |
| # Load video | |
| clip = VideoFileClip(input_video).subclip(start_time, end_time) | |
| # Change playback speed if applicable | |
| if playback_speed != 1: | |
| clip = speedx(clip, factor=playback_speed) | |
| # Rotate the video if applicable | |
| if rotation_degrees != 0: | |
| clip = rotate(clip, rotation_degrees) | |
| # Adjust brightness if applicable | |
| if brightness_factor != 1: | |
| clip = multiply_brightness(clip, brightness_factor) | |
| # Adjust contrast if applicable | |
| if contrast != 1: | |
| clip = lum_contrast(clip, contrast=contrast - 1) | |
| # Adjust color saturation if applicable | |
| if saturation != 1: | |
| clip = colorx(clip, factor=saturation) | |
| # Add text watermark if applicable | |
| if watermark_text: | |
| txt_clip = TextClip(watermark_text, fontsize=24, color='white', bg_color='black') | |
| txt_clip = txt_clip.set_position(('right', 'bottom')).set_duration(clip.duration) | |
| clip = CompositeVideoClip([clip, txt_clip]) | |
| # Define output paths | |
| output_video_path = "edited_video.mp4" | |
| output_gif_path = "edited_video.gif" | |
| if not create_gif: | |
| # Write video file with the edits | |
| clip.write_videofile(output_video_path, codec="libx264", audio_codec="aac") | |
| return output_video_path | |
| else: | |
| # Write GIF file with the edits | |
| clip.write_gif(output_gif_path) | |
| return output_gif_path | |
| iface = gr.Interface( | |
| fn=video_editor, | |
| inputs=[ | |
| gr.Video(source="upload", type="file", label="Input Video"), | |
| gr.Textbox(value="00:00:00", label="Start Time (HH:MM:SS)"), | |
| gr.Textbox(value="00:00:10", label="End Time (HH:MM:SS)"), | |
| gr.Slider(minimum=0.25, maximum=4.0, value=1, step=0.25, label="Playback Speed"), | |
| gr.Checkbox(label="Create GIF instead of video", value=False), | |
| gr.Slider(minimum=0, maximum=360, value=0, step=90, label="Rotation (Degrees)"), | |
| gr.Textbox(value="", label="Watermark Text"), | |
| gr.Slider(minimum=0.5, maximum=2.0, value=1, step=0.1, label="Brightness Factor"), | |
| gr.Slider(minimum=0.1, maximum=2.0, value=1, step=0.1, label="Contrast"), | |
| gr.Slider(minimum=0, maximum=2.0, value=1, step=0.1, label="Saturation") | |
| ], | |
| outputs=gr.Video(label="Output Video/GIF"), | |
| examples=[ | |
| ["sample_video.mp4", "00:00:02", "00:00:05", 1, False, 0, "", 1, 1, 1], | |
| ] | |
| ) | |
| iface.launch() |