Update app.py

app.py

@@ -1,59 +1,72 @@
 import gradio as gr
-from moviepy.editor import *
-from PIL import ImageFont, ImageDraw, Image
-
-def edit_video(video_file, name, author, font, filter):
-    # Load the video
-    clip = VideoFileClip(video_file)
-
-    # Apply the chosen filter
-    clip = eval(f"clip.{filter}()")
-
-    # Load the font based on user input
-    font_path = f"{font}.ttf"
-    font_size = clip.size[0] // 10
-    font = ImageFont.truetype(font_path, font_size)
-
-    # Create an image with the author text
-    author_text = f"Author: {author}"
-    author_image = Image.new('RGB', (clip.w // 2, clip.h // 8), (0, 0, 0))
-    author_draw = ImageDraw.Draw(author_image)
-    author_draw.text((10, 10), author_text, font=font, fill=(255, 255, 255))
-
-    # Create an image with the bigger text
-    bigger_text = f"Author: {author}"
-    bigger_image = Image.new('RGB', (clip.w // 2, clip.h // 4), (0, 0, 0))
-    bigger_draw = ImageDraw.Draw(bigger_image)
-    bigger_draw.text((10, 10), bigger_text, font=font, fill=(255, 255, 255))
-
-    # Create a composite video with the texts
-    text_clip1 = ImageClip(author_image).set_duration(4).set_position(("right", "bottom"))
-    text_clip2 = ImageClip(bigger_image).set_duration(4).set_position(("right", "bottom"))
-    composite_clip = CompositeVideoClip([clip, text_clip1, text_clip2])
-
-    # Save the edited video
-    output_file = f"{name}_edited.mp4"
-    composite_clip.write_videofile(output_file, codec='libx264')
-
-    return output_file
-
-def video_editor(name, author, font, filter, video_file):
-    edited_video = edit_video(video_file, name, author, font, filter)
-    return edited_video
-
-# Define the inputs and outputs for Gradio app
-inputs = [
-    gr.Textbox(label="Video Name"),
-    gr.Textbox(label="Author Name"),
-    gr.Radio(["Arial", "Barbie", "MrBeast", "OptiAgle"], label="Font"),
-    gr.Radio(["blur", "rotate", "resize", "mirror"], label="Filter"),
-    gr.inputs.File(label="Video File", type="file")
-]
-
-outputs = gr.outputs.File(label="Edited Video")
-
-# Create a Gradio app interface
-grapp = gr.Interface(fn=video_editor, inputs=inputs, outputs=outputs, capture_session=True)
-
-# Run the Gradio app
-grapp.launch()
+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()