Additional formats

nomadkaraoke · Jan 15, 2025 · e216da1 · e216da1
1 parent 02749be
commit e216da1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/audio_separator/separator/separator.py b/audio_separator/separator/separator.py
@@ -757,7 +757,7 @@ def separate(self, audio_file_path, custom_output_names=None):
                 for root, dirs, files in os.walk(path):
                     for file in files:
                         # Check the file extension to ensure it's an audio file
-                        if file.endswith(('.wav', '.mp3', '.flac', '.ogg', '.m4a')):  # Add other formats if needed
+                        if file.endswith((".wav", ".flac", ".mp3", ".ogg", ".opus", ".m4a", ".aiff", ".ac3")):  # Add other formats if needed
                             full_path = os.path.join(root, file)
                             self.logger.info(f"Processing file: {full_path}")
                             try:

diff --git a/audio_separator/utils/cli.py b/audio_separator/utils/cli.py
@@ -192,7 +192,7 @@ def main():
             for root, dirs, files in os.walk(path):
                 for file in files:
                     # Check the file extension to ensure it's an audio file
-                    if file.endswith(('.wav', '.mp3', '.flac', '.ogg', '.m4a')):  # Add other formats if needed
+                    if file.endswith((".wav", ".flac", ".mp3", ".ogg", ".opus", ".m4a", ".aiff", ".ac3")):  # Add other formats if needed
                         audio_files.append(os.path.join(root, file))
         else:
             # If the path is a file, add it to the list