from pydantic import BaseModel
from typing import List, Optional
from .question import Question, Answer
from .checklist import ChecklistItem


class SessionStartResponse(BaseModel):
    session_id: str
    round: int
    questions: List[Question]


class TranscribeResponse(BaseModel):
    transcript: str


class SubmitResponse(BaseModel):
    round: int
    is_complete: bool
    questions: Optional[List[Question]] = None
    round_summary: Optional[str] = None
    checklist_preview: Optional[str] = None
    transcripts: Optional[List[str]] = None


class SessionResultsResponse(BaseModel):
    session_id: str
    checklist: List[ChecklistItem]
    markdown: str