mirror of
https://github.com/ParisNeo/lollms-webui.git
synced 2024-12-26 23:41:03 +00:00
105 lines
3.3 KiB
Python
105 lines
3.3 KiB
Python
"""
|
|
project: lollms_webui
|
|
file: lilypond_execution_engine.py
|
|
author: LilyPond integration
|
|
description:
|
|
This is a utility for executing LilyPond code in LOLLMS
|
|
"""
|
|
|
|
import shutil
|
|
import subprocess
|
|
import time
|
|
from pathlib import Path
|
|
|
|
import pipmaster as pm
|
|
from ascii_colors import trace_exception
|
|
from lollms.client_session import Client
|
|
from lollms.utilities import discussion_path_to_url, show_yes_no_dialog
|
|
|
|
from lollms_webui import LOLLMSWebUI
|
|
|
|
lollmsElfServer: LOLLMSWebUI = LOLLMSWebUI.get_instance()
|
|
|
|
|
|
def check_and_install_lilypond():
|
|
"""Check if LilyPond is installed and install it if needed"""
|
|
if not pm.is_installed("lilypond"):
|
|
if not show_yes_no_dialog(
|
|
"Installation", "LilyPond is not installed. Do you want to install it?"
|
|
):
|
|
return {"status": False, "error": "User refused LilyPond installation!"}
|
|
try:
|
|
pm.install("lilypond")
|
|
return {"status": True}
|
|
except Exception as ex:
|
|
return {"status": False, "error": f"Failed to install LilyPond: {str(ex)}"}
|
|
return {"status": True}
|
|
|
|
|
|
def execute_lilypond(code, client: Client, message_id):
|
|
"""Execute LilyPond code and return the result"""
|
|
try:
|
|
# Check LilyPond installation
|
|
check_result = check_and_install_lilypond()
|
|
if not check_result["status"]:
|
|
return {"output": check_result["error"], "execution_time": 0}
|
|
|
|
# Start timer
|
|
start_time = time.time()
|
|
|
|
# Import LilyPond after installation check
|
|
import lilypond
|
|
|
|
# Create work directory in discussion folder
|
|
root_folder = client.discussion.discussion_folder
|
|
root_folder.mkdir(parents=True, exist_ok=True)
|
|
|
|
# Create LilyPond file
|
|
ly_file = root_folder / f"score_{message_id}.ly"
|
|
ly_file.write_text(code, encoding="utf8")
|
|
|
|
# Get the PDF and MIDI outputs
|
|
pdf_file = ly_file.with_suffix(".pdf")
|
|
midi_file = ly_file.with_suffix(".mid")
|
|
|
|
# Compile the file
|
|
subprocess.run(
|
|
[lilypond.executable(), str(ly_file)], check=True, cwd=root_folder
|
|
)
|
|
|
|
# Create links to files
|
|
if (
|
|
not "http" in lollmsElfServer.config.host
|
|
and not "https" in lollmsElfServer.config.host
|
|
):
|
|
host = "http://" + lollmsElfServer.config.host
|
|
else:
|
|
host = lollmsElfServer.config.host
|
|
|
|
pdf_link = (
|
|
f"{host}:{lollmsElfServer.config.port}{discussion_path_to_url(pdf_file)}"
|
|
)
|
|
midi_link = (
|
|
f"{host}:{lollmsElfServer.config.port}{discussion_path_to_url(midi_file)}"
|
|
)
|
|
|
|
# Create output HTML
|
|
output = f"""
|
|
<div>
|
|
<h3>LilyPond Output:</h3>
|
|
<p><a href="{pdf_link}" target="_blank">View PDF Score</a></p>
|
|
<p><a href="{midi_link}" target="_blank">Download MIDI</a></p>
|
|
<embed src="{pdf_link}" type="application/pdf" width="100%" height="600px">
|
|
</div>
|
|
"""
|
|
|
|
execution_time = time.time() - start_time
|
|
return {"output": output, "execution_time": execution_time}
|
|
|
|
except Exception as ex:
|
|
trace = trace_exception(ex)
|
|
return {
|
|
"output": f"Error executing LilyPond code:\n{trace}",
|
|
"execution_time": 0,
|
|
}
|