"""Public names for relevance inputs, decisions, and screening results."""

from __future__ import annotations

from dataclasses import dataclass

from ._core.accelerated.compact import (
    ArrowCompactFacts,
    CompactColumnScreenBatch,
    ScreeningResult,
)
from ._core.models import (
    Article,
    ArticleAdmissibilityStatus,
    EvidencePath,
    ExposureFamily,
    ExposureRelevanceStatus,
    RankedCandidate,
    RelevanceDecision,
)

Prediction = RelevanceDecision
RankedPrediction = RankedCandidate
PythonScreeningBatch = CompactColumnScreenBatch
ArrowScreeningBatch = ArrowCompactFacts
ScreeningBatch = PythonScreeningBatch | ArrowScreeningBatch


@dataclass(frozen=True, slots=True)
class BackendInfo:
    """Availability and acceleration details for one backend."""

    name: str
    available: bool
    accelerated: bool
    detail: str


__all__ = [
    "ArrowScreeningBatch",
    "Article",
    "ArticleAdmissibilityStatus",
    "BackendInfo",
    "EvidencePath",
    "ExposureFamily",
    "ExposureRelevanceStatus",
    "Prediction",
    "PythonScreeningBatch",
    "RankedPrediction",
    "ScreeningBatch",
    "ScreeningResult",
]