Update app.py

app.py

@@ -293,74 +293,6 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
         outputs_flat.extend([batch[0], batch[1]])
     return [final_composition, generated_batches, block_lines] + outputs_flat
 
-# -----------------------------
-# BATCH HANDLING FUNCTIONS
-# -----------------------------
-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, midi_score = save_midi(final_composition)
-        block_lines.append(midi_score[-1][1] / 1000 if final_composition else 0)
-        TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
-            midi_score,
-            output_signature='Godzilla Piano Transformer',
-            output_file_name=midi_fname,
-            track_name='Project Los Angeles',
-            list_of_MIDI_patches=[0]*16,
-            verbose=False
-        )
-        midi_plot = TMIDIX.plot_ms_SONG(
-            midi_score,
-            plot_title='Godzilla Piano Transformer Composition',
-            block_lines_times_list=block_lines[:-1],
-            return_plt=True
-        )
-        midi_audio = midi_to_colab_audio(midi_fname + '.mid',
-                                         soundfont_path=SOUDFONT_PATH,
-                                         sample_rate=16000,
-                                         output_for_gradio=True)
-        print("Added batch #", batch_number)
-        print_sep()
-        return (16000, midi_audio), midi_plot, midi_fname + '.mid', final_composition, generated_batches, block_lines
-    else:
-        return None, None, None, [], [], []
-
-def remove_batch(batch_number, num_tokens, final_composition, generated_batches, block_lines):
-    """Remove tokens from the final composition and update outputs."""
-    if final_composition and len(final_composition) > num_tokens:
-        final_composition = final_composition[:-num_tokens]
-        if block_lines:
-            block_lines.pop()
-        midi_fname, midi_score = save_midi(final_composition)
-        TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
-            midi_score,
-            output_signature='Godzilla Piano Transformer',
-            output_file_name=midi_fname,
-            track_name='Project Los Angeles',
-            list_of_MIDI_patches=[0]*16,
-            verbose=False
-        )
-        midi_plot = TMIDIX.plot_ms_SONG(
-            midi_score,
-            plot_title='Godzilla Piano Transformer Composition',
-            block_lines_times_list=block_lines[:-1],
-            return_plt=True
-        )
-        midi_audio = midi_to_colab_audio(midi_fname + '.mid',
-                                         soundfont_path=SOUDFONT_PATH,
-                                         sample_rate=16000,
-                                         output_for_gradio=True)
-        print("Removed batch #", batch_number)
-        print_sep()
-        return (16000, midi_audio), midi_plot, midi_fname + '.mid', final_composition, generated_batches, block_lines
-    else:
-        return None, None, None, [], [], []
-
-def clear():
-    """Clear outputs and reset state."""
-    return None, None, None, [], []
-
 def reset(final_composition=[], generated_batches=[], block_lines=[]):
     """Reset composition state."""
     return [], [], []
@@ -370,12 +302,12 @@ def reset(final_composition=[], generated_batches=[], block_lines=[]):
 # -----------------------------
 with gr.Blocks() as demo:
 
-    gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>Godzilla Piano Transformer</h1>")
-    gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>Fast 807M 4k solo Piano music transformer trained on 1.14M+ MIDIs (2.7M+ samples)</h1>")
+    gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>MuseCraft Piano Transformer</h1>")
+    gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>Solo Piano music transformer for MuseCraft project</h1>")
     gr.HTML("""
-        Check out <a href="https://huggingface.co/datasets/asigalov61/Godzilla-Piano">Godzilla Piano dataset</a> on Hugging Face
+        Check out <a href="https://github.com/MIDIAI/MuseCraft">MuseCraft project</a> on GitHub
         <p>
-            <a href="https://huggingface.co/spaces/asigalov61/Godzilla-Piano-Transformer?duplicate=true">
+            <a href="https://huggingface.co/spaces/projectlosangeles/MuseCraft-Piano-Transformer?duplicate=true">
                 <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-md.svg" alt="Duplicate in Hugging Face">
             </a>
         </p>
@@ -384,8 +316,6 @@ with gr.Blocks() as demo:
 
     # Global state variables for composition
     final_composition = gr.State([])
-    generated_batches = gr.State([])
-    block_lines = gr.State([])
 
     gr.Markdown("## Upload seed MIDI or click 'Generate' for a random output")
     input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
@@ -399,42 +329,14 @@ with gr.Blocks() as demo:
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
     generate_btn = gr.Button("Generate", variant="primary")
 
-    gr.Markdown("## Batch Previews")
-    outputs = [final_composition, generated_batches, block_lines]
-    # Two outputs (audio and plot) for each batch
-    for i in range(NUM_OUT_BATCHES):
-        with gr.Tab(f"Batch # {i}"):
-            audio_output = gr.Audio(label=f"Batch # {i} MIDI Audio", format="mp3")
-            plot_output = gr.Plot(label=f"Batch # {i} MIDI Plot")
-            outputs.extend([audio_output, plot_output])
+    gr.Markdown("## MIDI Output")
+            
     generate_btn.click(
         generate_music_and_state,
-        [input_midi, num_prime_tokens, num_gen_tokens, num_mem_tokens, model_temperature,
-         final_composition, generated_batches, block_lines],
-        outputs
+        [input_midi, num_prime_tokens, num_gen_tokens, num_mem_tokens, model_temperature],
+        generated_MIDI_file
     )
 
-    gr.Markdown("## Add/Remove Batch")
-    batch_number = gr.Slider(0, NUM_OUT_BATCHES - 1, value=0, step=1, label="Batch number to add/remove")
-    add_btn = gr.Button("Add batch", variant="primary")
-    remove_btn = gr.Button("Remove batch", variant="stop")
-    clear_btn = gr.ClearButton()
-
-    final_audio_output = gr.Audio(label="Final MIDI audio", format="mp3")
-    final_plot_output = gr.Plot(label="Final MIDI plot")
-    final_file_output = gr.File(label="Final MIDI file")
-
-    add_btn.click(
-        add_batch,
-        [batch_number, final_composition, generated_batches, block_lines],
-        [final_audio_output, final_plot_output, final_file_output, final_composition, generated_batches, block_lines]
-    )
-    remove_btn.click(
-        remove_batch,
-        [batch_number, num_gen_tokens, final_composition, generated_batches, block_lines],
-        [final_audio_output, final_plot_output, final_file_output, final_composition, generated_batches, block_lines]
-    )
-    clear_btn.click(clear, inputs=None,
-                    outputs=[final_audio_output, final_plot_output, final_file_output, final_composition, block_lines])
+    generated_MIDI_file = gr.File(label="Generated MIDI file")
 
 demo.launch()