from pydantic import BaseModel
from typing import List

class ImageRequest(BaseModel):
    desc: str
    layout: str
    style: str
    image_number: int
    branch_id: int


class ImageResponse(BaseModel):
    image_urls: List[str]


class ModuleOut(BaseModel):
    module_id: int
    name: str
    url: str

    class Config:
        orm_mode = True