import os
import re
import time
import subprocess
import gradio as gr

os.environ['QT_QPA_PLATFORM']='offscreen' 
# os.environ['MPLCONFIGDIR'] = os.getcwd() + "/configs/"

# 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/{conversation_id}", exist_ok=True)
    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:
        ts = time.time()
        # Write the ABC text to a temporary file
        tmp_abc = f"tmp/{conversation_id}/{ts}.abc"
        with open(tmp_abc, "w") as abc_file:
            abc_file.write(abc_notation[0])
        # Convert abc notation to midi
        tmp_midi = f'tmp/{conversation_id}/{ts}.mid'
        subprocess.run(["abc2midi", str(tmp_abc), "-o", tmp_midi])
        # Convert abc notation to SVG
        svg_file = f'tmp/{conversation_id}/{ts}.svg'
        audio_file = f'tmp/{conversation_id}/{ts}.mp3'
        subprocess.run(["musescore", "-o", svg_file, tmp_midi], capture_output=True, text=True)
        subprocess.run(["musescore","-o", audio_file, tmp_midi])
        return svg_file.replace(".svg", "-1.svg"), audio_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()