Joel Mathew Thomas
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import subprocess
|
|
import os
|
|
import soundfile as sf
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
def separate_audio_with_demucs(input_file, output_dir: str):
|
|
"""
|
|
Use subprocess to run Demucs separation on an input audio file.
|
|
Parameters:
|
|
input_file (str): Path to the input audio file to be separated.
|
|
output_dir (str): Directory where the separated output will be saved.
|
|
"""
|
|
|
|
|
|
# Create the output directory if it does not exist
|
|
os.makedirs(output_dir, exist_ok=True)
|
|
|
|
# Demucs command to separate the temp audio file
|
|
command = [ 'demucs', '--out', output_dir, input_file]
|
|
|
|
try:
|
|
# Run the command as a subprocess
|
|
subprocess.run(command, check=True)
|
|
print(f"Separation completed. Output saved to {output_dir}")
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"An error occured while running demucs : {e}")
|
|
except FileNotFoundError:
|
|
print("Demucs is not installed or not found in the system PATH")
|
|
finally:
|
|
# Cleanup the temporary file
|
|
os.remove(input_file)
|
|
|
|
|
|
|