Add song list

app.py

@@ -1,6 +1,7 @@
 from transformers import pipeline
 import gradio as gr
 from song_guesser import SongGuesser
+from list_songs import ListSongs
 
 pipe = pipeline(model="SamuelHarner/whisper-tuned")  # change to "your-username/the-name-you-picked"
 
@@ -21,4 +22,8 @@ iface = gr.Interface(
     description="Sing a Swedish Christmas song and see if we can guess it. If our guess is wrong, try singing more clearly, more of the lyrics, or another song and maybe we will get it the next time!",
 )
 
+markdown_text = "The songs that we can guess are: \n" + ListSongs.get_song_list()
+
+iface.add_component(gr.Markdown(markdown_text))
+
 iface.launch()

list_songs.py

@@ -0,0 +1,14 @@
+import os
+
+class ListSongs:
+    @staticmethod
+    def get_song_list():
+        folder_path = './swedish_christmas_songs'
+
+        list_output = ""
+
+        for filename in sorted(os.listdir(folder_path)):
+            if filename.endswith('.txt'):  # Check if the file is a text file
+                list_output += "- " + filename[:-4].replace("_", " ") + "\n"
+
+        return list_output.strip()