aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat（新功能）: pose transform 逻辑修改

Browse Source 
fix（修复bug）:
docs（文档变更）:
refactor（重构）:
test(增加测试):
This commit is contained in:
zhouchengrong
2025-03-20 10:41:36 +08:00
parent 157f75b0a6
commit 754aee7118
3 changed files with 2 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/api/api_pose_transform.py
View File

@@ -24,8 +24,7 @@ def pose_transform(request_item: PoseTransformModel, background_tasks: Backgroun
{ {
"tasks_id": "123-89", "tasks_id": "123-89",
"image_url": "aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png", "image_url": "aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png",
"pose_id": "1", "pose_id": "1"
"result_type" : "gif"
} }
""" """
try: try:

1
app/schemas/pose_transform.py
View File

@@ -5,4 +5,3 @@ class PoseTransformModel(BaseModel):
image_url: str image_url: str
tasks_id: str tasks_id: str
pose_id: str pose_id: str
result_type: str

7
app/service/generate_image/service_pose_transform.py
View File

@@ -39,12 +39,7 @@ class PoseTransformService:
self.tasks_id = request_data.tasks_id self.tasks_id = request_data.tasks_id
self.user_id = self.tasks_id[self.tasks_id.rfind('-') + 1:] self.user_id = self.tasks_id[self.tasks_id.rfind('-') + 1:]
self.result_type = request_data.result_type self.result_type = request_data.result_type
if self.result_type == "gif": self.gen_product_data = {'tasks_id': self.tasks_id, 'status': 'SUCCESS', 'message': "success", 'gif_url': 'test/mannequin_name.png', 'video_url': 'test/mannequin_name.png', 'type': self.result_type}
self.gen_product_data = {'tasks_id': self.tasks_id, 'status': 'SUCCESS', 'message': "success", 'gif_url': 'test/mannequin_name.png', 'video_url': '', 'type': self.result_type}
elif self.result_type == "video":
self.gen_product_data = {'tasks_id': self.tasks_id, 'status': 'SUCCESS', 'message': "success", 'gif_url': '', 'video_url': 'test/mannequin_name.png', 'type': self.result_type}
elif self.result_type == "all":
self.gen_product_data = {'tasks_id': self.tasks_id, 'status': 'SUCCESS', 'message': "success", 'gif_url': 'test/mannequin_name.png', 'video_url': 'test/mannequin_name.png', 'type': self.result_type}
self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data)) self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data))
self.redis_client.expire(self.tasks_id, 600) self.redis_client.expire(self.tasks_id, 600)