Create app.py

app.py

@@ -0,0 +1,72 @@
+import gradio as gr
+import subprocess
+import os
+import zipfile
+import shutil
+
+def process_videos(video_file_list, logo_file, outro_file):
+    if not os.path.exists("outputs"):
+        os.makedirs("outputs")
+    output_paths = []
+
+    # Save logo and outro files to disk
+    logo_path = "logo.png"
+    outro_path = "outro.mp4"
+    with open(logo_path, "wb") as f:
+        f.write(logo_file.read())
+    with open(outro_path, "wb") as f:
+        f.write(outro_file.read())
+
+    for video in video_file_list:
+        input_path = video.name
+        with open(input_path, "wb") as f:
+            f.write(video.read())
+
+        watermarked_path = os.path.join("outputs", f"watermarked_{os.path.basename(input_path)}")
+        final_path = os.path.join("outputs", f"final_{os.path.basename(input_path)}")
+
+        # Add watermark using FFmpeg
+        subprocess.run(
+            f"ffmpeg -y -i {input_path} -i {logo_path} -filter_complex 'overlay=10:10' -codec:a copy {watermarked_path}",
+            shell=True
+        )
+
+        # Add outro
+        subprocess.run(
+            f"ffmpeg -y -i {watermarked_path} -i {outro_path} -filter_complex '[0:v][1:v]concat=n=2:v=1:a=0[outv]' -map '[outv]' -c:v libx264 {final_path}",
+            shell=True
+        )
+
+        output_paths.append(final_path)
+
+        # Cleanup input and intermediate file
+        os.remove(input_path)
+        os.remove(watermarked_path)
+
+    # Create ZIP archive of outputs
+    zip_filename = "processed_videos.zip"
+    with zipfile.ZipFile(zip_filename, 'w') as zipf:
+        for filepath in output_paths:
+            zipf.write(filepath)
+            os.remove(filepath)  # Remove after adding to ZIP
+
+    # Cleanup logo and outro files
+    os.remove(logo_path)
+    os.remove(outro_path)
+
+    return zip_filename
+
+demo = gr.Interface(
+    fn=process_videos,
+    inputs=[
+        gr.Files(label="Upload Videos", file_types=[".mp4"], type="file", multiple=True),
+        gr.File(label="Upload Logo (PNG)", file_types=[".png"]),
+        gr.File(label="Upload Outro Video (MP4)", file_types=[".mp4"])
+    ],
+    outputs=gr.File(label="Download Processed Videos ZIP"),
+    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()