sync ms

app.py

@@ -28,19 +28,18 @@ def audio2midi(audio_path: str):
     # Transcriptor
     transcriptor = PianoTranscription(
         device="cuda" if torch.cuda.is_available() else "cpu", checkpoint_path=None
-    )  # device: 'cuda' | 'cpu'
-    # Transcribe and write out to MIDI file
-    basename = os.path.basename(audio_path).split(".")[-2]
-    midi_path = f"./example/{basename}.mid"
+    )
+    # device: 'cuda' | 'cpu' Transcribe and write out to MIDI file
+    midi_path = f"./example/temp.mid"
     # midi_path = audio_path.replace(audio_path.split(".")[-1], "mid")
     transcriptor.transcribe(audio, midi_path)
-    return midi_path
+    return midi_path, os.path.basename(audio_path).split(".")[-2].capitalize()
 
 
 def inference(audio_path: str):
-    midi_path = audio2midi(audio_path)
-    xml_path, title, artist = midi2xml(midi_path)
-    abc = midi2abc(midi_path, title, artist)
+    midi_path, title = audio2midi(audio_path)
+    xml_path = midi2xml(midi_path, title)
+    abc = midi2abc(midi_path, title)
     mxl_path = xml2mxl(xml_path)
     pdf_path, jpg_path = mxl2jpg(mxl_path)
     return midi_path, pdf_path, xml_path, mxl_path, abc, jpg_path

convert.py

@@ -57,25 +57,19 @@ else:
     mscore = "D:/Program Files/MuseScore 3/bin/MuseScore3.exe"
 
 
-def add_title_to_xml(xml_path: str):
+def add_title_to_xml(xml_path: str, title: str):
     """
     给 MIDI 文件添加标题，标题为去后缀的文件名。
     参数：
     - midi_file_path: MIDI 文件的路径
     """
-    # 获取文件名（不带路径）
-    file_name = os.path.basename(xml_path)
-    title = file_name.split(".")[0]
-    artist = "Transcripted by AI"
     # 读取 MIDI 文件
     midi_data = converter.parse(xml_path)
     # 将标题添加到 MIDI 文件中
-    midi_data.metadata.title = title
     midi_data.metadata.movementName = title
-    midi_data.metadata.composer = artist
+    midi_data.metadata.composer = "Transcripted by AI"
     # 保存修改后的 MIDI 文件
     midi_data.write("musicxml", fp=xml_path)
-    return title, artist
 
 
 def xml2abc(xml_path: str):
@@ -96,13 +90,13 @@ def xml2mxl(xml_path: str):
     return mxl_file
 
 
-def midi2xml(mid_file: str):
+def midi2xml(mid_file: str, title: str):
     xml_file = mid_file.replace(".mid", ".musicxml")
     command = [mscore, "-o", xml_file, mid_file]
     result = subprocess.run(command)
-    title, artist = add_title_to_xml(xml_file)
+    add_title_to_xml(xml_file, title)
     print(result)
-    return xml_file, title, artist
+    return xml_file
 
 
 def xml2midi(xml_file: str):

midi2abc.py

@@ -580,7 +580,7 @@ def main(argv):
         print(result)
 
 
-def midi2abc(filename, title, artist):
+def midi2abc(filename, title):
     key = None
     default_len = Fraction(1, 16)
     metre = Fraction(3, 4)
@@ -605,7 +605,7 @@ def midi2abc(filename, title, artist):
         default_len,
         bars_per_line,
         title,
-        artist,
+        "Transcripted by AI",
         "",
         no_triplets,
         no_broken_rythms,