Update app.py

app.py

@@ -391,79 +391,6 @@ def generate_callback_wrapper(input_midi,
 
 #==================================================================================
 
-def add_batch(batch_number, final_composition, generated_batches, block_lines, model_state=[]):
-
-    if generated_batches:
-        final_composition.extend(generated_batches[batch_number])
-
-        # Save MIDI to a temporary file
-        midi_score = save_midi(final_composition, model_selector=model_state[2])
-    
-        block_lines.append(midi_score[-1][1] / 1000)
-
-        # MIDI plot
-        midi_plot = TMIDIX.plot_ms_SONG(midi_score, 
-                                        plot_title='Guided Accompaniment Transformer Composition',
-                                        block_lines_times_list=block_lines[:-1],
-                                        return_plt=True)
-        
-        # File name
-        fname = 'Guided-Accompaniment-Transformer-Music-Composition'
-        
-        # Save audio to a temporary file
-        midi_audio = midi_to_colab_audio(fname + '.mid', 
-                                        soundfont_path=SOUDFONT_PATH,
-                                        sample_rate=16000,
-                                        output_for_gradio=True
-                                        )
-    
-        print('Added batch #', batch_number)
-        print('=' * 70)
-
-        return (16000, midi_audio), midi_plot, 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, model_state=[]):
-
-    if final_composition:
-
-        if len(final_composition) > num_tokens:
-            final_composition = final_composition[:-num_tokens]
-            block_lines.pop()
-    
-        # Save MIDI to a temporary file
-        midi_score = save_midi(final_composition, model_selector=model_state[2])
-    
-        # MIDI plot
-        midi_plot = TMIDIX.plot_ms_SONG(midi_score, 
-                                        plot_title='Guided Accompaniment Transformer Composition',
-                                        block_lines_times_list=block_lines[:-1],
-                                        return_plt=True)
-    
-        # File name
-        fname = 'Guided-Accompaniment-Transformer-Music-Composition'
-        
-        # Save audio to a temporary file
-        midi_audio = midi_to_colab_audio(fname + '.mid', 
-                                        soundfont_path=SOUDFONT_PATH,
-                                        sample_rate=16000,
-                                        output_for_gradio=True
-                                        )
-        
-        print('Removed batch #', batch_number)
-        print('=' * 70)
-        
-        return (16000, midi_audio), midi_plot, fname+'.mid', final_composition, generated_batches, block_lines
-
-    else:
-        return None, None, None, [], [], []
-
-#==================================================================================
-
 def reset(final_composition=[], generated_batches=[], block_lines=[], model_state=[]):
     
     final_composition = []
@@ -579,27 +506,6 @@ with gr.Blocks() as demo:
                        outputs
                       )
     
-    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")
-    
-    final_audio_output = gr.Audio(label="Final MIDI audio", format="mp3", elem_id="midi_audio")
-    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, model_state],
-                  [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, model_state], 
-                     [final_audio_output, final_plot_output, final_file_output, final_composition, generated_batches, block_lines])
-
     #==================================================================================
 
     demo.unload(reset_demo)