import uuid
from datetime import datetime
from typing import Any

from pydantic import BaseModel

from app.models.protocol import ProtocolStatus


class ProtocolResponse(BaseModel):
    id: uuid.UUID
    transcription_id: uuid.UUID
    template_id: uuid.UUID
    content: dict[str, Any]
    raw_text: str
    llm_model: str
    status: ProtocolStatus
    created_at: datetime
    updated_at: datetime

    model_config = {"from_attributes": True}


class ProtocolList(BaseModel):
    items: list[ProtocolResponse]
    total: int


class ProtocolRegenerate(BaseModel):
    template_id: uuid.UUID | None = None