Update app.py

app.py

@@ -146,6 +146,10 @@ def main():
     "Elige con que modelo entrenar",
     ("Maestro", "Lakh"))
 
+    option_musica = st.selectbox(
+    "Elige instrumento a generar las melodias",
+    ("Piano", "Chromatic Percussion", "Organ", "Guitar", "Bass", "Strings", "Ensemble", "Brass",
+    "Reed", "Pipe", "Synth Lead", "Synth Pad", "Synth Effects", "Ethnic", "Percussive", "Sound Effects"))
     
     
     if uploaded_file and option is not None:
@@ -177,7 +181,44 @@ def main():
         
         # Convertir MIDI generado por el modelo a archivo WAV
         pm = pretty_midi.PrettyMIDI(sample_file)
-        instrument_name = pretty_midi.program_to_instrument_name(pm.instruments[0].program)
+        instrument_name = ""
+
+        if option_musica is not None:
+            if option_musica=="Piano":
+                instrument_name="Acoustic Grand Piano"
+            elif option_musica=="Chromatic Percussion":
+                instrument_name="Celesta"
+            elif option_musica=="Organ":
+                instrument_name="Hammond Organ"
+            elif option_musica=="Guitar":
+                instrument_name="Acoustic Guitar (nylon)"
+            elif option_musica=="Bass":
+                instrument_name="Acoustic Bass"
+            elif option_musica=="Strings":
+                instrument_name="Violin"
+            elif option_musica=="Ensemble":
+                instrument_name="String Ensemble 1"
+            elif option_musica=="Brass":
+                instrument_name="Trumpet"
+            elif option_musica=="Reed":
+                instrument_name="Soprano Sax"
+            elif option_musica=="Pipe":
+                instrument_name="Piccolo"
+            elif option_musica=="Synth Lead":
+                instrument_name="Lead 2 (sawtooth)"
+            elif option_musica=="Synth Pad":
+                instrument_name="Pad 2 (warm)"
+            elif option_musica=="Synth Effects":
+                instrument_name="FX 2 (soundtrack)"
+            elif option_musica=="Ethnic":
+                instrument_name="Banjo"
+            elif option_musica=="Percussive":
+                instrument_name="Melodic Tom"
+            elif option_musica=="Sound Effects":
+                instrument_name="Guitar Fret Noise"
+            else:
+                instrument_name=pretty_midi.program_to_instrument_name(pm.instruments[0].program)
+        
         raw_notes = midi_to_notes(sample_file)
         key_order = ['pitch', 'step', 'duration']
         seq_length = 25