add more songs

app.py

@@ -7,10 +7,20 @@ from time import sleep
 import argparse
 import cv2
 import numpy as np
+import logging
 
 
-SONG_1 = "au clair de la lune"
-SONG_2 = "ainsi font, font, font"
+SONG_1 = "Au clair de la lune"
+SONG_2 = "Ainsi font, font, font"
+SONG_3 = "Dodo l'enfant do"
+SONG_4 = "A la claire fontaine"
+
+SONG = {
+    SONG_1: "fa fa fa sol la - sol - fa la sol sol fa - fa fa fa sol la - sol - fa la sol sol fa - - sol sol sol sol re - re - sol fa mi re do - fa fa fa sol la - sol - fa la sol sol fa",
+    SONG_2: "mi mi do - mi - sol - sol - la sol fa mi re mi do mi mi do - mi - sol - sol - la sol fa mi re do",
+    SONG_3: "mi - do - mi mi do - re mi fa mi re sol mi do - mi - do - mi mi do - re mi fa mi re sol do",
+    SONG_4: "fa - fa la la sol la sol - fa - fa la la sol la - la - la sol fa la do la do - do la fa la sol - fa - fa la la sol fa la fa la - la sol fa la sol fa"
+}
 
 
 def select_song(song: str = SONG_1, context={}):
@@ -40,11 +50,15 @@ def play_note(audio_signal: np.ndarray, context={}):
             sleep(0.01)
             print("waiting for file")
     assert file_name.exists()
-    if context["time_index"] == 0:
-        context["__stop"]()
-    else:
-        context["__set_audio"](file_name)
-        context["__play"]()
+    try:
+        if context["time_index"] == 0:
+            context["__stop"]()
+        else:
+            context["__set_audio"](file_name)
+            context["__play"]()
+    except Exception as e:
+        logging.warning(
+            f"Error playing note {note}: {e}, not expected to work with MPL backend for instance")
 
 
 def display_current_color(context={}):
@@ -127,10 +141,7 @@ def xylo_player():
 
 def song_player(context={}):
     song = context["song"]
-    SONG = {
-        SONG_1: "fa fa fa sol la - sol - la sol sol fa - fa fa fa sol la",
-        SONG_2: "mi mi do - mi - sol - sol - la sol fa mi re mi do mi mi do - mi - sol - sol - la sol fa mi re do"
-    }
+
     song_str = SONG.get(song, "")
     image_song, target_note = generate_song(
         song_str, current_time=context.get("time_index", 0))
@@ -176,7 +187,7 @@ def generate_song(song_str, current_time=None) -> Tuple[np.ndarray, str]:
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='Xylophone synthesizer')
     parser.add_argument('-b', '--backend', type=str,
-                        default='gradio', choices=['gradio', 'qt'])
+                        default='gradio', choices=['gradio', 'qt', 'mpl'])
     args = parser.parse_args()
     all_notes = list(NOTE_FREQUENCIES.keys())
     icon_list = [Path(f"__{note}.jpg") for note in all_notes]
@@ -184,7 +195,7 @@ if __name__ == '__main__':
         img = get_color(note, size=(512, 512))
         Image.save_image(img, icon)
     interactive(note=Control("C4", all_notes, icons=icon_list))(select_note)
-    interactive(song=(SONG_1, [SONG_1, SONG_2]))(select_song)
+    interactive(song=(SONG_1, list(SONG.keys())))(select_song)
     interactive_pipeline(
         gui=args.backend,
         cache=False,