models.py

from typing import List
from pydantic import BaseModel

class ExtractionResult(BaseModel):
    answers: List[str]

class EvaluationResult(BaseModel):
    total_marks: int
    grade: str
    percentage: float

class EvaluationRequest(BaseModel):
    answer_key: ExtractionResult
    student: ExtractionResult