Update app.py

app.py

@@ -1,7 +1,29 @@
 import gradio as gr
 import numpy
 import pathlib
+from music21 import stream,chord,duration,clef
 
+def PredictionToChord(prediction):
+    noteToChord = []
+    drumComponent = {"Hihat":"A4","Snare":"D4","KickBass":"G3","Crash":"B4","Tom":"E4","Ride":"G4"}
+    for component in prediction.split(" "):
+        if(component in drumComponent):
+            noteToChord.append(drumComponent[component])
+    
+    return noteToChord
+
+def PredictionToNotation(predict):
+    score = stream.Score()
+    drum_part = stream.Part()
+    drum_part.id = 'Drums'
+    for prediction in predict:
+        n=chord.Chord(PredictionToChord(prediction))
+        n.duration = duration.Duration(0.5)
+        drum_part.append(n)
+    drum_part.clef = clef.PercussionClef()
+    score.insert(0, drum_part)
+    score.show('lily')
+    
 def transcribe():
     
     return "drum sheet"