#======================================================
# https://huggingface.co/spaces/asigalov61/MIDI-Aligner
#======================================================

import os.path

import time as reqtime
import datetime
from pytz import timezone

import gradio as gr

import random
import tqdm

from midi_to_colab_audio import midi_to_colab_audio

import TMIDIX

import matplotlib.pyplot as plt

# =================================================================================================
                       
def AlignMIDI(input_midi, 
              bar_dtime, 
              min_adj_dtime
             ):
    
    print('=' * 70)
    print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
    start_time = reqtime.time()

    print('=' * 70)

    fn = os.path.basename(input_midi.name)
    fn1 = fn.split('.')[0]

    print('-' * 70)
    print('Input file name:', fn)
    print('Req bar dur:', bar_dtime)
    print('Req min adj dtime:', min_adj_dtime)
    print('-' * 70)

    print('=' * 70)
    print('Aligning MIDI...')
    print('=' * 70)

    #===============================================================================
    # Loading
    #===============================================================================
    
    raw_score = TMIDIX.midi2single_track_ms_score(input_midi.name)
    
    escore_notes = TMIDIX.advanced_score_processor(raw_score, 
                                                   return_enhanced_score_notes=True,
                                                   apply_sustain=True
                                                  )[0]
    
    if len(escore_notes) > 0:

        #===============================================================================
        # Pre-processing
        #===============================================================================
    
        escore_notes = TMIDIX.augment_enhanced_score_notes(escore_notes, timings_divider=20)

        escore_notes = TMIDIX.remove_duplicate_pitches_from_escore_notes(escore_notes)

        escore_notes = TMIDIX.fix_escore_notes_durations(escore_notes, min_notes_gap=0)
        
        escore_notes = TMIDIX.recalculate_score_timings(escore_notes)
        
        #===============================================================================
        # Aligning MIDI to bars
        #===============================================================================

        output = TMIDIX.advanced_align_escore_notes_to_bars(escore_notes, 
                                                            bar_dtime=bar_dtime,
                                                            dtimes_adj_thresh=min_adj_dtime,
                                                            min_dur_gap=0
                                                           )
        print('=' * 70)
        print('Done!')
        print('=' * 70)
        
        #===============================================================================
        # Rendering
        #===============================================================================
        
        print('Rendering results...')
        
        print('=' * 70)
        print('Sample INTs', output[:12])
        print('=' * 70)
        
        output_score, patches, overflow_patches = TMIDIX.patch_enhanced_score_notes(output)

        detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(output_score,
                                                                  output_signature = 'MIDI Aligner',
                                                                  output_file_name = fn1,
                                                                  track_name='Project Los Angeles',
                                                                  list_of_MIDI_patches=patches,
                                                                  timings_multiplier=20
                                                                  )
        
        new_fn = fn1+'.mid'
                
        
        audio = midi_to_colab_audio(new_fn, 
                                    soundfont_path=soundfont,
                                    sample_rate=16000,
                                    volume_scale=10,
                                    output_for_gradio=True
                                    )
        
        print('Done!')
        print('=' * 70)
    
        #========================================================

        output_midi = str(new_fn)
        output_audio = (16000, audio)
        
        output_plot = TMIDIX.plot_ms_SONG(output_score, 
                                          plot_title=fn1,
                                          timings_multiplier=20,
                                          return_plt=True
                                         )
    
        print('Output MIDI file name:', output_midi)
        print('Output MIDI title:', fn1)
        print('=' * 70) 
        
    
        #========================================================
        
        print('-' * 70)
        print('Req end time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
        print('-' * 70)
        print('Req execution time:', (reqtime.time() - start_time), 'sec')
    
        return output_midi, output_audio, output_plot

# =================================================================================================

if __name__ == "__main__":
    
    PDT = timezone('US/Pacific')
    
    print('=' * 70)
    print('App start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
    print('=' * 70)

    soundfont = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
   
    app = gr.Blocks()
    with app:
        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>MIDI Aligner</h1>")
        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Align any MIDI to bars</h1>")
        gr.Markdown(
            "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.MIDI-Aligner&style=flat)\n\n"
            "This is a demo for TMIDIX Python module from tegridy-tools\n\n"
            "Check out [tegridy-tools](https://github.com/asigalov61/tegridy-tools) on GitHub!\n\n"
        )
        gr.Markdown("## Upload your MIDI or select a sample example MIDI")

        input_midi = gr.File(label="Input MIDI")
        
        bar_dtime = gr.Slider(50, 200, value=200, step=50, label="Delta bar duration")
        min_adj_dtime = gr.Slider(1, 10, value=4, step=1, label="Minimum adjustment delta time")
        
        run_btn = gr.Button("Align", variant="primary")

        gr.Markdown("## Output results")

        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 = run_btn.click(AlignMIDI, [input_midi, 
                                              bar_dtime, 
                                              min_adj_dtime
                                             ],
                                              [output_midi, 
                                               output_audio, 
                                               output_plot
                                              ])

        gr.Examples([["Sharing The Night Together.kar", 200, 4], 
                     ["Deep Relaxation Melody #6.mid", 200, 4],
                    ],
                    [input_midi, 
                     bar_dtime, 
                     min_adj_dtime
                    ],
                    [output_midi, 
                     output_audio, 
                     output_plot
                    ],
                    AlignMIDI,
                    cache_examples=True,
                )
        
        app.launch()