Update main.py

main.py

@@ -13,7 +13,7 @@ from utils import (
 os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"
 
 DESCRIPTION = """
-<h1>🎵 SA Midi Generation Demo 🎶</h1>
+<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>
 
@@ -47,9 +47,14 @@ DESCRIPTION = """
 <p>Enjoy creating your own music!</p>
 """
 
-
+# Genrs
 genres = ["ROCK", "POP", "OTHER", "R&B/SOUL", "JAZZ", "ELECTRONIC", "RANDOM"]
 
+#Artists
+with open('artist_names.json', 'r') as f:
+    artist_names = json.load(f)
+#print("Loaded Artists names:", artist_names)
+
 demo = gr.Blocks()
 
 
@@ -65,13 +70,16 @@ def run():
                 genre = gr.Dropdown(
                     choices=genres, value="POP", label="Select the genre"
                 )
+                artist = gr.Dropdown(
+                    choices=artist_names, value=artist_names[0], label="Select the artist style"
+                )
                 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")
             with gr.Column():
-                with gr.Group():
+                with gr.Box():
                     audio_output = gr.Video(show_share_button=True)
                     midi_file = gr.File()
                     with gr.Row():
@@ -91,7 +99,7 @@ def run():
             num_tokens = gr.Text(visible=False)
         btn_from_scratch.click(
             fn=generate_song,
-            inputs=[genre, temp, empty_sequence, qpm],
+            inputs=[genre, artist, temp, empty_sequence, qpm],
             outputs=[
                 audio_output,
                 midi_file,
@@ -103,7 +111,7 @@ def run():
         )
         btn_continue.click(
             fn=generate_song,
-            inputs=[genre, temp, text_sequence, qpm],
+            inputs=[genre, artist, temp, text_sequence, qpm],
             outputs=[
                 audio_output,
                 midi_file,