Spaces:
Running
Running
| # ================================================================================================= | |
| # https://huggingface.co/spaces/asigalov61/MIDI-Chords-Mixer | |
| # ================================================================================================= | |
| import time as reqtime | |
| import datetime | |
| from pytz import timezone | |
| import copy | |
| import gradio as gr | |
| import random | |
| from midi_to_colab_audio import midi_to_colab_audio | |
| import TMIDIX | |
| # ================================================================================================= | |
| def Mix_Chords(input_output_as_solo_piano, | |
| input_remove_drums, | |
| input_output_tempo, | |
| input_transform, | |
| input_transpose_to_C4, | |
| input_transpose_value | |
| ): | |
| print('=' * 70) | |
| print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT))) | |
| start_time = reqtime.time() | |
| print('=' * 70) | |
| print('-' * 70) | |
| print('Requested settings:') | |
| print('-' * 70) | |
| print('Output as Solo Piano:', input_output_as_solo_piano) | |
| print('Remove drums:', input_remove_drums) | |
| print('Output tempo:', input_output_tempo) | |
| print('Transform:', input_transform) | |
| print('Transpose to C4:', input_transpose_to_C4) | |
| print('Transpose value:', input_transpose_value) | |
| print('-' * 70) | |
| print('Loading source and target MIDIs...') | |
| #============================================================================================================= | |
| sidx = random.randint(0, len(all_chords_chunks)-1) | |
| src_score = all_chords_chunks[sidx] | |
| sfn = src_score[0] | |
| src_score_chords = src_score[1] | |
| src_escore_notes = [] | |
| for s in src_score_chords: | |
| chord_notes = [['note'] + list(c)[:2] + [(list(c)[2] // 16)+1] + list(c)[3:] + [list(c)[2]] for c in s] | |
| for c in chord_notes[1:]: | |
| c[1] = 0 | |
| src_escore_notes.extend(chord_notes) | |
| src_escore_notes = TMIDIX.recalculate_score_timings(TMIDIX.delta_score_to_abs_score(src_escore_notes)) | |
| src_cscore = TMIDIX.chordify_score([1000, src_escore_notes]) | |
| src_counts = [chunks_lens[sidx]] + chords_drums_counts[sidx] | |
| src_cidxs = all_chords_indexes[sidx] | |
| src_avg_time = TMIDIX.escore_notes_averages(src_escore_notes)[0] | |
| #============================================================================================================= | |
| sidx = random.randint(0, len(all_chords_chunks)-1) | |
| trg_score = all_chords_chunks[sidx] | |
| tfn = trg_score[0] | |
| trg_score_chords = trg_score[1] | |
| trg_escore_notes = [] | |
| for s in trg_score_chords: | |
| chord_notes = [['note'] + list(c)[:2] + [(list(c)[2] // 16)+1] + list(c)[3:] + [list(c)[2]] for c in s] | |
| for c in chord_notes[1:]: | |
| c[1] = 0 | |
| trg_escore_notes.extend(chord_notes) | |
| trg_escore_notes = TMIDIX.recalculate_score_timings(TMIDIX.delta_score_to_abs_score(trg_escore_notes)) | |
| trg_cscore = TMIDIX.chordify_score([1000, trg_escore_notes]) | |
| trg_counts = [chunks_lens[sidx]] + chords_drums_counts[sidx] | |
| trg_cidxs = all_chords_indexes[sidx] | |
| trg_avg_time = TMIDIX.escore_notes_averages(trg_escore_notes)[0] | |
| #============================================================================================================= | |
| print('=' * 70) | |
| print('Done loading source and target MIDIs...!') | |
| print('=' * 70) | |
| print('Mixing...') | |
| max_chords_count = min(src_counts[1], trg_counts[1]) | |
| mixed_score = [] | |
| cidx = 0 | |
| for i, c in enumerate(src_cscore): | |
| if i == src_cidxs[cidx]: | |
| stime = c[0][1] | |
| sdurs = [cc[2] for cc in c] | |
| tchord = copy.deepcopy(trg_cscore[trg_cidxs[cidx]]) | |
| sdurs = sdurs + random.choices(sdurs, k=abs(len(tchord)-len(c))) | |
| for j, n in enumerate(tchord): | |
| n[1] = stime | |
| n[2] = sdurs[j] | |
| mixed_score.extend(tchord) | |
| cidx += 1 | |
| if cidx == max_chords_count: | |
| break | |
| else: | |
| mixed_score.extend(c) | |
| print('=' * 70) | |
| print('Done!') | |
| print('=' * 70) | |
| #=============================================================================== | |
| if input_output_as_solo_piano: | |
| csong = TMIDIX.chordify_score([1000, mixed_score]) | |
| mixed_score = [] | |
| for c in csong: | |
| pitches = [] | |
| for cc in c: | |
| ccc = copy.deepcopy(cc) | |
| if cc[3] != 9: | |
| if cc[4] not in pitches: | |
| ccc[3] = 0 | |
| ccc[6] = 0 | |
| mixed_score.append(ccc) | |
| pitches.append(cc[4]) | |
| else: | |
| mixed_score.append(ccc) | |
| if input_remove_drums: | |
| mixed_score = [e for e in mixed_score if e[3] != 9] | |
| if input_output_tempo == 'Source MIDI': | |
| time_k = src_avg_time / trg_avg_time | |
| mixed_score = TMIDIX.adjust_escore_notes_timings(mixed_score, time_k) | |
| elif input_output_tempo == 'Target MIDI': | |
| time_k = trg_avg_time / src_avg_time | |
| mixed_score = TMIDIX.adjust_escore_notes_timings(mixed_score, time_k) | |
| if input_transform == 'Flip Mix': | |
| mixed_score = TMIDIX.flip_enhanced_score_notes(mixed_score) | |
| elif input_transform == 'Reverse Mix': | |
| mixed_score = TMIDIX.reverse_enhanced_score_notes(mixed_score) | |
| if input_transpose_value != 0: | |
| mixed_score = TMIDIX.transpose_escore_notes(mixed_score, input_transpose_value) | |
| if input_transpose_to_C4: | |
| mixed_score = TMIDIX.transpose_escore_notes_to_pitch(mixed_score) | |
| #=============================================================================== | |
| print('Rendering results...') | |
| print('=' * 70) | |
| print('Sample INTs', mixed_score[:5]) | |
| print('=' * 70) | |
| fn1 = 'MIDI_Chords_Mixer_' + sfn + '_' + tfn | |
| output_score, patches, overflow_patches = TMIDIX.patch_enhanced_score_notes(mixed_score) | |
| detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(output_score, | |
| output_signature = 'MIDI Chords Mixer', | |
| output_file_name = fn1, | |
| track_name='Project Los Angeles', | |
| list_of_MIDI_patches=patches, | |
| timings_multiplier=16 | |
| ) | |
| 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_title = str(fn1) | |
| output_midi_summary = 'Source MIDI MD5: ' + sfn + '\n' | |
| output_midi_summary += 'Target MIDI MD5: ' + tfn | |
| output_midi = str(new_fn) | |
| output_audio = (16000, audio) | |
| for o in output_score: | |
| o[1] *= 16 | |
| o[2] *= 16 | |
| output_plot = TMIDIX.plot_ms_SONG(output_score, plot_title=output_midi_title, return_plt=True) | |
| print('Output MIDI file name:', output_midi) | |
| print('Output MIDI title:', output_midi_title) | |
| print('Output MIDI summary:', output_midi_summary) | |
| 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_title, output_midi_summary, 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" | |
| #===================================================================== | |
| all_chords_chunks = TMIDIX.Tegridy_Any_Pickle_File_Reader('Los_Angeles_Chords_Songs_MIDI_Dataset_79898_128_256') | |
| print('=' * 70) | |
| print('Prepping loaded data...') | |
| #===================================================================== | |
| chunks_lens = [len(a[1]) for a in all_chords_chunks] | |
| #===================================================================== | |
| chords_drums_counts = [] | |
| for a in all_chords_chunks: | |
| score = a[1] | |
| ccount = 0 | |
| dcount = 0 | |
| for s in score: | |
| if any(a[2] < 80 for a in s): | |
| ccount += 1 | |
| if all(a[2] == 128 for a in s): | |
| dcount += 1 | |
| chords_drums_counts.append([ccount, dcount]) | |
| #===================================================================== | |
| all_chords_indexes = [] | |
| for a in all_chords_chunks: | |
| score = a[1] | |
| chords_indexes = [] | |
| for i, s in enumerate(score): | |
| if any(a[2] < 80 for a in s): | |
| chords_indexes.append(i) | |
| all_chords_indexes.append(chords_indexes) | |
| #===================================================================== | |
| print('Done!') | |
| print('=' * 70) | |
| #===================================================================== | |
| app = gr.Blocks() | |
| with app: | |
| gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>MIDI Chords Mixer</h1>") | |
| gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Mix chords from one MIDI to another MIDI</h1>") | |
| gr.Markdown( | |
| "\n\n" | |
| "This is a demo for TMIDIX Python module from tegridy-tools and Los Angeles MIDI Dataset\n\n" | |
| "Check out [tegridy-tools](https://github.com/asigalov61/tegridy-tools) on GitHub!\n\n" | |
| "Check out [Los Angeles MIDI Dataset](https://github.com/asigalov61/Los-Angeles-MIDI-Dataset) on GitHub!\n\n" | |
| ) | |
| gr.Markdown("## Select mixing options") | |
| input_output_as_solo_piano = gr.Checkbox(label="Output as Solo Piano", value=False) | |
| input_remove_drums = gr.Checkbox(label="Remove drums from output", value=False) | |
| input_output_tempo = gr.Radio(["Source MIDI", "Target MIDI", "MIDI Mix"], value="MIDI Mix", label="Output tempo") | |
| input_transform = gr.Radio(["As-is", "Flip Mix", "Reverse Mix"], value="As-is", label="Transform") | |
| input_transpose_value = gr.Slider(-12, 12, value=0, step=1, label="Transpose value") | |
| input_transpose_to_C4 = gr.Checkbox(label="Transpose to C4", value=False) | |
| run_btn = gr.Button("Mix Chords", variant="primary") | |
| gr.Markdown("## Mixing results") | |
| 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="mp3", 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(Mix_Chords, [input_output_as_solo_piano, | |
| input_remove_drums, | |
| input_output_tempo, | |
| input_transform, | |
| input_transpose_to_C4, | |
| input_transpose_value | |
| ], | |
| [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot]) | |
| app.queue().launch() |