hf-lin
pdf version
01fd204
raw
history blame
No virus
1.54 kB
import os
os.environ['QT_QPA_PLATFORM']='offscreen'
# os.environ['MPLCONFIGDIR'] = os.getcwd() + "/configs/"
import re
import subprocess
import gradio as gr
import time
from symusic import Score, Synthesizer
import torchaudio
import torch
# for rendering abc notation
default_abc = """
X:1
L:1/8
M:2/4
K:G
|:"G" G>A Bc | dB dB |"C" ce ce |"D7" dB A2 |"G" G>A Bc | dB dB |"Am" cA"D7" FA |"G" AG G2 ::
"Em" g2"D" f>e | de Bd |"C" ce ce |"D7" dB A2 |"G" g2"D" f>e | de Bd |"Am" cA"D7" FA |"G" AG G2 :|
"""
def parse_abc_notation(text='', conversation_id='debug'):
# os.makedirs(f"tmp/", exist_ok=True)
ts = time.time()
abc_pattern = r'(X:\d+\n(?:[^\n]*\n)+)'
abc_notation = re.findall(abc_pattern, text+'\n')
print(f'extract abc block: {abc_notation}')
if abc_notation:
# Convert ABC to midi
s = Score.from_abc(abc_notation[0])
wav_file = f'{ts}.mp3'
audio = Synthesizer().render(s, stereo=True)
torchaudio.save(wav_file, torch.FloatTensor(audio), 44100)
# Convert abc notation to SVG
tmp_midi = f'{ts}.mid'
s.dump_midi(tmp_midi)
svg_file = f'{ts}.pdf'
subprocess.run(["./MuseScore-4.1.1.232071203-x86_64.AppImage", "-f", "-o", svg_file, tmp_midi])
return svg_file, wav_file
else:
return None, None
gradio_app = gr.Interface(
parse_abc_notation,
inputs=["text"],
outputs=[gr.Image(label="svg"), gr.Audio(label="audio")],
title="ABC notation parse",
examples=[default_abc]
)
gradio_app.launch()