from pydantic import BaseModel
from typing import List


class QueryDTO(BaseModel):
    category: str
    text: str


class DetailsDTO(BaseModel):
    exam_id: str
    date: int
    performance_comment: str
    detailed_summary: str


class WeakAreaDTO(BaseModel):
    area: str
    comment: str


class TrainingContentDTO(BaseModel):
    details: List[DetailsDTO]
    weak_areas: List[WeakAreaDTO]
    queries: List[QueryDTO]


class TipsDTO(BaseModel):
    tip_ids: List[str]