from pydantic import BaseModel, EmailStr
from typing import Optional

class EmailRequest(BaseModel):
    email: EmailStr

class VerifyRequest(BaseModel):
    email: EmailStr
    code: str

class OnboardingRequest(BaseModel):
    user_id: int
    firstName: Optional[str] = None
    lastName: Optional[str] = None
    university: Optional[str] = None
    career: Optional[str] = None
    semester: Optional[int] = None
    strengths: Optional[list] = None
    weaknesses: Optional[list] = None
    studyStyle: Optional[str] = None
    careerInterests: Optional[list] = None
    futureRoles: Optional[list] = None
    skillsToLearn: Optional[list] = None

class UserRequest(BaseModel):
    user_id: int

class MatchRequest(BaseModel):
    user_id: int
    candidate_id: int
    candidate_name: str
    message: Optional[str] = None
    match_type: Optional[str] = 'MENTOR'
    compatibility_score: Optional[int] = 0

class MatchRespondRequest(BaseModel):
    user_id: int
    action: str
    message: Optional[str] = None