from datetime import datetime
from pydantic import BaseModel


class HTMLGenerateRequest(BaseModel):
    html_content: str


class HTMLGenerateResponse(BaseModel):
    message: str
    unique_id: str
    url: str


class HTMLFileResponse(BaseModel):
    id: int
    unique_id: str
    filename: str
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True