models.py

from __future__ import annotations
from pydantic import BaseModel
from typing import List,  Optional

class Employee(BaseModel):
    id: str
    name: str
    role: str   
    department: str
    level: str
    project_id: List[str]
    hourly_rate: Optional[float] = None

class EmployeeSkillGap(BaseModel):
    employee_id: str
    skill: str


class TrainingMaterial(BaseModel):
    id: str
    title: str
    url: str
    skills_covered: List[str]
    cost: Optional[float] = None
    duration_hours: Optional[float] = None

class EmployeeTrainingPlan(BaseModel):
    employee_name: str
    role: str
    skills_gaps: List[EmployeeSkillGap]
    training_plan: List[TrainingMaterial]

class AnalysisResult(BaseModel):
    project_id: str
    project_name: str
    budget: int
    deadline: str
    status: str
    team: List[EmployeeTrainingPlan]
    corny_summary: str