from __future__ import annotations

from typing import Optional

from pydantic import BaseModel


class SanitizeRequest(BaseModel):
    text: str
    model_name: Optional[str] = None


class ModelInfo(BaseModel):
    id: str
    label: str
    provider: str
    params: str
    quality: str
    quality_score: int
    speed: str
    entity_types: list[str]
    description: str


class EntityDetail(BaseModel):
    type: str
    original_value: str
    replacement: str
    start: int
    end: int
    score: Optional[float] = None


class SanitizeStats(BaseModel):
    total_entities: int
    by_type: dict[str, int]
    original_length: int
    sanitized_length: int
    sensitive_ratio: float


class SanitizeResponse(BaseModel):
    original_text: str
    sanitized_text: str
    entities: list[EntityDetail]
    stats: SanitizeStats