Update app.py

app.py

@@ -120,7 +120,27 @@ def render_midi(input_midi, render_type, soundfont_bank, render_sample_rate, mel
         for e in output_score:
             if e[3] != 9:
                 e[4] = 127 - e[4]
-        
+
+    elif render_type == 'Repair':
+        output_score = []
+
+        for c in cscore:
+            tones_chord = sorted(set([t[4] % 12 for t in c if t[3] != 9]))
+            drums_events = [t for t in c if t[3] == 9]
+
+            new_tones_chord = TMIDIX.check_and_fix_tones_chord(tones_chord)
+            
+            if tones_chord:
+                if len(c) > 1:
+                    output_score.extend([c[0]])
+                    for cc in c[1:]
+                        if cc[3] != 9 and (cc[4] % 12) in new_tones_chord:
+                            output_score.extend([cc])
+                    output_score.extend(drums_events)
+                else:
+                    
+                    output_score.extend([c[0]] + drums_events)
+                        
     print('Done processing!')
     print('=' * 70)
     
@@ -247,7 +267,7 @@ if __name__ == "__main__":
 
         gr.Markdown("## Select desired render type")
 
-        render_type = gr.Radio(["Render as-is", "Extract melody", "Transform"], label="Render type", value="Render as-is")
+        render_type = gr.Radio(["Render as-is", "Extract melody", "Transform", "Repair"], label="Render type", value="Render as-is")
 
         gr.Markdown("## Select desired render options")