choose chord progression

app.py

@@ -4,7 +4,9 @@ from musiclang import Score
 from midi2audio import FluidSynth
 import os
 
-def musiclang(nb_tokens, temperature, chord_progression, tempo, midi_file, bar_range):
+
+
+def inner_loop(nb_tokens, temperature, chord_progression, tempo, midi_file, bar_range):
     top_p = 1.0
     seed = 0
 
@@ -59,9 +61,18 @@ def musiclang(nb_tokens, temperature, chord_progression, tempo, midi_file, bar_r
     mp3_path = 'result.mp3'
     FluidSynth("/usr/share/sounds/sf2/FluidR3_GM.sf2").midi_to_audio(midi_path, wav_path)
     os.system(f'ffmpeg -i {wav_path} -acodec libmp3lame -y -loglevel quiet -stats {mp3_path}')
+    return mp3_path, midi_path, chord_repr
+
+def musiclang(nb_tokens, temperature, chord_progression, tempo, midi_file, bar_range):
+    exception = None
+    mp3_path, midi_path, chord_repr = None, None, None
+    try:
+        mp3_path, midi_path, chord_repr = inner_loop(nb_tokens, temperature, chord_progression, tempo, midi_file, 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
+    return mp3_path, midi_path, chord_repr, exception
 
 # Update Gradio interface to include MIDI file upload and bar range selection
 iface = gr.Interface(
@@ -77,12 +88,15 @@ iface = gr.Interface(
     outputs=[
         gr.Audio(label="Generated Music"),
         gr.File(label="Download MIDI"),
-        gr.Textbox(label="Inferred output Chord Progression", lines=2, value="")
+        gr.Textbox(label="Inferred output Chord Progression", lines=2, value=""),
+        gr.Textbox(label="Info Message")  # Initially hidden, shown only if there's an error
     ],
     title="Controllable Symbolic Music Generation with MusicLang Predict",
-    description="""Customize the music generation by specifying the number of tokens, temperature, chord progression, tempo, and optionally uploading a MIDI file to use as a prompt. Specify the bar range for the MIDI prompt.
+    description="""
+    \n Simple application that wraps <a href="https://github.com/musiclang/musiclang_predict">musiclang predict</a>.
+    \n Customize the music generation by specifying the number of tokens, temperature, chord progression, tempo, and optionally uploading a MIDI file to use as a prompt. Specify the bar range for the MIDI prompt.
     \nChord qualities: M, m, 7, m7, m7b5, sus2, sus4, M7, dim, dim7. You can also specify the bass if it belongs to the chord (e.g., Bm/D).
-    \nIf no chord progression or MIDI file is given, it generates a free sample with the specified number of tokens."""
+    If no chord progression or MIDI file is given, it generates a free sample with the specified number of tokens."""
 )
 
 iface.launch()