Update app.py

app.py

@@ -256,15 +256,15 @@ def run(audio_files, file_m, file_index):
 
 def process_audio(audio_file, uploaded_files, file_m, file_index):
     if audio_file is not None:
-        result, first_audio = run([audio_file], file_m, file_index)
+        result, _ = run([audio_file], file_m, file_index)
     elif uploaded_files is not None:
-        result, first_audio = run(uploaded_files, file_m, file_index)
+        result, _ = run(uploaded_files, file_m, file_index)
     
-    # Ensure first_audio is correctly formatted as a valid file path
-    if not os.path.exists(first_audio):
-        raise ValueError(f"Processed audio file not found at: {first_audio}")
-    
-    return result, gr.update(value=first_audio, visible=True)
+    # Return the list of processed audio file paths and the first file for initial display
+    return result, result[0], gr.update(choices=result, visible=True)
+
+def update_audio_selection(selected_file):
+    return gr.update(value=selected_file, visible=True)
 
 
 def switch_input(input_type):
@@ -324,7 +324,11 @@ def get_gui(theme):
         model = model_conf()
         indx = index_conf()
         button_base = button_conf()
-        output_file, output_audio = output_conf()
+
+        dropdown = gr.Dropdown(choices=[], label="Select Processed Audio", visible=False)
+        output_audio = gr.Audio(label="Play Selected Audio", visible=False)
+        
+        # output_file, output_audio = output_conf()
 
         button_base.click(
             process_audio,
@@ -332,6 +336,12 @@ def get_gui(theme):
             outputs=[output_file, output_audio],
         )
 
+        dropdown.change(
+        update_audio_selection,
+        inputs=[dropdown],
+        outputs=output_audio,
+    )
+
         # gr.Examples(
         #     examples=[
         #         ["./test.ogg", "./model.pth", "./model.index"],