Configure Verovio resources in Theme Lab

apps/theme_lab/app.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import argparse
+from contextlib import asynccontextmanager
 from html import escape
 import os
 import pickle
@@ -58,7 +59,13 @@ _transformer_lock = threading.Lock()
 _transformer_checkpoint = None
 
 
-app = FastAPI(title="Theme Lab", version="0.1.0")
+@asynccontextmanager
+async def lifespan(_: FastAPI):
+    start_background_warmup()
+    yield
+
+
+app = FastAPI(title="Theme Lab", version="0.1.0", lifespan=lifespan)
 app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
 app.mount("/svgs", StaticFiles(directory=ROOT / "svgs"), name="svgs")
 app.mount("/midis", StaticFiles(directory=ROOT / "midis"), name="midis")
@@ -122,12 +129,21 @@ def index() -> FileResponse:
 def health() -> dict[str, object]:
     verovio_cli = shutil.which("verovio") is not None
     verovio_python = has_python_verovio_renderer()
+    verovio_resource_path = None
+    if verovio_python:
+        try:
+            import verovio
+
+            verovio_resource_path = configure_verovio_resources(verovio)
+        except Exception:
+            verovio_resource_path = None
     return {
         "ok": True,
         "database": DB_PATH.exists(),
         "verovio": verovio_cli or verovio_python,
         "verovio_cli": verovio_cli,
         "verovio_python": verovio_python,
+        "verovio_resource_path": verovio_resource_path,
         "score_preview_render_version": SCORE_PREVIEW_RENDER_VERSION,
         "markov_cache_entries": len(_markov_cache),
         "default_markov_cache": DEFAULT_MARKOV_CACHE.exists(),
@@ -475,6 +491,34 @@ def has_python_verovio_renderer() -> bool:
     return True
 
 
+def packaged_verovio_resource_path(verovio_module) -> Path | None:
+    module_file = getattr(verovio_module, "__file__", None)
+    if not module_file:
+        return None
+    data_path = Path(module_file).resolve().parent / "data"
+    if not data_path.exists():
+        return None
+    required = ["Bravura.xml", "Leipzig.xml"]
+    if not all((data_path / name).exists() for name in required):
+        return None
+    return data_path
+
+
+def configure_verovio_resources(verovio_module, toolkit=None) -> str | None:
+    data_path = packaged_verovio_resource_path(verovio_module)
+    if data_path is None:
+        return None
+    resource_path = str(data_path)
+    try:
+        if hasattr(verovio_module, "setDefaultResourcePath"):
+            verovio_module.setDefaultResourcePath(resource_path)
+        if toolkit is not None and hasattr(toolkit, "setResourcePath"):
+            toolkit.setResourcePath(resource_path)
+    except Exception:
+        return None
+    return resource_path
+
+
 def musicxml_preview_data(musicxml_data: str, *, strip_ties: bool = False, strip_beams: bool = False) -> str | None:
     try:
         root = ET.fromstring(musicxml_data)
@@ -507,6 +551,7 @@ def render_musicxml_with_python_verovio(
     try:
         import verovio
 
+        configure_verovio_resources(verovio)
         musicxml_data = musicxml_path.read_text(encoding="utf-8")
         if strip_ties or strip_beams:
             musicxml_data = (
@@ -514,6 +559,7 @@ def render_musicxml_with_python_verovio(
                 or musicxml_data
             )
         toolkit = verovio.toolkit()
+        configure_verovio_resources(verovio, toolkit)
         toolkit.setOptions({"scale": 42})
         svg = toolkit.renderData(musicxml_data, {}) if hasattr(toolkit, "renderData") else ""
         if not svg:
@@ -710,7 +756,6 @@ def load_default_transformer_checkpoint() -> None:
         _transformer_checkpoint = checkpoint
 
 
-@app.on_event("startup")
 def start_background_warmup() -> None:
     loaded_precomputed = load_precomputed_markov_backend()
     if not loaded_precomputed and os.environ.get("THEME_LAB_WARM_MARKOV", "").lower() in {"1", "true", "yes"}: