add examples

app.py

@@ -156,6 +156,16 @@ def update_ito_output(all_results, selected_step):
     return (args.sample_rate, current_output), ito_param_output, selected_result['log']
 
 
+# Define the path to the examples folder
+EXAMPLES_DIR = "examples/"
+example_files = [f for f in os.listdir(EXAMPLES_DIR) if f.endswith('.mp3')]
+
+# Create lists for input and reference examples
+input_examples = [f"input_{i}.mp3" for i in range(1, len(example_files)//2 + 1)]
+reference_examples = [f"reference_{i}.mp3" for i in range(1, len(example_files)//2 + 1)]
+
+
+
 """ APP display """
 with gr.Blocks() as demo:
     gr.Markdown("# ITO-Master: Inference Time Optimization for Mastering Style Transfer")
@@ -172,7 +182,13 @@ with gr.Blocks() as demo:
             input_audio = gr.Audio(label="Source Audio $x_{in}$")
             reference_audio = gr.Audio(label="Reference Style Audio $x_{ref}$")
         
+        # Dropdowns for selecting example files
+        with gr.Row():
+            input_example_dropdown = gr.Dropdown(label="Select Input Example", choices=input_examples)
+            reference_example_dropdown = gr.Dropdown(label="Select Reference Example", choices=reference_examples)
+
         process_button = gr.Button("Process Mastering Style Transfer")
+        gr.Markdown('<span style="color: lightgray; font-style: italic;">all output samples are normalized to -12dB</span>')
         
         with gr.Row():
             with gr.Column():
@@ -180,9 +196,16 @@ with gr.Blocks() as demo:
                 normalized_input = gr.Audio(label="Normalized Source Audio", type='numpy')
             param_output = gr.Textbox(label="Predicted Parameters", lines=5)
 
+        def process_audio_with_examples(input_audio, reference_audio, input_example, reference_example):
+            if input_example:
+                input_audio = sf.read(os.path.join(EXAMPLES_DIR, input_example))[0]
+            if reference_example:
+                reference_audio = sf.read(os.path.join(EXAMPLES_DIR, reference_example))[0]
+            return process_audio(input_audio, reference_audio)
+
         process_button.click(
-            process_audio, 
-            inputs=[input_audio, reference_audio], 
+            process_audio_with_examples, 
+            inputs=[input_audio, reference_audio, input_example_dropdown, reference_example_dropdown], 
             outputs=[output_audio, param_output, normalized_input]
         )
 
@@ -195,6 +218,7 @@ with gr.Blocks() as demo:
             reference_audio_yt = gr.Audio(label="Reference Style Audio (Do not put when using YouTube URL)")
         
         process_button_yt = gr.Button("Process Mastering Style Transfer")
+        gr.Markdown('<span style="color: lightgray; font-style: italic;">all output samples are normalized to -12dB</span>')
         
         with gr.Row():
             output_audio_yt = gr.Audio(label="Output Audio", type='numpy')
@@ -260,6 +284,7 @@ with gr.Blocks() as demo:
     )
 
     ito_button = gr.Button("Perform ITO")
+    gr.Markdown('<span style="color: lightgray; font-style: italic;">all output samples are normalized to -12dB</span>')
 
     with gr.Row():
         with gr.Column():