Update app.py

app.py

@@ -49,18 +49,19 @@ st.subheader('Upload your audio file to process with AudioLDM2 or StableAudio')
 
 uploaded_file = st.file_uploader("Choose an audio file", type=["wav", "mp3", "ogg"])
 model_choice = st.selectbox('Choose a model for processing', ('AudioLDM2', 'StableAudio'))
+prompt_text = st.text_input('Enter a prompt for the audio processing')
 
-if uploaded_file is not None:
+if uploaded_file is not None and prompt_text:
     st.audio(uploaded_file, format="audio/mpeg")
 
     # Dummy processing function to demonstrate functionality
-    def process_audio(file, model):
+    def process_audio(file, model, prompt):
         # Replace this with actual audio processing code
         processed_audio_path = "processed_audio.wav"
         return processed_audio_path
 
     if st.button('Process Audio'):
-        processed_audio = process_audio(uploaded_file, model_choice)
+        processed_audio = process_audio(uploaded_file, model_choice, prompt_text)
         st.audio(processed_audio, format="audio/mpeg", loop=False)
         st.download_button(label="Download Processed Audio", data=processed_audio, file_name="processed_audio.wav", mime="audio/wav")