Spaces:
Running
Running
File size: 2,992 Bytes
440bac0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | import pretty_midi
import os
import logging
logger = logging.getLogger(__name__)
def parse_midi_file(filepath: str) -> dict:
"""
Parsea un archivo MIDI (.mid o .midi) y extrae todas sus pistas, notas, tempo y clasificación GM.
"""
if not os.path.exists(filepath):
raise FileNotFoundError(f"No se encontró el archivo MIDI en: {filepath}")
try:
pm = pretty_midi.PrettyMIDI(filepath)
tempo = pm.estimate_tempo()
duration = pm.get_end_time()
tracks = []
for idx, inst in enumerate(pm.instruments):
notes = []
for note in inst.notes:
notes.append({
"note": pretty_midi.note_number_to_name(note.pitch),
"midi": note.pitch,
"start": round(note.start, 3),
"duration": round(note.end - note.start, 3),
"velocity": note.velocity
})
# Si no tiene notas, la ignoramos para mantener el payload limpio
if not notes:
continue
if inst.is_drum:
inst_type = "drums"
name = "Batería / Percusión"
else:
prog = inst.program
if 0 <= prog <= 7:
inst_type = "piano"
name = f"Piano (GM {prog})"
elif 24 <= prog <= 31:
inst_type = "guitar"
name = f"Guitarra (GM {prog})"
elif 32 <= prog <= 39:
inst_type = "bass"
name = f"Bajo (GM {prog})"
elif 16 <= prog <= 23:
inst_type = "organ"
name = f"Órgano (GM {prog})"
elif 40 <= prog <= 55:
inst_type = "strings"
name = f"Cuerdas/Orquesta (GM {prog})"
elif 56 <= prog <= 71:
inst_type = "brass"
name = f"Metales/Vientos (GM {prog})"
else:
inst_type = "synth"
name = f"Sintetizador (GM {prog})"
tracks.append({
"id": idx,
"nombre": name,
"program": inst.program,
"is_drum": inst.is_drum,
"tipo_instrumento": inst_type,
"total_notas": len(notes),
"notas": notes
})
return {
"success": True,
"tempo_bpm": round(tempo, 1) if tempo > 0 else 120.0,
"duracion_segundos": round(duration, 2),
"total_pistas": len(tracks),
"pistas": tracks
}
except Exception as e:
logger.error(f"Error parseando archivo MIDI {filepath}: {e}")
return {
"success": False,
"error": f"El archivo no es un MIDI válido o está dañado: {str(e)}"
}
|