Use lossless ffmpeg copy when trimming inputs

gradio_app.py

@@ -6,6 +6,7 @@ import datetime
 import shutil
 from pathlib import Path
 from moviepy.editor import VideoFileClip
+import subprocess
 import gradio as gr
 
 # Download Weights
@@ -64,10 +65,29 @@ propainter = Propainter(propainter_model_dir, device=device)
 # Helper function to trim videos
 def trim_video(input_path, output_path, max_duration=10):
     clip = VideoFileClip(input_path)
-    trimmed_clip = clip.subclip(0, min(max_duration, clip.duration))
-    trimmed_clip.write_videofile(output_path, codec="libx264", audio_codec="aac")
+    duration = min(max_duration, clip.duration)
     clip.close()
-    trimmed_clip.close()
+
+    # Preserve original encoding to avoid reintroducing H.264 macroblocking artefacts
+    subprocess.run(
+        [
+            "ffmpeg",
+            "-hide_banner",
+            "-loglevel",
+            "error",
+            "-y",
+            "-ss",
+            "0",
+            "-i",
+            input_path,
+            "-t",
+            f"{duration:.6f}",
+            "-c",
+            "copy",
+            output_path,
+        ],
+        check=True,
+    )
 
 @spaces.GPU(duration=1200)
 def infer(input_video, input_mask):