Update app.py

app.py

@@ -129,35 +129,66 @@ def mse_with_positive_pressure(y_true: tf.Tensor, y_pred: tf.Tensor):
 def calcular_duracion_midi(archivo_midi):
     midi = pretty_midi.PrettyMIDI(archivo_midi)
     return midi.get_end_time()
+
+def obtain_statistics(midi_file: str) -> pd.DataFrame:
+    # Convertir archivo MIDI a DataFrame de notas
+    notes_df = midi_to_notes(midi_file)
+
+    # Calcular estadísticas
+    mean_pitch = np.mean(notes_df['pitch'])
+    std_pitch = np.std(notes_df['pitch'])
+    median_pitch = np.median(notes_df['pitch'])
+
+    mean_step = np.mean(notes_df['step'])
+    std_step = np.std(notes_df['step'])
+    median_step = np.median(notes_df['step'])
+
+    mean_duration = np.mean(notes_df['duration'])
+    std_duration = np.std(notes_df['duration'])
+    median_duration = np.median(notes_df['duration'])
+
+    # Crear DataFrame con estadísticas
+    statistics_data = {
+        'Estadísticas': ['Media', 'Desviación Estandar', 'Mediana'],
+        'Tono': [mean_pitch, std_pitch, median_pitch],
+        'Paso': [mean_step, std_step, median_step],
+        'Duración': [mean_duration, std_duration, median_duration]
+    }
+    statistics_df = pd.DataFrame(statistics_data)
+
+    return statistics_df
     
 def main():
     seed = 42
     tf.random.set_seed(seed)
     np.random.seed(seed)
 
-    st.title('GENERADOR DE MELODIAS CON RNN')
-
-    
     # Rutas de archivos
     #sample_file = 'Preludes 2 Through Major keys 39.mid'
+    st.title('GENERADOR DE MELODIAS CON RNN')
+        
     out_file = 'output.mid'
     uploaded_file = st.file_uploader("Sube un archivo MIDI")
+
     
     model=''
     pesos=''
-    
+
     with st.container(height = None, border = True):
         option = st.selectbox(
-        "Elige con que modelo entrenar",
-        ("Maestro", "Lakh"))
+            "Elige con qué 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"))
+            "Elige instrumento para generar las melodías",
+            ("Piano", "Percusión cromática", "Organo", "Guitarra", "Bajo", "Instrumentos de cuerda", "Conjunto", "Laton",
+             "Junco", "Pipa", "Instrumento sintetizador", "Pad sintetizador", "Efecto sintetizador", "Etnico", "Percusion", "Efectos de sonido")
+        )
     
         num_predictions = st.number_input("Ingrese el número de notas:", min_value=100, max_value=150, value=120, step=1)
-    
+        
+        
     if uploaded_file and option is not None:
 
         if option=="Maestro":
@@ -167,22 +198,23 @@ def main():
             model="mi_modelo03_music.h5"
             pesos="mi_pesos03_music.h5"
         
-        st.subheader("Archivo cargado:")
-        st.write(uploaded_file.name)
-        
-        # Guardar el archivo en una ubicación temporal
-        with open(uploaded_file.name, 'wb') as f:
-            f.write(uploaded_file.getbuffer())
+        st.subheader("Archivo cargado")
+        with st.container(height = None, border = True):
+            st.write(uploaded_file.name)
             
-        sample_file=uploaded_file.name
-
-        # Duracion del MIDI
-        duracion = calcular_duracion_midi(sample_file)
-        minutos, segundos = divmod(duracion, 60)
-        
-        st.write(f"La duración del archivo MIDI subido es: {int(minutos)}:{int(segundos):02d}")
+            # Guardar el archivo en una ubicación temporal
+            with open(uploaded_file.name, 'wb') as f:
+                f.write(uploaded_file.getbuffer())
+                
+            sample_file=uploaded_file.name
+    
+            # Duracion del MIDI
+            duracion = calcular_duracion_midi(sample_file)
+            minutos, segundos = divmod(duracion, 60)
+            
+            st.write(f"La duración del archivo MIDI subido es: {int(minutos)}:{int(segundos):02d}")
 
