new layout / download possible

playground.py

@@ -48,9 +48,13 @@ def generator(
     add_bars=False,
     add_bar_count=1,
 ):
-
     genesis = GenerateMidiText(model, tokenizer, piece_by_track)
-    track = {"label": label}
+    track = {
+        "label": label,
+        "instrument": instrument,
+        "temperature": temp,
+        "density": density,
+    }
     inst = next(
         (
             inst
@@ -61,7 +65,7 @@ def generator(
     )["family_number"]
 
     inst_index = -1  # default to last generated
-    if state != []:
+    if piece_by_track != []:
         for index, instrum in enumerate(state):
             if instrum["label"] == track["label"]:
                 inst_index = index  # changing if exists
@@ -102,7 +106,7 @@ def generator(
     piano_roll = plot_piano_roll(mixed_inst_midi)
     track["text"] = inst_text
     state.append(track)
-
+    output_file = "./mixed.mid"
     return (
         inst_text,
         (44100, inst_audio),
@@ -111,6 +115,7 @@ def generator(
         (44100, mixed_audio),
         regenerate,
         genesis.piece_by_track,
+        output_file,
     )
 
 
@@ -121,56 +126,36 @@ def generated_text_from_state(state):
     return generated_text_from_state
 
 
-def save_midi_to_folder(state, midi_path):
-    # check if midi_path exists using os
-    if os.path.exists(midi_path):
-        print(f"The path {midi_path} already exists")
-    else:
-        os.mkdir(midi_path)
-        print(f"Path '{midi_path}' created")
-
-    # making sure that the path ends with a slash
-    if midi_path[-1] != "/":
-        midi_path += "/"
-
-    generated_text = generated_text_from_state(state)
-    decoder.get_midi(generated_text, f"{midi_path}your_awesome_midi.mid")
-    status = f"MIDI file saved to: {midi_path}your_awesome_midi.mid"
-
-    return status
-
-
-def instrument_row(default_inst, row_id):
-    with gr.Row():
-        row = gr.Variable(row_id)
-        with gr.Column(scale=1, min_width=100):
-            inst = gr.Dropdown(
-                sorted([inst["transfer_to"] for inst in INSTRUMENT_TRANSFER_CLASSES])
-                + ["Drums"],
-                value=default_inst,
-                label="Instrument",
-            )
-            temp = gr.Dropdown(
-                [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1],
-                value=0.7,
-                label="Creativity",
-            )
-            density = gr.Dropdown([1, 2, 3], value=3, label="Note Density")
-
-        with gr.Column(scale=3):
-            output_txt = gr.Textbox(
-                label="output", lines=10, max_lines=10, show_label=False
-            )
-        with gr.Column(scale=1, min_width=100):
-            inst_audio = gr.Audio(label="TRACK Audio", show_label=True)
-            regenerate = gr.Checkbox(value=False, label="Regenerate", visible=False)
-            # add_bars = gr.Checkbox(value=False, label="Add Bars")
-            # add_bar_count = gr.Dropdown([1, 2, 4, 8], value=1, label="Add Bars")
-            gen_btn = gr.Button("Generate")
+def instrument_col(default_inst, col_id):
+    inst_label = gr.Variable(col_id)
+    with gr.Column(scale=1, min_width=100):
+        track_md = gr.Markdown(f"""## TRACK {col_id+1}""")
+        inst = gr.Dropdown(
+            sorted([inst["transfer_to"] for inst in INSTRUMENT_TRANSFER_CLASSES])
+            + ["Drums"],
+            value=default_inst,
+            label="Instrument",
+        )
+        temp = gr.Dropdown(
+            [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1],
+            value=0.7,
+            label="Creativity",
+        )
+        density = gr.Dropdown([1, 2, 3], value=3, label="Note Density")
+        regenerate = gr.State(
+            value=False
+        )  # initial state should be to generate (not regenerate)
+        # add_bars = gr.Checkbox(value=False, label="Add Bars")
+        # add_bar_count = gr.Dropdown([1, 2, 4, 8], value=1, label="Add Bars")
+        gen_btn = gr.Button("Generate")
+        inst_audio = gr.Audio(label="TRACK Audio", show_label=True)
+        output_txt = gr.Textbox(
+            label="output", lines=10, max_lines=10, show_label=False, visible=False
+        )
 
     gen_btn.click(
         fn=generator,
-        inputs=[row, regenerate, temp, density, inst, state, piece_by_track],
+        inputs=[inst_label, regenerate, temp, density, inst, state, piece_by_track],
         outputs=[
             output_txt,
             inst_audio,
@@ -179,6 +164,7 @@ def instrument_row(default_inst, row_id):
             mixed_audio,
             regenerate,
             piece_by_track,
+            output_file,
         ],
     )
 
@@ -188,46 +174,35 @@ with gr.Blocks() as demo:
     state = gr.State([])
     title = gr.Markdown(
         """ # Demo-App of The-Jam-Machine
-    A Generative AI trained on text transcription of MIDI music """
+    ## A Generative AI trained on text transcription of MIDI music """
     )
 
-    track1_md = gr.Markdown(""" ## Mixed Audio and Piano Roll """)
-    mixed_audio = gr.Audio(label="Mixed Audio")
-    piano_roll = gr.Plot(label="Piano Roll", show_label=False)
     description = gr.Markdown(
         """
-        For each **TRACK**, choose your **instrument** along with **creativity** (temperature) and **note density**. Then, hit the **Generate** Button!
-        You can have a look at the generated text; but most importantly, check the **piano roll** and listen to the TRACK audio!
-        If you don't like the track, hit the generate button to regenerate it! Generate more tracks and listen to the **mixed audio**!
+        For each **TRACK**, choose your **instrument** along with **creativity** (temperature) and **note density**. 
+        Then, hit the **Generate** Button, and after a few seconds a track should have been generated. 
+        Check the **piano roll** and listen to the TRACK! If you don't like the track, hit the generate button to regenerate it. 
+        You can then generate more tracks and listen to the **mixed audio**! \n
+        Does it sound nice? Maybe a little robotic and laking some depth... Well, you can download the MIDI file and use it in the DAW of your choice to edit the instruments and add some effects of your choices!
         """
     )
-    track1_md = gr.Markdown(""" ## TRACK 1 """)
-    instrument_row("Drums", 0)
-    track1_md = gr.Markdown(""" ## TRACK 2 """)
-    instrument_row("Synth Bass 1", 1)
-    track1_md = gr.Markdown(""" ## TRACK 3 """)
-    instrument_row("Synth Lead Square", 2)
-    # instrument_row("Piano")
-    #
-    # save = gr.Markdown(""" ## SAVE AS MIDI file - enter a valid path """)
-    # with gr.Row():
-    #     # with gr.Column(scale=1, min_width=100):
-    #     saving_path = gr.Textbox(
-    #         value="/Users/ChatGPT/Downloads/TheJamMachine/",
-    #         show_label=False,
-    #         label="Saving Path",
-    #         interactive=True,
-    #     )
-    #     # with gr.Column(scale=1, min_width=100):
-    #     gen_btn = gr.Button("Save MIDI")
-    #     # with gr.Column(scale=1, min_width=100):
-    #     status = gr.Textbox(show_label=False)
-
-    # gen_btn.click(fn=save_midi_to_folder, inputs=[state, saving_path], outputs=status)
-
-
-demo.launch(debug=True)
 
+    aud_md = gr.Markdown(f""" ## Mixed Audio, Piano Roll and MIDI Download """)
+    with gr.Row(equal_height=True, variant="default"):
+        mixed_audio = gr.Audio(label="Mixed Audio", show_label=False)
+        output_file = gr.File(
+            label="Download",
+            show_label=False,
+        )
+    with gr.Row(variant="compact"):
+        piano_roll = gr.Plot(label="Piano Roll", show_label=False)
+
+    with gr.Row(variant="default"):
+        instrument_col("Drums", 0)
+        instrument_col("Synth Bass 1", 1)
+        instrument_col("Synth Lead Square", 2)
+
+demo.launch(debug=True, server_name="0.0.0.0", share=False)
 """
 TODO: add improvise button
 TODO: cleanup input output of generator