from enum import Enum

from pydantic import BaseModel


class FileType(str, Enum):
    IMAGE = "image"
    VIDEO = "video"
    AUDIO = "audio"
    DOCUMENT = "document"
    FILE = "file"


class FileUploadResponse(BaseModel):
    id: int
    analysis_run_id: str
    file_type: FileType
    file_name: str
    file_url: str
    size_bytes: int | None = None


class FileListItem(BaseModel):
    id: int
    file_type: FileType
    file_name: str
    file_url: str
    size_bytes: int | None = None
    created_at: str