import uuid
from datetime import datetime
from pydantic import BaseModel


class TaskResponse(BaseModel):
    id: uuid.UUID
    template_id: uuid.UUID
    status: str
    result_file_path: str | None = None
    error_msg: str | None = None
    created_at: datetime
    finished_at: datetime | None = None

    model_config = {"from_attributes": True}


class TaskDetailResponse(BaseModel):
    id: uuid.UUID
    template_id: uuid.UUID
    template_name: str = ""
    status: str
    result_file_path: str | None = None
    error_msg: str | None = None
    created_at: datetime
    finished_at: datetime | None = None
    points: list[dict] = []

    model_config = {"from_attributes": True}


class GenerateResponse(BaseModel):
    task_id: uuid.UUID
    status: str


class SingleTestRequest(BaseModel):
    prompt: str
    model_id: uuid.UUID | None = None
    ref_file_path: str | None = None


class SingleTestResponse(BaseModel):
    result: str