from pydantic import BaseModel
from typing import Optional

class AudioInput(BaseModel):
    audio_base64: str

class DetectionResult(BaseModel):
    label: str  # "AI_GENERATED" or "HUMAN"
    confidence: float
    message: Optional[str] = None