Update app.py

app.py

@@ -2,68 +2,54 @@ 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")
+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 = []
 
-    # 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)}")
+    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 {input_path} -i {logo_path} -filter_complex 'overlay=10:10' -codec:a copy {watermarked_path}",
-            shell=True
+            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
+        # Add outro by concatenating video and 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
+            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)
 
-        # Cleanup input and intermediate file
-        os.remove(input_path)
+        # Clean intermediate watermarked file
         os.remove(watermarked_path)
 
-    # Create ZIP archive of outputs
+    # 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)
+            zipf.write(filepath, os.path.basename(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"),
-        gr.File(label="Upload Logo (PNG)", file_types=[".png"]),
-        gr.File(label="Upload Outro Video (MP4)", file_types=[".mp4"])
+        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 Processed Videos ZIP"),
+    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."
 )