2nd commit

2026-05-20 03:14:42 +03:30
parent b5a838bbd7
commit 987b17f63e
2 changed files with 0 additions and 0 deletions
@@ -0,0 +1,2 @@
+from .basic_converter import media_to_notes
+__all__=["media_to_notes"]
@@ -0,0 +1,73 @@
+#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}.")