added ffmeg functionality

app.py

@@ -3,6 +3,7 @@ import os
 import cv2
 import numpy as np
 import shutil
+import subprocess
 from SinglePhoto import FaceSwapper
 
 wellcomingMessage = """
@@ -49,6 +50,7 @@ def swap_video(src_img, src_idx, video, dst_idx):
     frames_dir = "VideoSwapping/video_frames"
     swapped_dir = "VideoSwapping/swapped_frames"
     output_video_path = "VideoSwapping/output_tmp_output_video.mp4"
+    final_output_path = "VideoSwapping/output_with_audio.mp4"
 
     os.makedirs(os.path.dirname(src_path), exist_ok=True)
     os.makedirs(os.path.dirname(dst_video_path), exist_ok=True)
@@ -83,6 +85,14 @@ def swap_video(src_img, src_idx, video, dst_idx):
     frames_to_video(swapped_dir, output_video_path, fps)
     log += f"Combined swapped frames into video {output_video_path}\n"
 
+    # Add audio from original video
+    ok, audio_log = add_audio_to_video(dst_video_path, output_video_path, final_output_path)
+    if ok:
+        log += f"Added audio to {final_output_path}\n"
+    else:
+        log += f"Audio muxing failed: {audio_log}\n"
+        final_output_path = output_video_path  # fallback to video without audio
+
     try:
         if os.path.exists(src_path):
             os.remove(src_path)
@@ -96,7 +106,29 @@ def swap_video(src_img, src_idx, video, dst_idx):
     except Exception as cleanup_error:
         log += f"Cleanup error: {cleanup_error}\n"
 
-    return output_video_path, log
+    return final_output_path, log
+
+def add_audio_to_video(original_video_path, video_no_audio_path, output_path):
+    """
+    Uses ffmpeg to mux audio from original_video_path into video_no_audio_path.
+    """
+    cmd = [
+        "ffmpeg",
+        "-y",
+        "-i", video_no_audio_path,
+        "-i", original_video_path,
+        "-c:v", "copy",
+        "-c:a", "aac",
+        "-map", "0:v:0",
+        "-map", "1:a:0?",
+        "-shortest",
+        output_path
+    ]
+    try:
+        subprocess.run(cmd, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        return True, ""
+    except subprocess.CalledProcessError as e:
+        return False, e.stderr.decode()
 
 def swap_multi_src_single_dst(src_imgs, dst_img, dst_idx):
     log = ""

packages.txt

@@ -0,0 +1 @@
+ffmpeg