from pydantic import BaseModel


class BrandDnaModel(BaseModel):
    image_url: str
    is_brand_dna: bool


class GenerateBrandModel(BaseModel):
    user_id: str
    prompt: str