from pydantic import BaseModel
from typing import List


class FetchTipsDTO(BaseModel):
    context: str
    question: str
    answer: str
    correct_answer: str


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]