Spaces:
Runtime error
Runtime error
File size: 2,343 Bytes
c56a4ac b8e4d8f c56a4ac b8e4d8f fb25e22 c56a4ac b8e4d8f fb25e22 b8e4d8f fb25e22 b8e4d8f fb25e22 b8e4d8f c56a4ac cf9f7d3 b37ad8a c56a4ac cf9f7d3 c56a4ac b8e4d8f c56a4ac cf9f7d3 c56a4ac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import gradio as gr
import os
import random
import subprocess
def separate_audio(audio_path, stem_count):
print(f"{audio_path=}")
head, tail = os.path.split(audio_path)
gradio_temp_path = head
audio_filename = tail.split('.')[0]
print(f"{gradio_temp_path=}")
print(f"{audio_filename=}")
print(f"{stem_count=}")
# Construct the spleeter command
command = f"spleeter separate -p spleeter:{stem_count}stems -o {gradio_temp_path} {audio_path}"
command = command.split()
print(f"{command=}")
# Run the spleeter command and capture the output
result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
print(f"stdout: {result.stdout.decode('utf-8')}")
print(f"stderr: {result.stderr.decode('utf-8')}")
# Check if the command executed successfully
if result.returncode != 0:
print(f"Error executing command: {result.returncode}")
return []
# Generate a random number for unique folder naming
randomnumber = str(random.randint(111111111, 999999999))
paths = []
# Define the output paths for the separated stems
if stem_count == 2:
paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/accompaniment.wav"))
paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/vocals.wav"))
elif stem_count == 4 or stem_count == 5:
paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/vocals.wav"))
paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/drums.wav"))
paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/bass.wav"))
paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/other.wav"))
if stem_count == 5:
paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/piano.wav"))
# Return the audio file paths as a list of Audio components
return [gr.Audio(path) for path in paths]
# Define the Gradio interface
iface = gr.Interface(
fn=separate_audio,
inputs=[gr.Audio(type="filepath"), gr.Radio([2, 4, 5])],
outputs=[gr.Audio() for _ in range(max_stem_count)] # Adjust the range according to the max number of stems
)
# Launch the interface
iface.launch()
|