Add app.py

app.py

@@ -2,6 +2,7 @@ import shutil
 
 import gradio as gr
 import mido
+from matplotlib import pyplot as plt
 from mido import Message, MidiFile, MidiTrack
 import numpy as np
 import os
@@ -76,6 +77,33 @@ def convert_midi_to_audio(midi_path):
     except Exception as e:
         return f"Error converting MIDI to audio: {str(e)}"
 
+
+def generate_piano_roll(chords):
+    fig, ax = plt.subplots(figsize=(8, 4))
+
+    for i, chord in enumerate(chords):
+        for note in chord:
+            ax.broken_barh([(i * 1, 0.8)], (note.pitch.midi - 0.4, 0.8), facecolors='blue')
+
+    ax.set_xlabel("Chord Progression")
+    ax.set_ylabel("MIDI Note Number")
+    min_min_chords = 128
+    max_max_chords = 0
+    for ch in chords:
+        for note in ch:
+            if note.pitch.midi < min_min_chords:
+                min_min_chords = note.pitch.midi
+            if note.pitch.midi > max_max_chords:
+                max_max_chords = note.pitch.midi
+    ax.set_yticks(range(min_min_chords, max_max_chords + 1, 2))
+    ax.set_xticks(range(len(chords)))
+    ax.set_xticklabels([f"Chord {i + 1}" for i in range(len(chords))])
+
+    plt.grid(True, linestyle='--', alpha=0.5)
+    plt.savefig("piano_roll.png")
+    return "piano_roll.png"
+
+
 def launch_score_editor(midi_path):
     try:
         score = converter.parse(midi_path)
@@ -94,9 +122,10 @@ def process_hexachord(hexachord_str):
 
     chords = generate_chords(notes)
     midi_path = create_midi(chords)
+    piano_roll_path = generate_piano_roll(chords)
     audio_path = convert_midi_to_audio(midi_path)
 
-    return midi_path, audio_path
+    return midi_path, piano_roll_path, audio_path
 
 
 with gr.Blocks() as ui:
@@ -105,12 +134,13 @@ with gr.Blocks() as ui:
     hexachord_input = gr.Textbox(label="Enter 6 MIDI note numbers (separated by spaces)", value="60 62 64 65 67 69")
     generate_button = gr.Button("Generate Chords")
     midi_output = gr.File(label="Download MIDI File")
+    piano_roll_output = gr.Image(label="Piano Roll Visualization")
     audio_output = gr.Audio(label="Play Generated Chords", value=None, interactive=False)
 
     generate_button.click(
         fn=process_hexachord,
         inputs=[hexachord_input],
-        outputs=[midi_output, audio_output]
+        outputs=[midi_output, piano_roll_output, audio_output]
     )
 
 on_huggingface = "HUGGINGFACE_SPACE" in os.environ