11

app.py

@@ -176,6 +176,18 @@ def process_audio(test_file):
     return audio1, audio2
 
 
+# List of demo audio files
+demo_audio_files = [
+    ("Demo Audio 1", "demo/item0_mix.wav"),
+    ("Demo Audio 2", "demo/item1_mix.wav"),
+    ("Demo Audio 3", "demo/item2_mix.wav"),
+    ("Demo Audio 4", "demo/item3_mix.wav"),
+    ("Demo Audio 5", "demo/item4_mix.wav"),
+]
+
+def update_audio_input(choice):
+    return choice
+
 # CSS styling (optional)
 css = """
 #col-container {
@@ -198,6 +210,12 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
             # Input: Upload audio file
             with gr.Row():
                 gt_file_input = gr.Audio(label="Upload Audio to Separate", type="filepath", value="demo/item0_mix.wav")
+                # Dropdown for demo audio selection
+                demo_selector = gr.Dropdown(
+                    label="Select Demo Audio",
+                    choices=[name for name, _ in demo_audio_files],
+                    value="Demo Audio 1"
+                )
                 button = gr.Button("Generate", scale=1)
             
             # Output Component for edited audio
@@ -205,6 +223,13 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
                 result1 = gr.Audio(label="Separated Audio 1", type="numpy")
                 result2 = gr.Audio(label="Separated Audio 2", type="numpy")
 
+            # Update the audio input with the selected demo audio file
+            demo_selector.change(
+                fn=lambda choice: next(path for name, path in demo_audio_files if name == choice),
+                inputs=demo_selector,
+                outputs=gt_file_input
+            )
+
             # Define the trigger and input-output linking
             button.click(
                 fn=process_audio,