__init__.py

"""
CU-1 UI Element Detector

A powerful UI element detection library for identifying and extracting
information from user interface screenshots.
"""

try:
    # When imported as a proper package
    from .cu1_detector import (
        CU1Detector,
        predict,
        get_predictions_json,
        get_prediction_image,
        get_detector
    )
except Exception:
    # Fallback for direct import context (e.g., pytest collecting project root)
    from cu1_detector import (  # type: ignore
        CU1Detector,
        predict,
        get_predictions_json,
        get_prediction_image,
        get_detector
    )

__version__ = "1.0.0"
__all__ = [
    "CU1Detector",
    "predict",
    "get_predictions_json",
    "get_prediction_image",
    "get_detector"
]