Duración del MIDI (#2)

- Duración del MIDI (78b79a635bcca50e16dec02a2ed58be5cde62401)


Co-authored-by: Ariel Maldonado <Just5ebas@users.noreply.huggingface.co>

app.py

@@ -126,6 +126,10 @@ def mse_with_positive_pressure(y_true: tf.Tensor, y_pred: tf.Tensor):
   positive_pressure = 10 * tf.maximum(-y_pred, 0.0)
   return tf.reduce_mean(mse + positive_pressure)
 
+def calcular_duracion_midi(archivo_midi):
+    midi = pretty_midi.PrettyMIDI(archivo_midi)
+    return midi.get_end_time()
+    
 def main():
     seed = 42
     tf.random.set_seed(seed)
@@ -172,6 +176,12 @@ def main():
             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 es: {int(minutos)} minutos y {segundos:.2f} segundos.")
         
         # Cargar modelo y pesos
         with custom_object_scope({'mse_with_positive_pressure': mse_with_positive_pressure}):
@@ -257,5 +267,11 @@ def main():
             mime='audio/midi'  
         )
 
+        # Duracion del MIDI resultante
+        duracion_f = calcular_duracion_midi(sample_file)
+        minutos_f, segundos_f = divmod(duracion_f, 60)
+        
+        st.write(f"La duración del archivo MIDI es: {int(minutos_f)} minutos y {segundos_f:.2f} segundos.")
+
 if __name__ == "__main__":
     main()