From 069db4061b8030a82dcdee5b119f9bdc68b97479 Mon Sep 17 00:00:00 2001
From: Fabrice Quenneville <fabquenneville@gmail.com>
Date: Wed, 23 Oct 2024 10:10:22 -0400
Subject: [PATCH] Add script to remove audio tracks from video files using
 ffmpeg

- Published a Python script that removes specified audio tracks from video files.
- Supports individual files and recursive directory processing.
- Added argument parsing with options for specifying audio track, file, and directory.
- Implemented color-coded output for clearer feedback.
- Integrated argcomplete for command-line autocompletion.
---
 scripts/video_remove_audio.py | 95 +++++++++++++++++++++++++++++++++++
 1 file changed, 95 insertions(+)
 create mode 100755 scripts/video_remove_audio.py

diff --git a/scripts/video_remove_audio.py b/scripts/video_remove_audio.py
new file mode 100755
index 0000000..67651d5
--- /dev/null
+++ b/scripts/video_remove_audio.py
@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+
+import os
+import argparse
+import argcomplete
+import colorama
+import subprocess
+
+colorama.init()
+
+creset = colorama.Fore.RESET
+ccyan = colorama.Fore.CYAN
+cyellow = colorama.Fore.YELLOW
+cgreen = colorama.Fore.GREEN
+cred = colorama.Fore.RED
+
+
+# Function to remove audio track from a video using ffmpeg
+def remove_audio_track(file_path, track):
+    print(f"{cgreen}Processing file: {file_path}{creset}")
+    output_file = f"{os.path.splitext(file_path)[0]}_no_audio{os.path.splitext(file_path)[1]}"
+    # ffmpeg command to remove the specified audio track and keep other streams
+    command = [
+        "ffmpeg", "-i", file_path, "-map", "0", "-map", f"-0:a:{track}", "-c",
+        "copy", output_file
+    ]
+
+    try:
+        result = subprocess.run(command,
+                                stdout=subprocess.PIPE,
+                                stderr=subprocess.PIPE,
+                                text=True)
+
+        if result.returncode == 0:
+            print(
+                f"{ccyan}Audio track {track} removed. Output saved to {output_file}{creset}\n"
+            )
+        else:
+            print(
+                f"{cred}Command failed with return code {result.returncode}{creset}\n"
+            )
+            print(f"{cred}Error output:", result.stderr, f"{creset}\n")
+
+    except subprocess.CalledProcessError as e:
+        print(f"{cred}Error removing audio track: {e}{creset}\n")
+
+
+# Function to recursively process videos in a directory
+def process_directory(dir_path, track):
+    for root, _, files in os.walk(dir_path):
+        for file in files:
+            if file.endswith((".mp4", ".mkv", ".avi",
+                              ".mov")):  # Add more formats as needed
+                file_path = os.path.join(root, file)
+                remove_audio_track(file_path, track)
+
+
+def main():
+    # Set up argument parser
+    parser = argparse.ArgumentParser(
+        description="Remove audio track from video files.")
+    parser.add_argument("--track",
+                        type=int,
+                        default=0,
+                        help="Audio track index to remove (default is 0).")
+    parser.add_argument("--file",
+                        type=str,
+                        help="Path to a specific video file.")
+    parser.add_argument(
+        "--dir",
+        type=str,
+        default=os.getcwd(),
+        help="Directory to process (default is current directory).")
+
+    # Enable autocomplete for argparse
+    argcomplete.autocomplete(parser)
+
+    args = parser.parse_args()
+
+    # Process single file if provided
+    if args.file:
+        if os.path.isfile(args.file):
+            remove_audio_track(args.file, args.track)
+        else:
+            print(f"File {args.file} does not exist.")
+    # Otherwise, process all files in the specified directory
+    else:
+        if os.path.isdir(args.dir):
+            process_directory(args.dir, args.track)
+        else:
+            print(f"Directory {args.dir} does not exist.")
+
+
+if __name__ == "__main__":
+    main()