74 lines
2.9 KiB
Python
74 lines
2.9 KiB
Python
#Importing Libraries
|
|
from imageio_ffmpeg import get_ffmpeg_exe
|
|
from subprocess import run
|
|
from librosa import load
|
|
from noisereduce import reduce_noise
|
|
from soundfile import write
|
|
from basic_pitch.inference import predict_and_save, ICASSP_2022_MODEL_PATH
|
|
from music21.midi import MidiFile, translate, environment
|
|
from os.path import splitext, basename, join, dirname
|
|
from os import makedirs
|
|
from lilypond import executable
|
|
from tempfile import TemporaryDirectory
|
|
from shutil import copy2
|
|
from platform import system
|
|
lilypond_path=str(executable())
|
|
if system()=="Windows" and not lilypond_path.lower().endswith('.exe'):
|
|
lilypond_path+='.exe'
|
|
us=environment.UserSettings()
|
|
us['lilypondPath']=lilypond_path
|
|
|
|
#Main Function
|
|
def media_to_notes(input_path,output_path,lilypond_path=lilypond_path):
|
|
makedirs(dirname(output_path),exist_ok=True)
|
|
file_extension=splitext(input_path)[1]
|
|
file_name=splitext(basename(input_path))[0]
|
|
|
|
#Saving Between Files in Temp Directory
|
|
with TemporaryDirectory() as tmp:
|
|
#Correcting File Format
|
|
if file_extension.lower()!='.wav':
|
|
run([get_ffmpeg_exe(),'-y','-i',input_path,join(tmp,file_name+'.wav')],check=True)
|
|
wav_path=join(tmp,file_name+'.wav')
|
|
else:
|
|
wav_path=input_path
|
|
#Noise Reduction
|
|
y,sr=load(wav_path,sr=None,mono=True)
|
|
noise_reduced_music=reduce_noise(y=y,sr=sr,stationary=True,prop_decrease=0.6)
|
|
write(join(tmp,file_name+'_denoised.wav'),noise_reduced_music,sr)
|
|
|
|
#Making The MIDI
|
|
predict_and_save(
|
|
audio_path_list=[join(tmp,file_name+'_denoised.wav')],
|
|
output_directory=tmp,
|
|
save_midi=True,
|
|
sonify_midi=False,
|
|
save_model_outputs=False,
|
|
save_notes=True,
|
|
model_or_model_path=ICASSP_2022_MODEL_PATH)
|
|
|
|
#Making The Ly
|
|
mf=MidiFile()
|
|
mf.open(join(tmp,file_name+'_denoised_basic_pitch.mid'))
|
|
mf.read()
|
|
mf.close()
|
|
s=translate.midiFileToStream(mf)
|
|
s.write('lily', fp=join(tmp,file_name+'.ly'))
|
|
|
|
#Correcting The Layout
|
|
with open(join(tmp,file_name+'.ly'),'r',encoding='utf-8') as f:
|
|
ly=f.read()
|
|
if r'\include "lilypond-book-preamble.ly"' in ly:
|
|
ly=ly.replace(r'\include "lilypond-book-preamble.ly"',rf'\include "{file_name}_wrapper.ly"')
|
|
else:
|
|
ly=rf'\include "{file_name}_wrapper.ly"'+'\n'+ly
|
|
with open(join(tmp,file_name+'_wrapper.ly'),'w',encoding='utf-8') as f:
|
|
f.write(r'\header { tagline = ##f }')
|
|
with open(join(tmp,file_name+'.ly'),'w',encoding='utf-8') as f:
|
|
f.write(ly)
|
|
|
|
#Making The PDF
|
|
run([lilypond_path,"-I",tmp,"-o",join(tmp,file_name),join(tmp,file_name+'.ly')],check=True)
|
|
copy2(join(tmp,file_name+'.pdf'),output_path)
|
|
print(f"Notes PDF saved to {output_path}.")
|