activity_web/backend/combined_loader.py

from __future__ import annotations

import importlib.util
import sys
from functools import lru_cache
from pathlib import Path
from types import ModuleType
from .startup import ensure_insightface_models_flat


def _pipeline_script_path() -> Path:
    return Path(__file__).resolve().parents[2] / "COMBINED PIPELINE" / "combined_pipeline.py"


@lru_cache(maxsize=1)
def _load_module() -> ModuleType:
    script = _pipeline_script_path()
    spec = importlib.util.spec_from_file_location("combined_pipeline", script)
    if spec is None or spec.loader is None:
        raise RuntimeError(f"Cannot load combined pipeline from {script}")
    module = importlib.util.module_from_spec(spec)
    sys.modules[spec.name] = module
    # Ensure commonly-used InsightFace model folders are flattened
    try:
        ensure_insightface_models_flat(["antelopev2", "buffalo_l"])
    except Exception:
        pass
    spec.loader.exec_module(module)
    return module


@lru_cache(maxsize=1)
def get_combined_pipeline():
    return _load_module().CombinedPipeline()