change app

app.py

@@ -11,7 +11,7 @@ mastering_transfer = MasteringStyleTransfer(args)
 
 def process_audio(input_audio, reference_audio, perform_ito, ito_reference_audio=None):
     # Process the audio files
-    output_audio, predicted_params, ito_output_audio, ito_predicted_params, _, sr, _ = mastering_transfer.process_audio(
+    output_audio, predicted_params, ito_output_audio, ito_predicted_params, ito_log, sr, _ = mastering_transfer.process_audio(
         input_audio, reference_audio, ito_reference_audio if ito_reference_audio else reference_audio, {}, perform_ito
     )
     
@@ -22,7 +22,7 @@ def process_audio(input_audio, reference_audio, perform_ito, ito_reference_audio
     # Generate top 10 differences if ITO was performed
     top_10_diff = mastering_transfer.get_top_10_diff_string(predicted_params, ito_predicted_params) if ito_predicted_params is not None else "ITO not performed"
     
-    return "output_mastered.wav", "ito_output_mastered.wav" if ito_output_audio is not None else None, param_output, ito_param_output, top_10_diff
+    return "output_mastered.wav", "ito_output_mastered.wav" if ito_output_audio is not None else None, param_output, ito_param_output, top_10_diff, ito_log
 
 def process_with_ito(input_audio, reference_audio, perform_ito, use_same_reference, ito_reference_audio):
     ito_ref = reference_audio if use_same_reference else ito_reference_audio
@@ -46,6 +46,7 @@ def process_youtube_with_ito(input_url, reference_url, perform_ito, use_same_ref
 
 with gr.Blocks() as demo:
     gr.Markdown("# Mastering Style Transfer Demo")
+
     with gr.Tab("Upload Audio"):
         input_audio = gr.Audio(label="Input Audio")
         reference_audio = gr.Audio(label="Reference Audio")
@@ -73,7 +74,7 @@ with gr.Blocks() as demo:
         submit_button.click(
             process_with_ito, 
             inputs=[input_audio, reference_audio, perform_ito, use_same_reference, ito_reference_audio], 
-            outputs=[output_audio, ito_output_audio, param_output, ito_param_output, top_10_diff]
+            outputs=[output_audio, ito_output_audio, param_output, ito_param_output, top_10_diff, ito_log]
         )
 
     with gr.Tab("YouTube URLs"):