TuKouKouClient/python/models.py

from middleware import *


def success(data):
    """
    成功
    """
    return {
        "code": 0,
        "msg": "请求成功",
        "data": data,
    }


def failed(msg):
    """
    失败
    """
    return {
        "code": -1,
        "msg": msg,
    }


class CheckSelectImages(BaseModel):
    # 检查目录
    path_type: int = Field(default=0, description="地址类型;0图像;1目录")
    path: str = Field(default=None, description="目录地址")
    image_list: list[str] = Field(default=[], description="图像地址")


class SegmentImages(BaseModel):
    # 抠图
    token: str = Field(default=None, description="hlm token信息")
    image_type: int = Field(default=0, description="图像类型;0非服装;1服装")
    segment_type: int = Field(default=0, description="抠图精细度;0普通;1精细")
    output_type: int = Field(default=0, description="抠图精细度;0透明;1白底")
    need_cutout_images: list = Field(default=None, description="图像地址集合")


class RequestHlm(BaseModel):
    # 抠图
    token: str = Field(default=None, description="hlm token信息")