from pydantic import BaseModel
class BrandDnaModel(BaseModel):
image_url: str
is_brand_dna: bool
class GenerateBrandModel(BaseModel):
user_id: str
prompt: str