from pydantic import BaseModel, EmailStr
from typing import Optional

class UserRegister(BaseModel):
    username: str
    password: str
    email: Optional[str] = None

class UserLogin(BaseModel):
    username: str
    password: str

class FileRecord(BaseModel):
    filename: str
    content_type: str
    bucket_path: str
    uploaded_by: str
    upload_time: str
    ai_description: Optional[str] = None  # AI 识图后生成的描述