Update main.py

main.py

@@ -14,33 +14,6 @@ os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"
 
 DESCRIPTION = """
 <h1>🎵 Multitrack Midi Generator 🎶</h1>
-<h3>AI-driven Music Composer: Creating Music One Instrument at a Time!</h3>
-<p>This interactive application uses an AI model to generate music sequences based on a chosen genre and various user inputs. The apps constructs the piece instrument by instrument</p>
-<div style="display: flex; justify-content: space-between;">
-    <div style="width: 45%; margin-right: 5%;">
-        <h2>Features:</h2>
-        <ul>
-            <li>🎼 Select the genre for the music.</li>
-            <li>🌡️ Use the "Temperature" slider to adjust the randomness of the music generated (higher values will produce more random outputs).</li>
-            <li>⏱️ Adjust the "Tempo" slider to change the speed of the music.</li>
-            <li>🎹 Use the buttons to generate a new song from scratch, continue generation with the current settings, remove the last added instrument, regenerate the last added instrument with a new one, or change the tempo of the current song.</li>
-        </ul>
-    </div>
-    <div style="width: 45%; margin-left: 5%;">
-        <h2>Outputs:</h2>
-        <p>The app outputs the following:</p>
-        <ul>
-            <li>🎧 The audio of the generated song.</li>
-            <li>📁 A MIDI file of the song.</li>
-            <li>📊 A plot of the song's sequence.</li>
-            <li>🎸 A list of the generated instruments.</li>
-            <li>📝 The text sequence of the song.</li>
-        </ul>
-    </div>
-</div>
-<hr style="margin-top: 2em; margin-bottom: 2em;">
-<p>This application is built upon the inspiring work of <a href="https://www.linkedin.com/in/dr-tristan-behrens-734967a2" target="_blank">Dr. Tristan Behrens</a></p>
-<p>Enjoy creating your own music!</p>
 """
 
 
@@ -52,7 +25,6 @@ demo = gr.Blocks()
 def run():
     with demo:
         gr.HTML(DESCRIPTION)
-        gr.DuplicateButton(value="Duplicate Space for private use")
         with gr.Row():
             with gr.Column():
                 temp = gr.Slider(
@@ -62,12 +34,12 @@ def run():
                     choices=genres, value="POP", label="Select the genre"
                 )
                 with gr.Row():
-                    btn_from_scratch = gr.Button("🧹 Start from scratch")
-                    btn_continue = gr.Button("➡️ Continue Generation")
-                    btn_remove_last = gr.Button("↩️ Remove last instrument")
-                    btn_regenerate_last = gr.Button("🔄 Regenerate last instrument")
+                    btn_from_scratch = gr.Button("Start")
+                    btn_continue = gr.Button("Continue Generation")
+                    btn_remove_last = gr.Button("Remove last instrument")
+                    btn_regenerate_last = gr.Button("Regenerate last instrument")
             with gr.Column():
-                with gr.Box():
+                with gr.Group():
                     audio_output = gr.Video(show_share_button=True)
                     midi_file = gr.File()
                     with gr.Row():
@@ -85,18 +57,6 @@ def run():
             empty_sequence = gr.Text(visible=False)
         with gr.Row():
             num_tokens = gr.Text(visible=False)
-        btn_from_scratch.click(
-            fn=generate_song,
-            inputs=[genre, temp, empty_sequence, qpm],
-            outputs=[
-                audio_output,
-                midi_file,
-                plot_output,
-                instruments_output,
-                text_sequence,
-                num_tokens,
-            ],
-        )
         btn_continue.click(
             fn=generate_song,
             inputs=[genre, temp, text_sequence, qpm],