from __future__ import annotations

from typing import Dict, List, Optional

from pydantic import BaseModel, Field


class UtterancePayload(BaseModel):
    start: float = Field(..., ge=0)
    end: float = Field(..., ge=0)
    text: str
    speaker: Optional[int] = Field(default=None, ge=0)


class TranscriptExportRequest(BaseModel):
    format: str
    utterances: List[UtterancePayload]
    include_timestamps: bool = True
    title: Optional[str] = None


class SummaryExportRequest(BaseModel):
    format: str
    summary: str
    metadata: Optional[Dict[str, str]] = None
    title: Optional[str] = None