asigalov61's picture
Update app.py
f742df5 verified
raw
history blame
12.1 kB
# =================================================================================================
# 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_song.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 + '\n'
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...')
#=====================================================================
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(
"![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.MIDI-Chords-Mixer&style=flat)\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="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()