Update main.py

main.py

@@ -6,6 +6,7 @@ from utils import (
     generate_song,
     remove_last_instrument,
     regenerate_last_instrument,
+    generate_new_instrument,
     change_tempo,
 )
 
@@ -38,6 +39,7 @@ def run():
                     btn_continue = gr.Button("Generate Again")
                     btn_remove_last = gr.Button("Remove last instrument")
                     btn_regenerate_last = gr.Button("Regenerate last instrument")
+                    btn_generate_new = gr.Button("Regenerate new instrument")
             with gr.Column():
                 with gr.Group():
                     audio_output = gr.Video(show_share_button=True)
@@ -105,6 +107,18 @@ def run():
                 num_tokens,
             ],
         )
+        btn_generate_new.click(
+            fn=generate_new_instrument,
+            inputs=[text_sequence, qpm],
+            outputs=[
+                audio_output,
+                midi_file,
+                plot_output,
+                instruments_output,
+                text_sequence,
+                num_tokens,
+            ],
+        )        
         btn_qpm.click(
             fn=change_tempo,
             inputs=[text_sequence, qpm],