from . import DocumentLayoutAnalyzer
from . import onnx
from . import pymupdf_util

from pymupdf.layout._build import git_sha
from pymupdf.layout._build import platform_python_implementation
from pymupdf.layout._build import version
from pymupdf.layout._build import version_tuple

import pymupdf


def activate():
    """Create a layout analyzer function using an ONNX model."""
    if callable(pymupdf._get_layout):
        return  # already activated
    MODEL = DocumentLayoutAnalyzer.get_model()

    def _get_layout(*args, **kwargs):
        return MODEL.predict(*args, **kwargs)

    pymupdf._get_layout = _get_layout


activate()