Update app.py

app.py

@@ -188,7 +188,7 @@ def save_midi(tokens, batch_number=None):
         list_of_MIDI_patches=patches,
         verbose=False
     )
-    return fname
+    return fname, song_events
 
 # -----------------------------
 # MUSIC GENERATION FUNCTION (Combined)
@@ -223,10 +223,10 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
     # Load seed from MIDI if there is no existing composition.
     if not final_composition and input_midi is not None:
         final_composition = load_midi(input_midi)[:num_prime_tokens]
-        midi_fname = save_midi(final_composition)
+        midi_fname, song_events = save_midi(final_composition)
         # Use the last note's time as a marker.
         midi_score = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
-            final_composition,
+            song_events,
             output_signature='Godzilla Piano Transformer',
             output_file_name=midi_fname,
             track_name='Project Los Angeles',
@@ -245,7 +245,7 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
         plot_kwargs = {'plot_title': f'Batch # {i}', 'return_plt': True}
         if len(final_composition) > PREVIEW_LENGTH:
             plot_kwargs['preview_length_in_notes'] = int(PREVIEW_LENGTH / 3)
-        midi_score = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
+        TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
             preview_tokens + tokens,
             output_signature='Godzilla Piano Transformer',
             output_file_name=midi_fname,
@@ -253,7 +253,7 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
             list_of_MIDI_patches=[0]*16,
             verbose=False
         )
-        midi_plot = TMIDIX.plot_ms_SONG(midi_score, **plot_kwargs)
+        midi_plot = TMIDIX.plot_ms_SONG(preview_tokens + tokens, **plot_kwargs)
         midi_audio = midi_to_colab_audio(midi_fname + '.mid',
                                          soundfont_path=SOUDFONT_PATH,
                                          sample_rate=16000,