from pydantic import BaseModel


class GenerateImageModel(BaseModel):
    tasks_id: str
    prompt: str
    image_url: str


class GenerateSingleLogoImageModel(BaseModel):
    tasks_id: str
    prompt: str
    seed: str


class GenerateProductImageModel(BaseModel):
    tasks_id: str
    prompt: str
    image_url: str