Update app.py

app.py

@@ -4,7 +4,7 @@ import tempfile
 import subprocess
 
 # Set Streamlit app title
-st.title("Audio Processing App")
+st.title("Audio Separation App")
 
 # Function to process the audio file
 def separate_audio(audio_path):
@@ -17,10 +17,7 @@ def separate_audio(audio_path):
     print(f"{gradio_temp_path=}")
     print(f"{audio_filename=}")
     
-    # command = f"spleeter separate -o {gradio_temp_path} -p spleeter:2stems {audio_path}"
     command = f"spleeter separate -p spleeter:2stems {audio_path}"
-    # command = f"ls {gradio_temp_path}"
-    # command = f"cp {audio_path} output/test.wav"
     command = command.split()
     print(f"{command=}")
 
@@ -37,22 +34,71 @@ def separate_audio(audio_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]
+        
+
+    
 # Streamlit app content
-st.write("Upload an audio file (.wav)")
+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,3,5))
 
 if uploaded_file is not None:
 
     # Save the uploaded file to a temporary location
-    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+    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
-    vocals_path, accompaniment_path = separate_audio(temp_file_path)
+    separated_audio_paths = separate_audio(temp_file_path)
 
     # Display the output files for download
     st.write("Output Files:")
-    st.audio(vocals_path, format="audio/wav", start_time=0)
-    st.audio(accompaniment_path, format="audio/wav", start_time=0)
+    # st.audio(vocals_path, format="audio/wav", start_time=0)
+    # st.audio(accompaniment_path, format="audio/wav", start_time=0)
+    for path in separated_audio_paths:
+        st.audio(path, format="audio/wav", start_time=0)