Update app.py

app.py

@@ -3,10 +3,6 @@ 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=}")
@@ -106,26 +102,24 @@ 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("Select stem count", (2,4,5))
+uploaded_file = st.file_uploader("", type=["wav","mp3"])
+selected_stem_count = 2
 
 if uploaded_file is not None:
 
-    if st.button("Submit"):
+    #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
+    # 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
-        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)
+    # 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)