sync modelscope

app.py

@@ -7,22 +7,12 @@ import argparse
 import gradio as gr
 from utils import *
 from config import *
-from render import *
-from music21 import converter
+from convert import *
 from transformers import GPT2Config
 import warnings
 warnings.filterwarnings('ignore')
 
 
-def abc_to_midi(abc_content, output_midi_path):
-    # 解析 ABC 格式的乐谱
-    score = converter.parse(abc_content)
-
-    # 将乐谱保存为 MIDI 文件
-    score.write('midi', fp=output_midi_path)
-    return output_midi_path
-
-
 def get_args(parser):
     parser.add_argument('-num_tunes', type=int, default=1,
                         help='the number of independently computed returned tunes')
@@ -172,10 +162,12 @@ def generate_abc(args, region):
     create_dir('./tmp')
     timestamp = time.strftime("%a_%d_%b_%Y_%H_%M_%S", time.localtime())
     out_midi = abc_to_midi(tunes, f'./tmp/[{region}]{timestamp}.mid')
-    pdf_file, jpg_file = midi2jpg(out_midi)
+    out_xml = abc_to_musicxml(tunes, f'./tmp/[{region}]{timestamp}.musicxml')
+    out_mxl = musicxml_to_mxl(f'./tmp/[{region}]{timestamp}.musicxml')
+    pdf_file, jpg_file = mxl2jpg(out_mxl)
     wav_file = midi2wav(out_midi)
 
-    return tunes, out_midi, pdf_file, jpg_file, wav_file
+    return tunes, out_midi, pdf_file, out_xml, out_mxl, jpg_file, wav_file
 
 
 def inference(region):
@@ -194,8 +186,8 @@ with gr.Blocks() as demo:
                 choices=[
                     'Mondstadt', 'Liyue', 'Inazuma', 'Sumeru', 'Fontaine'
                 ],
-                value='Liyue',
-                label='Region'
+                value='Mondstadt',
+                label='Region genre'
             )
             gen_btn = gr.Button("Generate")
 
@@ -203,13 +195,18 @@ with gr.Blocks() as demo:
             wav_output = gr.Audio(label='Audio', type='filepath')
             dld_midi = gr.components.File(label="Download MIDI")
             pdf_score = gr.components.File(label="Download PDF score")
+            dld_xml = gr.components.File(label="Download MusicXML")
+            dld_mxl = gr.components.File(label="Download MXL")
             abc_output = gr.TextArea(label='abc score')
             img_score = gr.Image(label='Staff', type='filepath')
 
     gen_btn.click(
         inference,
         inputs=region_opt,
-        outputs=[abc_output, dld_midi, pdf_score, img_score, wav_output]
+        outputs=[
+            abc_output, dld_midi, pdf_score,
+            dld_xml, dld_mxl, img_score, wav_output
+        ]
     )
 
 demo.launch(share=True)

render.py → convert.py

@@ -3,6 +3,7 @@ import sys
 import fitz
 import subprocess
 from PIL import Image
+from music21 import converter
 from utils import download
 
 if sys.platform.startswith('linux'):
@@ -23,6 +24,26 @@ else:
     mscore = "D:/Program Files/MuseScore 3/bin/MuseScore3.exe"
 
 
+def abc_to_midi(abc_content, output_midi_path):
+    score = converter.parse(abc_content, format='abc')
+    score.write('midi', fp=output_midi_path)
+    return output_midi_path
+
+
+def abc_to_musicxml(abc_content, output_xml_path):
+    score = converter.parse(abc_content, format='abc')
+    score.write('musicxml', fp=output_xml_path)
+    return output_xml_path
+
+
+def musicxml_to_mxl(xml_path):
+    mxl_file = xml_path.replace('.musicxml', '.mxl')
+    command = [mscore, "-o", mxl_file, xml_path]
+    result = subprocess.run(command)
+    print(result)
+    return mxl_file
+
+
 def midi2wav(mid_file: str):
     wav_file = mid_file.replace('.mid', '.wav')
     command = [mscore, "-o", wav_file, mid_file]
@@ -41,12 +62,17 @@ def pdf2img(pdf_path: str):
         # 将页面渲染为图像
         image = page.get_pixmap()
         # 将图像添加到列表
-        images.append(Image.frombytes(
-            "RGB", [image.width, image.height], image.samples)
+        images.append(
+            Image.frombytes(
+                "RGB",
+                [image.width, image.height],
+                image.samples
+            )
         )
     # 竖向合并图像
     merged_image = Image.new(
-        "RGB", (images[0].width, sum(image.height for image in images))
+        "RGB",
+        (images[0].width, sum(image.height for image in images))
     )
     y_offset = 0
     for image in images:
@@ -59,9 +85,9 @@ def pdf2img(pdf_path: str):
     return output_path
 
 
-def midi2jpg(mid_file: str):
-    pdf_score = mid_file.replace('.mid', '.pdf')
-    command = [mscore, "-o", pdf_score, mid_file]
+def mxl2jpg(mxl_file: str):
+    pdf_score = mxl_file.replace('.mxl', '.pdf')
+    command = [mscore, "-o", pdf_score, mxl_file]
     result = subprocess.run(command)
     print(result)
     return pdf_score, pdf2img(pdf_score)

utils.py

@@ -21,8 +21,10 @@ S:2
 B:9
 E:4
 B:9
+T:{region} Fragment
 L:1/8
 M:3/4
+C:Generated by AI
 K:D
 de |"D"'''