Spaces:
Runtime error
Runtime error
File size: 4,326 Bytes
e62200f 8482b0b e62200f 8482b0b 3e5fbbc 8482b0b b5fa5cc 8482b0b 4feb81f 8482b0b 3e5fbbc f94e46a 3e5fbbc f94e46a 3e5fbbc 8482b0b 3e5fbbc 8482b0b bd1f851 5f6adf7 8482b0b 20d74ef 3e5fbbc 20d74ef 8482b0b f94e46a 8482b0b 3e5fbbc |
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 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 |
import streamlit as st
import os
import tempfile
import subprocess
# Set Streamlit app title
st.title("Audio Separation App")
# Function to process the audio file
def separate_audio(audio_path):
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=}")
command = f"spleeter separate -p spleeter:2stems {audio_path}"
command = command.split()
print(f"{command=}")
result = subprocess.run(command)
print(result)
print("--------")
accompaniment_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/accompaniment.wav"
vocals_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"
print(f"{accompaniment_path=}")
print(os.path.exists(accompaniment_path))
print(f"{vocals_path=}")
print(os.path.exists(vocals_path))
return vocals_path, accompaniment_path
def separate_audio_by_stem(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=}")
command = f"spleeter separate -p spleeter:{stem_count}stems {audio_path}"
command = command.split()
print(f"{command=}")
result = subprocess.run(command)
print(result)
if stem_count == 2:
accompaniment_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/accompaniment.wav"
vocals_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"
print(f"{accompaniment_path=}")
print(os.path.exists(accompaniment_path))
print(f"{vocals_path=}")
print(os.path.exists(vocals_path))
return [vocals_path, accompaniment_path]
elif stem_count == 4:
vocals_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"
drums_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/drums.wav"
bass_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/bass.wav"
other_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/other.wav"
print(f"{vocals_path=} \t exists: {os.path.exists(vocals_path)}")
print(f"{drums_path=} \t exists: {os.path.exists(drums_path)}")
print(f"{bass_path=} \t exists: {os.path.exists(bass_path)}")
print(f"{other_path=} \t exists: {os.path.exists(other_path)}")
return [vocals_path, drums_path, bass_path, other_path]
elif stem_count == 5:
piano_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/piano.wav"
vocals_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"
drums_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/drums.wav"
bass_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/bass.wav"
other_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/other.wav"
print(f"{piano_path=} \t exists: {os.path.exists(vocals_path)}")
print(f"{vocals_path=} \t exists: {os.path.exists(vocals_path)}")
print(f"{drums_path=} \t exists: {os.path.exists(drums_path)}")
print(f"{bass_path=} \t exists: {os.path.exists(bass_path)}")
print(f"{other_path=} \t exists: {os.path.exists(other_path)}")
return [piano_path, vocals_path, drums_path, bass_path, other_path]
# Streamlit app content
st.write("Upload an audio file")
uploaded_file = st.file_uploader("Choose a file", type=["wav","mp3"])
selected_stem_count = st.radio("Select stem count", (2,4,5))
if uploaded_file is not None:
# Save the uploaded file to a temporary location
with tempfile.NamedTemporaryFile(delete=False) as temp_file:
temp_file.write(uploaded_file.read())
temp_file_path = temp_file.name
# Process the uploaded audio file
separated_audio_paths = separate_audio_by_stem(temp_file_path, selected_stem_count)
# Display the output files for download
st.write("Output Files:")
for path in separated_audio_paths:
st.audio(path, format="audio/wav", start_time=0)
|