Update app.py

app.py

@@ -1,9 +1,12 @@
 import streamlit as st
 import os
-import time
 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=}")
@@ -103,20 +106,26 @@ def separate_audio_by_stem(audio_path, stem_count):
             {'description': 'Other', '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("Stem count", (2,4,5))
+selected_stem_count = st.radio("Select stem count", (2,4,5))
 
 if uploaded_file is not None:
 
     if st.button("Submit"):
 
+        # 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
-            
-        separate_audios = separate_audio_by_stem(temp_file_path, selected_stem_count)
         
+        # Process the uploaded audio file
+        separate_audios = separate_audio_by_stem(temp_file_path, selected_stem_count)
+    
+        # Display the output files for download
+        st.write("Output Files:")
         for audio in separate_audios:
             st.write(audio['description'])
             st.audio(audio['path'], format="audio/wav", start_time=0)