from pydantic import BaseModel, Field


class SingleImageRequest(BaseModel):
    image: str = Field(default="", title="Image", description="Image to work on, must be a Base64 string containing the image's data.")

class SingleImageResponse(BaseModel):
    image: str = Field(default="", title="Image", description="Generated image, a Base64 string containing the image's data.")

class GenerateMaskRequest(SingleImageRequest):
    mask_type: int = Field(default="", title="Mask Type", description="Mask type to work with")