from pydantic import BaseModel
from typing import Dict, List, Optional


class ClaimRequest(BaseModel):
    claim: str
    model: str = 'logistic'


class EvidenceItem(BaseModel):
    statement: str
    label: str
    similarity: float
    reason: Optional[str] = None


class PredictionResponse(BaseModel):
    prediction: str
    probabilities: Optional[Dict[str, float]] = None
    evidence: Optional[List[EvidenceItem]] = None
    justification: Optional[str] = None