Update app.py

app.py

@@ -443,10 +443,14 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
     
       comp_cos_sims.append(cos_sim)
     
-    min_cos_sim = min(comp_cos_sims)
-    avg_cos_sim = sum(comp_cos_sims) / len(comp_cos_sims)
-    max_cos_sim = max(comp_cos_sims)
+    output_min_sim = min(comp_cos_sims)
+    output_avg_sim = sum(comp_cos_sims) / len(comp_cos_sims)
+    output_max_sim = max(comp_cos_sims)
 
+    print('Min sim:', output_min_sim)
+    print('Avg sim:', output_avg_sim)
+    print('max sim:', output_max_sim)
+    
     print('=' * 70)
     print('Done!')
     print('=' * 70)
@@ -476,7 +480,7 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
     print('-' * 70)
     print('Req execution time:', (reqtime.time() - start_time), 'sec')
 
-    return output_src_audio, output_src_plot, output_trg_audio, output_trg_plot
+    return output_src_audio, output_src_plot, output_trg_audio, output_trg_plot, output_min_sim, output_avg_sim, output_max_sim
 
 # =================================================================================================
 
@@ -519,7 +523,11 @@ if __name__ == "__main__":
         
         run_btn = gr.Button("compare", variant="primary")
 
-        gr.Markdown("## Inpainting results")
+        gr.Markdown("## MIDI comparison results")
+
+        output_min_sim = gr.Number(label="Minimum similarity")
+        output_avg_sim = gr.Number(label="Average similarity")
+        output_max_sim = gr.Number(label="Maximum similarity")
 
         output_src_audio = gr.Audio(label="Source MIDI audio", format="mp3", elem_id="midi_audio")
         output_src_plot = gr.Plot(label="Source MIDI plot")
@@ -528,14 +536,14 @@ if __name__ == "__main__":
         output_trg_plot = gr.Plot(label="Target MIDI plot")
         
         run_event = run_btn.click(CompareMIDIs, [input_src_midi, input_trg_midi, input_sampling_resolution, input_sampling_overlap],
-                                  [output_src_audio, output_src_plot, output_trg_audio, output_trg_plot])
+                                  [output_src_audio, output_src_plot, output_trg_audio, output_trg_plot, output_min_sim, output_avg_sim, output_max_sim])
 
         gr.Examples(
             [
              ["Honesty.kar", "Hotel California.mid", 200, 0]
             ],
             [input_src_midi, input_trg_midi, input_sampling_resolution, input_sampling_overlap],
-            [output_src_audio, output_src_plot, output_trg_audio, output_trg_plot],
+            [output_src_audio, output_src_plot, output_trg_audio, output_trg_plot, output_min_sim, output_avg_sim, output_max_sim],
             CompareMIDIs,
             cache_examples=True,
         )