add examples with app

.gitignore

@@ -2,6 +2,7 @@
 .idea/
 *.mid
 venv/
-*.mid
 *.wav
-*.mp3
+!examples/
+*.mp3
+gradio_cached_examples/

app.py

@@ -6,16 +6,14 @@ import os
 import tempfile
 
 
-def inner_loop(nb_tokens, temperature, chord_progression, tempo, midi_file, bar_range):
+def inner_loop(midi_file, chord_progression, tempo, temperature, nb_tokens, bar_range):
     top_p = 0.98
     seed = 0
-
-    print(midi_file)
     # Initialize the MusicLangPredictor
     ml = MusicLangPredictor('musiclang/musiclang-v2')
     tempo_message = ""  # Default message if no MIDI file is uploaded
     time_signature = (4, 4)
-    if midi_file is not None:
+    if midi_file is not None and midi_file != "":
         # Load the MIDI file and use it as the score prompt
         filepath = midi_file
         start_bar, end_bar = map(int, bar_range.split("-"))
@@ -76,47 +74,76 @@ def inner_loop(nb_tokens, temperature, chord_progression, tempo, midi_file, bar_
 
     return mp3_path, midi_path, chord_repr, tempo_message
 
-def musiclang(nb_tokens, temperature, chord_progression, tempo, midi_file, bar_range):
+def musiclang(midi_file, chord_progression, tempo, temperature, nb_tokens, bar_range):
     exception = None
     mp3_path, midi_path, chord_repr, tempo_message = None, None, None, ""
     try:
-        mp3_path, midi_path, chord_repr, tempo_message = inner_loop(nb_tokens, temperature, chord_progression, tempo,
-                                                                    midi_file, bar_range)
+        mp3_path, midi_path, chord_repr, tempo_message = inner_loop(midi_file, chord_progression, tempo, temperature, nb_tokens, bar_range)
     except Exception as e:
         exception = "Error : " + e.__class__.__name__ + " " + str(e)
     # Return the MP3 path for Gradio to display and the MIDI file path for download
-    return mp3_path, midi_path, chord_repr, tempo_message, exception
-
-# Update Gradio interface to include MIDI file upload and bar range selection
-iface = gr.Interface(
-    fn=musiclang,
-    inputs=[
-        gr.Number(label="Number of Tokens", value=1024, minimum=256, maximum=2048, step=256),
-        gr.Slider(label="Temperature", value=0.95, minimum=0.1, maximum=1.0, step=0.1),
-        gr.Textbox(label="Chord Progression (Optional)", placeholder="Am CM Dm/F E7 Am", lines=2, value=""),
-        gr.Slider(label="Tempo", value=120, minimum=60, maximum=240, step=1),
-        gr.File(label="Upload MIDI File", type="filepath", file_types=[".mid", ".midi"]),
-        gr.Textbox(label="Bar Range of input file", placeholder="0-4", value="0-4")
-    ],
-    outputs=[
-        gr.Audio(label="Generated Music"),
-        gr.File(label="Download MIDI"),
-        gr.Textbox(label="Inferred output Chord Progression", lines=2, value=""),
-        gr.Textbox(label="Tempo Used", value=""),  # Display the tempo used for generation
-        gr.Textbox(label="Info Message")  # Initially hidden, shown only if there's an error
-    ],
-    title="Controllable Symbolic Music Generation with MusicLang Predict",
-    description="""
-    \n This is the demo for MusicLang Predict, which offers advanced controllability features and high-quality music generation by manipulating symbolic music. Control your music generation by :
-    \n - Specifying the number of tokens
-    \n - <b>The temperature</b>: the level of creativity of MusicLang. The higher the temperature, the more creative the generation. We suggest you test different levels to find the one that suits your needs
-    \n - <b>Chord progression</b> : Available chord qualities include: M, m, 7, m7b5, sus2, sus4, m7, M7, dim, dim0. You can also specify the bass if it belongs to the chord (e.g., Bm/D)
-    \n - <b>The tempo</b>
-    \n - <b>Uploading a MIDI</b> file to use as a prompt
-    \n - Selecting a bar range of the input file (For example 0-4 means first four bars)
-    \n Note: The model generates a score file, not an audio one. Therefore, the audio rendering in this notebook serves as a quick preview of the generated music. For an optimized experience, we recommend downloading the midi file and opening it in your favorite Digital Audio Workstation (DAW).”
-    \n If no chord progression or MIDI file is given, it generates a free sample with the specified number of tokens.
-    \n Need more info ? Visit <a href="https://github.com/musiclang/musiclang_predict">our Github</a>"""
-)
-
-iface.launch()
+    return mp3_path, midi_path, exception
+
+
+with gr.Blocks() as demo:
+    # Introductory text
+    gr.Markdown("""
+    # Controllable Symbolic Music Generation with MusicLang Predict
+    [MusicLang Predict](https://github.com/musiclang/musiclang_predict) offers advanced controllability features and high-quality music generation by manipulating symbolic music.
+    You can for example use it to continue your composition with a specific chord progression.
+    """)
+
+    with gr.Row():
+        with gr.Column():
+            with gr.Row():
+                midi_file = gr.File(label="Prompt MIDI File (Optional)", type="filepath", file_types=[".mid", ".midi"],
+                                    elem_id='midi_file_input')
+                with gr.Column():
+                    bar_range = gr.Textbox(label="Bar Range of input file (eg: 0-4 for first four bars)", placeholder="0-4",
+                                           value="0-4", elem_id='bar_range_input')
+                    nb_tokens = gr.Number(label="Nb Tokens",
+                                          value=512, minimum=256, maximum=2048, step=256, elem_id='nb_tokens_input')
+                    temperature = gr.Slider(
+                        label="Temperature",
+                        value=0.95,
+                        visible=False,
+                        minimum=0.1, maximum=1.0, step=0.1, elem_id='temperature_input')
+                    tempo = gr.Slider(label="Tempo", value=120, minimum=60, maximum=240, step=1, elem_id='tempo_input')
+
+            with gr.Row():
+                chord_progression = gr.Textbox(
+                    label="Chord Progression (Optional)",
+                    placeholder="Am CM Dm7/F E7 Asus4", lines=2, value="", elem_id='chord_progression_input')
+
+            with gr.Row():
+                generate_btn = gr.Button("Generate", elem_id='generate_button')
+
+        with gr.Column():
+            info_message = gr.Textbox(label="Info Message", elem_id='info_message_output')
+            generated_music = gr.Audio(label="Preview generated Music", elem_id='generated_music_output')
+            generated_midi = gr.File(label="Download MIDI", elem_id='generated_midi_output')
+            generate_btn.click(
+                fn=musiclang,
+                inputs=[midi_file, chord_progression, tempo, temperature, nb_tokens, bar_range],
+                outputs=[generated_music, generated_midi, info_message]
+            )
+
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("## Examples")
+            gr.Examples(
+                examples=[["examples/Bach_847.mid", "", 120, 0.95, 512, "0-4"],
+                          ["examples/Bach_847.mid", "Cm C7/E Fm F#dim G7", 120, 0.95, 512, "0-4"],
+                          ["examples/Boney_m_ma_baker.mid", "", 120, 0.95, 512, "0-4"],
+                          ["examples/Eminem_slim_shady.mid", "Cm AbM BbM G7 Cm", 120, 0.95, 512, "0-4"],
+                          ["examples/Mozart_alla_turca.mid", "", 120, 0.95, 512, "0-4"],
+                          ["examples/Mozart_alla_turca.mid", "Am Em CM G7 E7 Am Am E7 Am", 120, 0.95, 512, "0-4"],
+                          ["examples/Daft_punk_Around_the_world.mid", "", 120, 0.95, 512, "0-4"],
+                          ],
+                inputs=[midi_file, chord_progression, tempo, temperature, nb_tokens, bar_range],
+                outputs=[generated_music, generated_midi, info_message],
+                fn=musiclang,
+                cache_examples=True,
+            )
+
+demo.launch()

flagged/log.csv

@@ -1,2 +0,0 @@
-Number of Tokens,Temperature,Chord Progression,Tempo,Upload MIDI File,Bar Range,Generated Music,Download MIDI,Inferred output Chord Progression,Info Message,flag,username,timestamp
-1024,0.9,,120,,0-4,"{""path"":""flagged/Generated Music/40e8a82a91668a20d35d/result.mp3"",""url"":""http://127.0.0.1:7860/file=/private/var/folders/52/q1zgqdvj1818brk2d_kw0stw0000gn/T/gradio/3926f3a3196bf0f0bbc8f54a4a464fa74a066dd9/result.mp3"",""size"":null,""orig_name"":""result.mp3"",""mime_type"":null,""is_stream"":false}","{""path"":""flagged/Download MIDI/1f6e813973ef7da5a542/test.mid"",""url"":""http://127.0.0.1:7860/file=/private/var/folders/52/q1zgqdvj1818brk2d_kw0stw0000gn/T/gradio/450e7e4a544b0fb7d114e7d3eadff382988e612b/test.mid"",""size"":2490,""orig_name"":""test.mid"",""mime_type"":null,""is_stream"":false}",Em CM Bm7b5/A Em GM Am Em/B Em GM Am Em/B Em GM Am Em/B Em GM Am Em/B Em GM Am Em/B Em Bm CM Em Em Bm CM Em GM Am Em/B Em GM Am Em/B Em GM Am Em/B,,,,2024-03-04 18:20:18.251715