Update app.py

app.py

@@ -56,12 +56,14 @@ def separate_audio_by_stem(audio_path, stem_count):
     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]
+
+        print(f"{accompaniment_path=} \t exists: {os.path.exists(accompaniment_path)}")
+        print(f"{vocals_path=} \t exists: {os.path.exists(vocals_path)}")
+        
+        return [
+            {'description':, 'Accompaniment', 'path':accompaniment_path},
+            {'description':, 'Vocals', 'path':vocals_path},
+        ]
     
     elif stem_count == 4:
         
@@ -75,12 +77,15 @@ def separate_audio_by_stem(audio_path, stem_count):
         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]
+        return [
+            {'description':, 'Vocals', 'path':vocals_path},
+            {'description':, 'Drums', 'path':drums_path},
+            {'description':, 'Bass', 'path':bass_path},
+            {'description':, 'Other', '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"
@@ -93,8 +98,13 @@ def separate_audio_by_stem(audio_path, stem_count):
         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]
+        return [
+            {'description':, 'Vocals', 'path':vocals_path},
+            {'description':, 'Piano', 'path':piano_path},
+            {'description':, 'Drums', 'path':drums_path},
+            {'description':, 'Bass', 'path':bass_path},
+            {'description':, 'Other', 'path':other_path},
+        ]
     
 # Streamlit app content
 st.write("Upload an audio file")
@@ -110,9 +120,10 @@ 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, selected_stem_count)
+    separate_audios = 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)
+    for audio in separate_audios:
+        st.write(audio.description)
+        st.audio(audio.path, format="audio/wav", start_time=0)