import uuid
from datetime import datetime
from pydantic import BaseModel, Field


class GenerationPointCreate(BaseModel):
    template_id: uuid.UUID
    position: dict = Field(..., description="选区位置信息")
    prompt: str
    model_id: uuid.UUID | None = None
    order: int = 0
    selected_text: str | None = None


class GenerationPointUpdate(BaseModel):
    position: dict | None = None
    prompt: str | None = None
    model_id: uuid.UUID | None = None
    order: int | None = None
    selected_text: str | None = None


class GenerationPointResponse(BaseModel):
    id: uuid.UUID
    template_id: uuid.UUID
    position: dict
    prompt: str
    model_id: uuid.UUID | None = None
    ref_file_path: str | None = None
    order: int
    selected_text: str | None = None
    created_at: datetime
    updated_at: datetime

    model_config = {"from_attributes": True}


class BatchOrderUpdate(BaseModel):
    points: list[dict] = Field(..., description="[{id: uuid, order: int}, ...]")