app/schemas/generate_image.py

from typing import List

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
"""


# product任务子项
class ProductItemModel(BaseModel):
    tasks_id: str
    image_strength: float
    prompt: str
    image_url: str
    product_type: str


# product批处理 集合
class BatchGenerateProductImageModel(BaseModel):
    batch_tasks_id: str
    user_id: str
    batch_data_list: List[ProductItemModel]


# relight任务子项
class RelightItemModel(BaseModel):
    tasks_id: str
    prompt: str
    image_url: str
    direction: str
    product_type: str


# relight批处理集合
class BatchGenerateRelightImageModel(BaseModel):
    batch_tasks_id: str
    user_id: str
    batch_data_list: List[RelightItemModel]


"""
    agent tool generate image
"""


class AgentTollGenerateImageModel(BaseModel):
    prompt: str
    category: str
    gender: str
    version: str
    size: int
feat（新功能）: batch generate product 入参回参修改 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-04 15:55:55 +08:00			`from typing import List`

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`
			`"""`


feat（新功能）: batch generate relight 入参回参修改 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-05 15:14:36 +08:00			`# product任务子项`
feat（新功能）: batch generate product 入参回参修改 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-04 15:55:55 +08:00			`class ProductItemModel(BaseModel):`
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			`tasks_id: str`
feat（新功能）: batch generate product 入参回参修改 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-04 15:55:55 +08:00			`image_strength: float`
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			`prompt: str`
			`image_url: str`
			`product_type: str`
feat（新功能）: batch generate product 入参回参修改 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-04 15:55:55 +08:00

feat（新功能）: batch generate relight 入参回参修改 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-05 15:14:36 +08:00			`# product批处理集合`
feat（新功能）: batch generate product 入参回参修改 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-04 15:55:55 +08:00			`class BatchGenerateProductImageModel(BaseModel):`
			`batch_tasks_id: str`
			`user_id: str`
			`batch_data_list: List[ProductItemModel]`
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

feat（新功能）: batch generate relight 入参回参修改 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-05 15:14:36 +08:00			`# relight任务子项`
			`class RelightItemModel(BaseModel):`
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			`tasks_id: str`
			`prompt: str`
			`image_url: str`
			`direction: str`
			`product_type: str`
feat（新功能）: batch generate relight 入参回参修改 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-05 15:14:36 +08:00

			`# relight批处理集合`
			`class BatchGenerateRelightImageModel(BaseModel):`
			`batch_tasks_id: str`
			`user_id: str`
			`batch_data_list: List[RelightItemModel]`
feat（新功能）: 新增agent 工具,图片生成接口 fix（修复bug）: docs（文档变更）: refactor（重构）: test(增加测试): 2025-06-30 11:29:19 +08:00

			`"""`
			`agent tool generate image`
			`"""`


			`class AgentTollGenerateImageModel(BaseModel):`
			`prompt: str`
			`category: str`
			`gender: str`
			`version: str`
			`size: int`