core/__init__.py

"""
Core utilities for Design System Extractor v2.
"""

from core.token_schema import (
    TokenSource,
    Confidence,
    Viewport,
    PageType,
    ColorToken,
    TypographyToken,
    SpacingToken,
    RadiusToken,
    ShadowToken,
    ExtractedTokens,
    NormalizedTokens,
    FinalTokens,
    WorkflowState,
)

from core.color_utils import (
    parse_color,
    normalize_hex,
    get_contrast_ratio,
    check_wcag_compliance,
    generate_color_ramp,
    generate_accessible_ramp,
    categorize_color,
    suggest_color_name,
)

# HF Inference is imported lazily to avoid circular imports
# Use: from core.hf_inference import get_inference_client

__all__ = [
    # Enums
    "TokenSource",
    "Confidence",
    "Viewport",
    "PageType",
    # Token models
    "ColorToken",
    "TypographyToken",
    "SpacingToken",
    "RadiusToken",
    "ShadowToken",
    # Result models
    "ExtractedTokens",
    "NormalizedTokens",
    "FinalTokens",
    "WorkflowState",
    # Color utilities
    "parse_color",
    "normalize_hex",
    "get_contrast_ratio",
    "check_wcag_compliance",
    "generate_color_ramp",
    "generate_accessible_ramp",
    "categorize_color",
    "suggest_color_name",
]