Update app.py

app.py

@@ -114,7 +114,7 @@ model.eval()
 # -----------------------------
 def render_midi_output(final_composition):
     """Generate MIDI score, plot, and audio from final composition."""
-    midi_score = save_midi(final_composition)
+    fname, midi_score = save_midi(final_composition)
     time_val = midi_score[-1][1] / 1000  # seconds marker from last note
     midi_plot = TMIDIX.plot_ms_SONG(
         midi_score,
@@ -122,7 +122,6 @@ def render_midi_output(final_composition):
         block_lines_times_list=[],
         return_plt=True
     )
-    fname = save_midi(final_composition)  # The file name is embedded in the saved MIDI.
     midi_audio = midi_to_colab_audio(
         fname + '.mid',
         soundfont_path=SOUDFONT_PATH,
@@ -260,12 +259,12 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
     output_batches = []
     for i, tokens in enumerate(batched_gen_tokens):
         preview_tokens = final_composition[-PREVIEW_LENGTH:]
-        midi_fname = save_midi(preview_tokens + tokens, batch_number=i)
+        midi_fname, midi_score = save_midi(preview_tokens + tokens, batch_number=i)
         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)
         TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
-            preview_tokens + tokens,
+            midi_score,
             output_signature='Godzilla Piano Transformer',
             output_file_name=midi_fname,
             track_name='Project Los Angeles',
@@ -298,7 +297,7 @@ def add_batch(batch_number, final_composition, generated_batches, block_lines):
     """Add tokens from the specified batch to the final composition and update outputs."""
     if generated_batches:
         final_composition.extend(generated_batches[batch_number])
-        midi_fname = save_midi(final_composition)
+        midi_fname, midi_score = save_midi(final_composition)
         block_lines.append(final_composition[-1] if final_composition else 0)
         midi_score = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
             final_composition,
@@ -330,8 +329,8 @@ def remove_batch(batch_number, num_tokens, final_composition, generated_batches,
         final_composition = final_composition[:-num_tokens]
         if block_lines:
             block_lines.pop()
-        midi_fname = save_midi(final_composition)
-        midi_score = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
+        midi_fname, midi_score = save_midi(final_composition)
+        TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
             final_composition,
             output_signature='Godzilla Piano Transformer',
             output_file_name=midi_fname,