-        st.subheader("Modelo elegido:")
+        st.subheader("Modelo elegido")
 
         with st.container(height = None, border = True):
             st.write(option, f" de tipo instrumental ", option_musica)
@@ -200,35 +232,35 @@ def main():
         if option_musica is not None:
             if option_musica=="Piano":
                 instrument_name="Acoustic Grand Piano"
-            elif option_musica=="Chromatic Percussion":
+            elif option_musica=="Percusión cromática":
                 instrument_name="Celesta"
-            elif option_musica=="Organ":
+            elif option_musica=="Organo":
                 instrument_name="Hammond Organ"
-            elif option_musica=="Guitar":
+            elif option_musica=="Guitarra":
                 instrument_name="Acoustic Guitar (nylon)"
-            elif option_musica=="Bass":
+            elif option_musica=="Bajo":
                 instrument_name="Acoustic Bass"
-            elif option_musica=="Strings":
+            elif option_musica=="Instrumentos de cuerda":
                 instrument_name="Violin"
-            elif option_musica=="Ensemble":
+            elif option_musica=="Conjunto":
                 instrument_name="String Ensemble 1"
-            elif option_musica=="Brass":
+            elif option_musica=="Laton":
                 instrument_name="Trumpet"
-            elif option_musica=="Reed":
+            elif option_musica=="Junco":
                 instrument_name="Soprano Sax"
-            elif option_musica=="Pipe":
+            elif option_musica=="Pipa":
                 instrument_name="Piccolo"
-            elif option_musica=="Synth Lead":
+            elif option_musica=="Instrumento sintetizador":
                 instrument_name="Lead 2 (sawtooth)"
-            elif option_musica=="Synth Pad":
+            elif option_musica=="Pad sintetizador":
                 instrument_name="Pad 2 (warm)"
-            elif option_musica=="Synth Effects":
+            elif option_musica=="Efecto sintetizador":
                 instrument_name="FX 2 (soundtrack)"
-            elif option_musica=="Ethnic":
+            elif option_musica=="Etnico":
                 instrument_name="Banjo"
-            elif option_musica=="Percussive":
+            elif option_musica=="Percusion":
                 instrument_name="Melodic Tom"
-            elif option_musica=="Sound Effects":
+            elif option_musica=="Efectos de sonido":
                 instrument_name="Guitar Fret Noise"
             else:
                 instrument_name=pretty_midi.program_to_instrument_name(pm.instruments[0].program)
@@ -264,6 +296,20 @@ def main():
     
         archivo_midi = open(out_file, 'rb').read()
 
+        csv_file = obtain_statistics(out_file)
+
+        # Guardar el archivo temporalmente
+        out_file = 'temp.mid'
+        with open(out_file, 'wb') as f:
+            f.write(uploaded_file.getbuffer())
+
+        # Obtener estadísticas
+        statistics_df = obtain_statistics(out_file)
+
+        # Mostrar estadísticas en Streamlit
+        st.write("### Estadísticas generadas:")
+        st.dataframe(statistics_df)
+            
         with st.container(height = None, border = True):
             st.download_button(
                 label="Descargar MIDI",
@@ -271,6 +317,14 @@ def main():
                 file_name=out_file,  # Nombre del archivo que se descargará
                 mime='audio/midi'  
             )
+            
+            # st.download_button(
+            #     label="Descargar CSV",
+            #     data=statistics_df.to_csv(index=False).encode('utf-8'),
+            #     file_name='statistics.csv',
+            #     mime='text/csv'
+            # )
+            
 
             # Duracion del MIDI resultante
             duracion_f = calcular_duracion_midi(out_file)