from pydantic import BaseModel
from typing import Optional

class HRAnalysis(BaseModel):
    Employee_ID: str
    DepartmentType: Optional[str]
    Satisfaction_Score: Optional[float]
    Performance_Score: Optional[int]
    Training_Program_Name: Optional[str]
    Training_Outcome: Optional[str]


class SatisfactionRequest(BaseModel):
    engagement_score: int
    work_life_balance_score: int
    performance_score: int

class PerformanceRequest(BaseModel):
    satisfaction_score: int
    engagement_score: int
    training_duration: int
    training_cost: float

class RetentionRequest(BaseModel):
    satisfaction_score: int
    engagement_score: int
    performance_score: int

class TrainingRequest(BaseModel):
    training_type: int
    training_duration: int
    training_cost: float