rollback

app.py

@@ -156,16 +156,6 @@ 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")
@@ -173,26 +163,17 @@ with gr.Blocks() as demo:
         gr.Markdown("Interactive demo of Inference Time Optimization (ITO) for Music Mastering Style Transfer. \
                     The mastering style transfer is performed by a differentiable audio processing model, and the predicted parameters are shown as the output. \
                     Perform mastering style transfer with an input source audio and a reference mastering style audio. On top of this result, you can perform ITO to optimize the reference embedding $z_{ref}$ to further gain control over the output mastering style.")
-        gr.Image("ito_snow.png", width=300, height=200, label="ITO pipeline")
+        gr.Image("ito_snow.png", width=500, height=300, label="ITO pipeline")
 
     gr.Markdown("## Step 1: Mastering Style Transfer")
 
     with gr.Tab("Upload Audio"):
         with gr.Row():
-            use_example_files = gr.Checkbox(label="Use Example Files", value=False)
-    
-        with gr.Row():
-            input_audio = gr.Audio(label="Source Audio $x_{in}$", interactive=not use_example_files.value)
-            reference_audio = gr.Audio(label="Reference Style Audio $x_{ref}$", interactive=not use_example_files.value)
-
-        # Dropdowns for selecting example files
-        with gr.Row():
-            input_example_dropdown = gr.Dropdown(label="Select Input Example", choices=input_examples, interactive=use_example_files.value)
-            reference_example_dropdown = gr.Dropdown(label="Select Reference Example", choices=reference_examples, interactive=use_example_files.value)
-
+            input_audio = gr.Audio(label="Source Audio $x_{in}$")
+            reference_audio = gr.Audio(label="Reference Style Audio $x_{ref}$")
+        
         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>')
-        gr.Markdown("all output samples are normalized to -12dB")
         
         with gr.Row():
             with gr.Column():
@@ -200,27 +181,12 @@ 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, use_examples):
-            if use_examples:
-                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_with_examples, 
-            inputs=[input_audio, reference_audio, input_example_dropdown, reference_example_dropdown, use_example_files], 
+            process_audio, 
+            inputs=[input_audio, reference_audio], 
             outputs=[output_audio, param_output, normalized_input]
         )
 
-        # Update the interactivity of the audio inputs and dropdowns based on the checkbox
-        use_example_files.change(
-            lambda use_examples: (not use_examples, not use_examples),
-            inputs=[use_example_files],
-            outputs=[input_audio, reference_audio]
-        )
-
     with gr.Tab("YouTube Audio"):
         with gr.Row():
             input_youtube_url = gr.Textbox(label="Input YouTube URL")