Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, vfx | |
| from moviepy.video.fx.all import blackwhite, colorx # Import specific effects from fx.all | |
| from moviepy.video.tools.drawing import color_gradient # For additional effects if needed | |
| import numpy as np | |
| # Function to adjust audio pitch | |
| def adjust_audio_pitch(input_video_path, pitch_factor): | |
| video = VideoFileClip(input_video_path) | |
| audio = video.audio | |
| if audio is None: | |
| return input_video_path # No audio to process | |
| # Adjust pitch (this is a simplified approach, more sophisticated methods may be needed) | |
| new_audio = audio.fx(vfx.pitch, pitch_factor) | |
| new_video = video.set_audio(new_audio) | |
| output_path = "temp_audio_adjusted.mp4" | |
| new_video.write_videofile(output_path, codec='libx264', audio_codec='aac') | |
| return output_path | |
| # Function to apply visual effects | |
| def apply_visual_effects(input_video_path, glitch_effect, mirror_effect, black_and_white, sepia, blur, brightness): | |
| video = VideoFileClip(input_video_path) | |
| if glitch_effect: | |
| # Simple glitch effect (you can enhance this by adding more complex glitches) | |
| def glitch_clip(clip): | |
| frames = [] | |
| for frame in clip.iter_frames(fps=clip.fps): | |
| if np.random.rand() < 0.1: # Randomly flip some frames horizontally | |
| frame = np.fliplr(frame) | |
| frames.append(frame) | |
| return clip.set_data(np.array(frames)) | |
| video = video.fl_image(lambda img: glitch_clip(video).get_frame(0)) | |
| if mirror_effect: | |
| video = video.fx(vfx.mirror_x) # Mirror reflection along X-axis | |
| if black_and_white: | |
| video = video.fx(blackwhite) | |
| if sepia: | |
| # Apply sepia effect using a custom function | |
| def sepia_effect(img): | |
| sepia_filter = np.array([[0.393, 0.769, 0.189], | |
| [0.349, 0.686, 0.168], | |
| [0.272, 0.534, 0.131]]) | |
| sepia_img = np.dot(img[...,:3], sepia_filter.T) | |
| sepia_img = np.clip(sepia_img, 0, 255).astype(np.uint8) | |
| return sepia_img | |
| video = video.fl_image(sepia_effect) | |
| if blur: | |
| video = video.fx(vfx.blur, blur_radius=5) | |
| if brightness != 0: | |
| video = video.fx(colorx, brightness) | |
| output_path = "temp_visual_effects.mp4" | |
| video.write_videofile(output_path, codec='libx264', audio_codec='aac') | |
| return output_path | |
| # Main function to handle all modifications | |
| def edit_video(input_video, pitch_factor, glitch_effect, mirror_effect, black_and_white, sepia, blur, brightness): | |
| temp_video = input_video.name if isinstance(input_video, gr.File) else input_video | |
| # Step 1: Adjust audio pitch | |
| temp_video = adjust_audio_pitch(temp_video, pitch_factor) | |
| # Step 2: Apply visual effects | |
| final_video = apply_visual_effects(temp_video, glitch_effect, mirror_effect, black_and_white, sepia, blur, brightness) | |
| return final_video | |
| # Create the Gradio Interface with a blue theme | |
| with gr.Blocks(theme="default-blue") as demo: | |
| gr.Markdown( | |
| """ | |
| # Neon Video Editor | |
| A sleek and modern video editor with advanced features. | |
| """ | |
| ) | |
| with gr.Row(): | |
| with gr.Column(): | |
| input_video = gr.File(label="Upload Video", file_types=["video"]) | |
| pitch_factor = gr.Slider(label="Audio Pitch Adjustment", minimum=-20, maximum=20, value=0, step=0.1) | |
| glitch_effect = gr.Checkbox(label="Apply Glitch Effect") | |
| mirror_effect = gr.Checkbox(label="Apply Mirror Reflection") | |
| black_and_white = gr.Checkbox(label="Apply Black & White Effect") | |
| sepia = gr.Checkbox(label="Apply Sepia Effect") | |
| blur = gr.Checkbox(label="Apply Blur Effect") | |
| brightness = gr.Slider(label="Adjust Brightness", minimum=-1, maximum=1, value=0, step=0.1) | |
| apply_button = gr.Button("Apply Changes") | |
| with gr.Column(): | |
| output_video = gr.Video(label="Edited Video Preview") | |
| apply_button.click( | |
| fn=edit_video, | |
| inputs=[input_video, pitch_factor, glitch_effect, mirror_effect, black_and_white, sepia, blur, brightness], | |
| outputs=output_video | |
| ) | |
| demo.launch() |