Update app.py

app.py

@@ -78,7 +78,23 @@ def separate_audio_by_stem(audio_path, stem_count):
     
         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")
@@ -94,7 +110,7 @@ if uploaded_file is not None:
         temp_file_path = temp_file.name
     
     # Process the uploaded audio file
-    separated_audio_paths = separate_audio_by_stem(temp_file_path)
+    separated_audio_paths = separate_audio_by_stem(temp_file_path, selected_stem_count)
 
     # Display the output files for download
     st.write("Output Files:")