Add app.py

app.py

@@ -13,6 +13,7 @@ from pydub import AudioSegment
 
 import hexachords
 
+
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 # configure.run()
 
@@ -27,11 +28,14 @@ def is_fsynth_installed():
     print('fluidsynth is NOT installed')
     return False
 
-def generate_chords(midi_pitches):
+def generate_chords(midi_pitches, itvl):
     # Placeholder for your actual chord generation function
     # Assuming hexachord is a list of MIDI note numbers
     hexa = hexachords.Hexachord()
-    cs1 = hexa.generate_chord_sequence_from_midi_pitches(midi_pitches, intrvl="P4")
+    interval_21 = 'P5'
+    if itvl == 'fourth':
+        interval_21 = 'P4'
+    cs1 = hexa.generate_chord_sequence_from_midi_pitches(midi_pitches, intrvl=interval_21)
     return cs1
 
 def create_midi(chords):
@@ -54,6 +58,7 @@ def create_midi(chords):
     mid.save(midi_path)
     return midi_path
 
+
 def convert_midi_to_audio(midi_path):
     if not shutil.which("fluidsynth"):
         try:
@@ -77,7 +82,6 @@ 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))
 
@@ -112,7 +116,7 @@ def launch_score_editor(midi_path):
     except Exception as e:
         return f"Error opening score editor: {str(e)}"
 
-def process_hexachord(hexachord_str):
+def process_hexachord(hexachord_str, itvl):
     try:
         notes = [int(n) for n in hexachord_str.split()]
         if len(notes) != 6 or len(set(notes)) != 6:
@@ -120,7 +124,7 @@ def process_hexachord(hexachord_str):
     except ValueError:
         return "Invalid input. Enter 6 MIDI note numbers separated by spaces."
 
-    chords = generate_chords(notes)
+    chords = generate_chords(notes, itvl)
     midi_path = create_midi(chords)
     piano_roll_path = generate_piano_roll(chords)
     audio_path = convert_midi_to_audio(midi_path)
@@ -132,6 +136,7 @@ with gr.Blocks() as ui:
     gr.Markdown("# Hexachord-based Chord Generator")
 
     hexachord_input = gr.Textbox(label="Enter 6 MIDI note numbers (separated by spaces)", value="60 62 64 65 67 69")
+    interval_switch = gr.Radio(choices=["fourth", "fifth"], label="Select Interval", value="fifth")
     generate_button = gr.Button("Generate Chords")
     midi_output = gr.File(label="Download MIDI File")
     piano_roll_output = gr.Image(label="Piano Roll Visualization")
@@ -139,7 +144,7 @@ with gr.Blocks() as ui:
 
     generate_button.click(
         fn=process_hexachord,
-        inputs=[hexachord_input],
+        inputs=[hexachord_input, interval_switch],
         outputs=[midi_output, piano_roll_output, audio_output]
     )
 

hexachords.py

@@ -4,7 +4,7 @@ from ortools.sat.python import cp_model
 class Hexachord:
 
     def generate_chord_sequence_from_midi_pitches(self, list_of_mp, intrvl="P5"):
-        return self.generate_chord_sequence([note.Note(mp).nameWithOctave for mp in list_of_mp])
+        return self.generate_chord_sequence([note.Note(mp).nameWithOctave for mp in list_of_mp], intrvl=intrvl)
 
     def generate_chord_sequence(self, list_of_notes, intrvl="P5"):