Update app.py

app.py

@@ -1,29 +1,5 @@
 #================================================================
-# https://huggingface.co/spaces/asigalov61/Advanced-MIDI-Renderer
-#================================================================
-# Packages:
-#
-#   sudo apt install fluidsynth
-#
-#================================================================
-# Requirements:
-#   
-#   pip install gradio
-#   pip install numpy
-#   pip install scipy
-#   pip install matplotlib
-#   pip install networkx
-#   pip install scikit-learn
-#
-#================================================================
-# Core modules:
-#
-# git clone --depth 1 https://github.com/asigalov61/tegridy-tools
-#
-# import TMIDIX
-# import TPLOTS
-# import midi_to_colab_audio
-#
+# https://huggingface.co/spaces/asigalov61/MIDI-Identification
 #================================================================
 
 import os
@@ -41,23 +17,20 @@ import statistics
 import gradio as gr
 
 import TMIDIX
-import TPLOTS
-
-from midi_to_colab_audio import midi_to_colab_audio
 
 #==========================================================================================================
 
-def Render_MIDI(input_midi, 
-                render_type, 
-                soundfont_bank, 
-                render_sample_rate, 
-                custom_render_patch,
-                render_align,
-                render_transpose_value,
-                render_transpose_to_C4,
-                render_output_as_solo_piano,
-                render_remove_drums 
-                ):
+def ID_MIDI(input_midi, 
+            render_type, 
+            soundfont_bank, 
+            render_sample_rate, 
+            custom_render_patch,
+            render_align,
+            render_transpose_value,
+            render_transpose_to_C4,
+            render_output_as_solo_piano,
+            render_remove_drums 
+            ):
     
     print('*' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -112,165 +85,7 @@ def Render_MIDI(input_midi,
     print('=' * 70)
     print('Processing...Please wait...')
 
-    output_score = copy.deepcopy(escore)
-
-    if render_type == "Extract melody":
-        output_score = TMIDIX.add_melody_to_enhanced_score_notes(escore, return_melody=True)
-        output_score = TMIDIX.recalculate_score_timings(output_score)
-
-    elif render_type == "Flip":
-        output_score = TMIDIX.flip_enhanced_score_notes(escore)
-        
-    elif render_type == "Reverse":
-        output_score = TMIDIX.reverse_enhanced_score_notes(escore)
-        
-    elif render_type == 'Repair Chords':
-        fixed_cscore = TMIDIX.advanced_check_and_fix_chords_in_chordified_score(cscore)[0]
-        output_score = TMIDIX.flatten(fixed_cscore)
-
-    elif render_type == "Add Drums Track":
-        nd_escore = [e for e in escore if e[3] != 9]
-        nd_escore = TMIDIX.augment_enhanced_score_notes(nd_escore)
-        output_score = TMIDIX.advanced_add_drums_to_escore_notes(nd_escore)
-
-        for e in output_score:
-            e[1] *= 16
-            e[2] *= 16
-
-    print('Done processing!')
-    print('=' * 70)
-    
-    print('Repatching if needed...')
-    print('=' * 70)
-
-    if -1 < custom_render_patch < 128:
-        for e in output_score:
-            if e[3] != 9:
-                e[6] = custom_render_patch
-        
-    print('Done repatching!')
-    print('=' * 70)
-    
-    print('Sample output events', output_score[:5])
-    print('=' * 70)
-    print('Final processing...')
-    
-    new_fn = fn1+'.mid'
-
-    if render_type != "Render as-is":
-
-        if render_transpose_value != 0:
-            output_score = TMIDIX.transpose_escore_notes(output_score, render_transpose_value)
-
-        if render_transpose_to_C4:
-            output_score = TMIDIX.transpose_escore_notes_to_pitch(output_score)
-
-        if render_align == "Start Times":
-            output_score = TMIDIX.recalculate_score_timings(output_score)
-            output_score = TMIDIX.align_escore_notes_to_bars(output_score)
-    
-        elif render_align == "Start Times and Durations":
-            output_score = TMIDIX.recalculate_score_timings(output_score)
-            output_score = TMIDIX.align_escore_notes_to_bars(output_score, trim_durations=True)
-    
-        elif render_align == "Start Times and Split Durations":
-            output_score = TMIDIX.recalculate_score_timings(output_score)
-            output_score = TMIDIX.align_escore_notes_to_bars(output_score, split_durations=True)
-
-        if render_type == "Longest Repeating Phrase":
-            zscore = TMIDIX.recalculate_score_timings(output_score)
-            lrno_score = TMIDIX.escore_notes_lrno_pattern_fast(zscore)
-
-            if lrno_score is not None:
-                output_score = lrno_score
-
-            else:
-                output_score = TMIDIX.recalculate_score_timings(TMIDIX.escore_notes_middle(output_score, 50))
-
-        if render_type == "Multi-Instrumental Summary":
-            zscore = TMIDIX.recalculate_score_timings(output_score)
-            c_escore_notes = TMIDIX.compress_patches_in_escore_notes_chords(zscore)
-            
-            if len(c_escore_notes) > 128:
-                cmatrix = TMIDIX.escore_notes_to_image_matrix(c_escore_notes, filter_out_zero_rows=True, filter_out_duplicate_rows=True)
-                smatrix = TPLOTS.square_image_matrix(cmatrix, num_pca_components=max(1, min(5, len(c_escore_notes) // 128)))
-                output_score = TMIDIX.image_matrix_to_original_escore_notes(smatrix)
-                
-                for o in output_score:
-                    o[1] *= 250
-                    o[2] *= 250            
-
-        if render_output_as_solo_piano:
-            output_score = TMIDIX.solo_piano_escore_notes(output_score, keep_drums=True)
-            
-        if render_remove_drums:
-            output_score = TMIDIX.strip_drums_from_escore_notes(output_score)
-            
-        if render_type == "Solo Piano Summary":
-            sp_escore_notes = TMIDIX.solo_piano_escore_notes(output_score, keep_drums=False)
-            zscore = TMIDIX.recalculate_score_timings(sp_escore_notes)
-
-            if len(zscore) > 128:
-                
-                bmatrix = TMIDIX.escore_notes_to_binary_matrix(zscore)
-                cmatrix = TMIDIX.compress_binary_matrix(bmatrix, only_compress_zeros=True)
-                smatrix = TPLOTS.square_binary_matrix(cmatrix, interpolation_order=max(1, min(5, len(zscore) // 128)))
-                output_score = TMIDIX.binary_matrix_to_original_escore_notes(smatrix)
-    
-                for o in output_score:
-                    o[1] *= 200
-                    o[2] *= 200
-            
-        SONG, patches, overflow_patches = TMIDIX.patch_enhanced_score_notes(output_score)
-                    
-        detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(SONG,
-                                                                  output_signature = 'Advanced MIDI Renderer',
-                                                                  output_file_name = fn1,
-                                                                  track_name='Project Los Angeles',
-                                                                  list_of_MIDI_patches=patches
-                                                                  )
 
-    else:
-        with open(new_fn, 'wb') as f:
-            f.write(fdata)
-            f.close()
-            
-    if soundfont_bank in ["Super GM",
-                            "Orpheus GM",
-                            "Live HQ GM",
-                            "Nice Strings + Orchestra", 
-                            "Real Choir", 
-                            "Super Game Boy", 
-                            "Proto Square"
-                            ]:
-        
-        sf2bank = ["Super GM",
-                    "Orpheus GM",
-                    "Live HQ GM",
-                    "Nice Strings + Orchestra", 
-                    "Real Choir", 
-                    "Super Game Boy", 
-                    "Proto Square"
-                    ].index(soundfont_bank)
-    
-    else:
-        sf2bank = 0
-
-    if render_sample_rate in ["16000", "32000", "44100"]:
-        srate = int(render_sample_rate)
-    
-    else:
-        srate = 16000
-
-    print('-' * 70)
-    print('Generating audio with SF2 bank', sf2bank, 'and', srate, 'Hz sample rate')
-    
-    audio = midi_to_colab_audio(new_fn, 
-                        soundfont_path=soundfonts[sf2bank],
-                        sample_rate=srate,
-                        volume_scale=10,
-                        output_for_gradio=True
-                        )
 
     print('-' * 70)
 
@@ -287,7 +102,7 @@ def Render_MIDI(input_midi,
     output_midi = str(new_fn)
     output_audio = (srate, audio)
     
-    output_plot = TMIDIX.plot_ms_SONG(output_score, plot_title=output_midi, return_plt=True)
+    output_plot = None
 
     print('Output MIDI file name:', output_midi)
     print('Output MIDI title:', output_midi_title)
@@ -316,24 +131,14 @@ if __name__ == "__main__":
     print('App start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
     print('=' * 70)
 
-    soundfonts = ["SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2",
-                  "Orpheus_18.06.2020.sf2",
-                  "Live HQ Natural SoundFont GM.sf2",
-                  "Nice-Strings-PlusOrchestra-v1.6.sf2", 
-                  "KBH-Real-Choir-V2.5.sf2", 
-                  "SuperGameBoy.sf2", 
-                  "ProtoSquare.sf2"
-                 ]
-
     app = gr.Blocks()
     
     with app:
         
-        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Advanced MIDI Renderer</h1>")
-        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Transform and render any MIDI</h1>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>MIDI Identification</h1>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Identify any MIDI</h1>")
         
-        gr.Markdown("![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Advanced-MIDI-Renderer&style=flat)\n\n"
-                    "This is a demo for tegridy-tools\n\n"
+        gr.Markdown("This is a demo for tegridy-tools\n\n"
                     "Please see [tegridy-tools](https://github.com/asigalov61/tegridy-tools) GitHub repo for more information\n\n"
                    )
         
@@ -341,93 +146,19 @@ if __name__ == "__main__":
         
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"], type="filepath")
 
-        gr.Markdown("## Select desired Sound Font bank and render sample rate")
-
-        soundfont_bank = gr.Radio(["Super GM",
-                                   "Orpheus GM",
-                                   "Live HQ GM",
-                                   "Nice Strings + Orchestra", 
-                                   "Real Choir", 
-                                   "Super Game Boy", 
-                                   "Proto Square"
-                                  ], 
-                                  label="SoundFont bank", 
-                                  value="Super GM"
-                                 )
-
-        render_sample_rate = gr.Radio(["16000", 
-                                       "32000", 
-                                       "44100"
-                                      ], 
-                                      label="MIDI audio render sample rate", 
-                                      value="16000"
-                                     )
-
-        gr.Markdown("## Select desired render type")
-
-        render_type = gr.Radio(["Render as-is", 
-                                "Custom render", 
-                                "Extract melody", 
-                                "Flip", 
-                                "Reverse", 
-                                "Repair Chords",
-                                "Longest Repeating Phrase",
-                                "Multi-Instrumental Summary",
-                                "Solo Piano Summary",
-                                "Add Drums Track"
-                               ], 
-                               label="Render type", 
-                               value="Render as-is"
-                              )
-
-        gr.Markdown("## Select custom render options")
-
-        custom_render_patch = gr.Slider(-1, 127, value=-1, label="Custom render MIDI patch")
-        
-        render_align = gr.Radio(["Do not align", 
-                                 "Start Times", 
-                                 "Start Times and Durations", 
-                                 "Start Times and Split Durations"
-                                ], 
-                                label="Align output to bars", 
-                                value="Do not align"
-                               )        
-        
-        render_transpose_value = gr.Slider(-12, 12, value=0, step=1, label="Transpose value")
-        render_transpose_to_C4 = gr.Checkbox(label="Transpose to C4", value=False)
-
-
-        render_output_as_solo_piano = gr.Checkbox(label="Output as Solo Piano", value=False)
-        render_remove_drums = gr.Checkbox(label="Remove drums", value=False)
-        
         submit = gr.Button("Render MIDI", variant="primary")
 
-        gr.Markdown("## Render results")
+        gr.Markdown("## MIDI identification results")
         
         output_midi_md5 = gr.Textbox(label="Output MIDI md5 hash")
         output_midi_title = gr.Textbox(label="Output MIDI title")
         output_midi_summary = gr.Textbox(label="Output MIDI summary")
-        output_audio = gr.Audio(label="Output MIDI audio", format="wav", elem_id="midi_audio")
-        output_plot = gr.Plot(label="Output MIDI score plot")
-        output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])
         
-        run_event = submit.click(Render_MIDI, [input_midi, 
-                                               render_type, 
-                                               soundfont_bank, 
-                                               render_sample_rate, 
-                                               custom_render_patch,
-                                               render_align,
-                                               render_transpose_value,
-                                               render_transpose_to_C4,
-                                               render_output_as_solo_piano,
-                                               render_remove_drums                                              
+        run_event = submit.click(ID_MIDI, [input_midi,                                         
                                               ],
                                                 [output_midi_md5, 
                                                  output_midi_title, 
                                                  output_midi_summary, 
-                                                 output_midi, 
-                                                 output_audio, 
-                                                 output_plot
                                                 ])
         
     app.queue().launch()