app/schemas/generate_image.py

from pydantic import BaseModel


class GenerateMultiViewModel(BaseModel):
    tasks_id: str
    image_url: str


class GenerateImageModel(BaseModel):
    tasks_id: str
    prompt: str
    image_url: str
    mode: str
    category: str
    gender: str
    version: str


class GenerateSingleLogoImageModel(BaseModel):
    tasks_id: str
    prompt: str
    seed: str


class GenerateProductImageModel(BaseModel):
    tasks_id: str
    prompt: str
    image_url: str
    image_strength: float
    product_type: str


class GenerateRelightImageModel(BaseModel):
    tasks_id: str
    prompt: str
    image_url: str
    direction: str
    product_type: str


"""
    batch generate image
"""


class BatchGenerateProductImageModel(BaseModel):
    tasks_id: str
    prompt: str
    image_url: str
    image_strength: float
    product_type: str
    batch_size: int


class BatchGenerateRelightImageModel(BaseModel):
    tasks_id: str
    prompt: str
    image_url: str
    direction: str
    product_type: str
    batch_size: int
feat generate 迁移 2024-04-15 18:07:25 +08:00			`from pydantic import BaseModel`


feat（新功能）: sketch 多视角图生成功能接口 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-01-08 15:44:59 +08:00			`class GenerateMultiViewModel(BaseModel):`
			`tasks_id: str`
			`image_url: str`


feat generate 迁移 2024-04-15 18:07:25 +08:00			`class GenerateImageModel(BaseModel):`
			`tasks_id: str`
feat generate 升级 attribute retrieve 迁移 2024-04-16 15:51:03 +08:00			`prompt: str`
			`image_url: str`
feat generate product image 部署 2024-06-12 15:32:25 +08:00			`mode: str`
			`category: str`
			`gender: str`
feat generate img fast version 部署 fix 2024-12-01 14:24:48 +08:00			`version: str`
feat generate slogan \| to product image \| slogan 接口部署 2024-05-30 15:01:39 +08:00

			`class GenerateSingleLogoImageModel(BaseModel):`
			`tasks_id: str`
			`prompt: str`
feat generate single logo 部署 2024-06-03 11:38:20 +08:00			`seed: str`
feat generate slogan \| to product image \| slogan 接口部署 2024-05-30 15:01:39 +08:00

			`class GenerateProductImageModel(BaseModel):`
			`tasks_id: str`
			`prompt: str`
			`image_url: str`
feat generate to product image 新增 image_strength参数 fix 2024-06-28 16:58:52 +08:00			`image_strength: float`
feat product image 新增product type 参数，解决single item 无法检测头部的问题 fix 2024-07-04 14:14:57 +08:00			`product_type: str`
feat 更新响应模板 fix 2024-06-17 10:45:45 +08:00

			`class GenerateRelightImageModel(BaseModel):`
			`tasks_id: str`
			`prompt: str`
			`image_url: str`
feat relight 增加打光方向 fix 2024-07-03 10:26:40 +08:00			`direction: str`
feat fix relight 新增single item模式 2024-07-05 15:45:48 +08:00			`product_type: str`
feat（新功能）: generate product relight pose_transform 开发，设置batch generate 的优先级为100 ，single generate 的优先级为1 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-04-21 10:04:40 +08:00

			`"""`
			`batch generate image`
			`"""`


			`class BatchGenerateProductImageModel(BaseModel):`
			`tasks_id: str`
			`prompt: str`
			`image_url: str`
			`image_strength: float`
			`product_type: str`
			`batch_size: int`


			`class BatchGenerateRelightImageModel(BaseModel):`
			`tasks_id: str`
			`prompt: str`
			`image_url: str`
			`direction: str`
			`product_type: str`
			`batch_size: int`