Update app.py

app.py

@@ -1,58 +1,48 @@
 import gradio as gr
-import subprocess
-import os
-import zipfile
-
-def process_videos(video_filepaths, logo_path, outro_path):
-    output_dir = "outputs"
-    if not os.path.exists(output_dir):
-        os.makedirs(output_dir)
-    output_paths = []
-
-    for video_path in video_filepaths:
-        basename = os.path.basename(video_path)
-        watermarked_path = os.path.join(output_dir, f"watermarked_{basename}")
-        final_path = os.path.join(output_dir, f"final_{basename}")
-
-        # Add watermark using FFmpeg
-        subprocess.run(
-            f"ffmpeg -y -i \"{video_path}\" -i \"{logo_path}\" -filter_complex \"overlay=10:10\" -codec:a copy \"{watermarked_path}\"",
-            shell=True,
-            check=True
-        )
-
-        # Add outro by concatenating video and outro
-        subprocess.run(
-            f"ffmpeg -y -i \"{watermarked_path}\" -i \"{outro_path}\" -filter_complex \"[0:v:0][0:a:0][1:v:0][1:a:0]concat=n=2:v=1:a=1[outv][outa]\" -map \"[outv]\" -map \"[outa]\" \"{final_path}\"",
-            shell=True,
-            check=True
-        )
-
-        output_paths.append(final_path)
-
-        # Clean intermediate watermarked file
-        os.remove(watermarked_path)
-
-    # Zip all processed videos
-    zip_filename = "processed_videos.zip"
-    with zipfile.ZipFile(zip_filename, 'w') as zipf:
-        for filepath in output_paths:
-            zipf.write(filepath, os.path.basename(filepath))
-            os.remove(filepath)  # Remove after adding to ZIP
-
-    return zip_filename
-
-demo = gr.Interface(
-    fn=process_videos,
-    inputs=[
-        gr.Files(label="Upload Videos (MP4)", file_types=[".mp4"], type="filepaths"),
-        gr.File(label="Upload Logo (PNG)", file_types=[".png"], type="filepath"),
-        gr.File(label="Upload Outro Video (MP4)", file_types=[".mp4"], type="filepath")
-    ],
-    outputs=gr.File(label="Download ZIP of Processed Videos"),
-    title="Bulk Video Watermark and Outro Adder",
-    description="Upload multiple videos, a logo image, and an outro video to add watermark and append outro in bulk."
-)
-
-if __name__ == "__main__":
-    demo.launch()
+from moviepy.editor import *
+
+def process_video(video_path, logo_path, outro_path, logo_position, logo_size):
+    # Load the video, logo, and outro
+    main_video = VideoFileClip(video_path)
+    logo = (ImageClip(logo_path)
+            .set_duration(main_video.duration)
+            .resize(height=int(logo_size))
+            .margin(right=8, top=8, opacity=0)
+            .set_pos((logo_position)))
+            
+    outro_video = VideoFileClip(outro_path)
+
+    # Add the watermark to the main video
+    final_video = CompositeVideoClip([main_video, logo])
+
+    # Concatenate the watermarked video with the outro
+    final_video = concatenate_videoclips([final_video, outro_video])
+
+    # Save the final video to a temporary file
+    output_path = "final_video.mp4"
+    final_video.write_videofile(output_path, codec="libx264")
+
+    return output_path
+
+# Define the Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("## Video Watermarker and Outro Adder")
+    with gr.Row():
+        with gr.Column():
+            video_input = gr.Video(label="Upload Your Video")
+            logo_input = gr.Image(label="Upload Your Logo", type="filepath")
+            outro_input = gr.Video(label="Upload Your Outro")
+            logo_position = gr.Dropdown(label="Logo Position", choices=["left", "center", "right"], value="right")
+            logo_size = gr.Slider(label="Logo Size", minimum=50, maximum=500, step=10, value=100)
+            process_button = gr.Button("Process Video")
+        with gr.Column():
+            video_output = gr.Video(label="Processed Video")
+
+    process_button.click(
+        fn=process_video,
+        inputs=[video_input, logo_input, outro_input, logo_position, logo_size],
+        outputs=video_output
+    )
+
+demo.launch()
+