app/models/user.py

from pydantic import BaseModel, EmailStr

class UserCreate(BaseModel):
    full_name: str
    email: EmailStr
    password: str

class UserLogin(BaseModel):
    email: EmailStr
    password: str

class User(BaseModel):
    email: EmailStr
    hashed_password: str
    role: str
    companies: list[str]
    chatbots: list[str]

class UserOut(BaseModel):
    email: EmailStr
    role: str
    companies: list[str]
    chatbots: list[str]

class SecurityQuestionsPayload(BaseModel):
    user_id: str
    questions: dict

class SecurityAnswerPayload(BaseModel):
    email: str
    answers: dict[str,str]

class ForgotPasswordEmailPayload(BaseModel):
    email: EmailStr

class ForgotPasswordResetPayload(BaseModel):
    email: EmailStr
    new_password: str