Update app.py

app.py

@@ -0,0 +1,109 @@
+#=====================================================
+# https://huggingface.co/spaces/asigalov61/MIDI-Doctor
+#=====================================================
+
+"""
+Check and heal your MIDI :)
+"""
+
+# -----------------------------
+# CONFIGURATION & GLOBALS
+# -----------------------------
+SEP = '=' * 70
+PDT = timezone('US/Pacific')
+
+# -----------------------------
+# PRINT START-UP INFO
+# -----------------------------
+def print_sep():
+    print(SEP)
+
+print_sep()
+print("MIDI Doctor Gradio App")
+print_sep()
+print("Loading modules...")
+
+import mididoctor
+
+print_sep()
+print("Done loading modules!")
+print_sep()
+
+# -----------------------------
+# MAIN FUNCTION
+# -----------------------------
+
+def Heal_MIDI(input_midi):
+    """
+    Heal MIDI
+    """
+    print_sep()
+    print("Request start time:", datetime.datetime.now(PDT).strftime("%Y-%m-%d %H:%M:%S"))
+
+    print_sep()
+    if input_midi is not None:
+        fn = os.path.basename(input_midi.name)
+        fn1 = fn.split('.')[0]
+        print('Input file name:', fn)
+        print_sep())
+
+        print('Healing MIDI...')
+        mididoctor.heal_midi(input_midi.name)
+
+        out_fname = './healed_midis/'+fn
+
+        print('Done!')
+        print_sep()
+        print("Request end time:", datetime.datetime.now(PDT).strftime("%Y-%m-%d %H:%M:%S"))
+        print_sep()
+        
+        return out_fname
+
+# -----------------------------
+# GRADIO INTERFACE SETUP
+# -----------------------------
+with gr.Blocks() as demo:
+
+    gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>MIDI Doctor</h1>")
+    gr.Markdown("<h1 style='text-align: left; margin-bottom: 1rem'>Check and heal your MIDIs</h1>")
+    gr.HTML("""
+        Check out <a href="https://github.com/asigalov61/mididoctor">MIDI Doctor</a> on GitHub
+        <p>
+            <a href="https://huggingface.co/spaces/asigalov61/MIDI-Doctor?duplicate=true">
+                <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-md.svg" alt="Duplicate in Hugging Face">
+            </a>
+        </p>
+    """)
+
+    gr.Markdown(
+        """
+        ## MIDI Doctor will check and heal
+        * MIDI signature
+        * MIDI channels range
+        * MIDI pitches range
+        * MIDI velocities range
+        * MIDI patch changes range
+        * Duplicate pitches
+        * Excessively short durations
+        * Excessively long durations
+        * Overlapping durations
+        * Text events consistency
+        * Excessively quiet dynamics
+        """
+    )
+
+    gr.Markdown("## Upload your MIDI")
+    input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
+    input_midi.upload()
+    
+    generate_btn = gr.Button("Heal", variant="primary")
+
+    generated_MIDI_file = gr.File(label="Healed MIDI file")
+            
+    generate_btn.click(
+        generate_music,
+        [input_midi],
+        Heal_MIDI
+    )
+
+demo.launch()