from pydantic import BaseModel
from datetime import datetime
from typing import Optional

# UserCreate: для создания пользователя
class UserCreate(BaseModel):
    username: str
    password: str

# CSVFileUpload: для загрузки CSV файла
class CSVFileUpload(BaseModel):
    file_path: str

# H5ModelCreate: для добавления модели
class H5ModelCreate(BaseModel):
    model_path: str

# ModelStatisticsCreate: для сохранения статистики модели
class ModelStatisticsCreate(BaseModel):
    model_id: int
    accuracy: float
    loss: float
    created_at: Optional[datetime] = None