diff --git a/app/api/api_pose_transform.py b/app/api/api_pose_transform.py index 4b66467..fe5fc5a 100644 --- a/app/api/api_pose_transform.py +++ b/app/api/api_pose_transform.py @@ -24,8 +24,7 @@ def pose_transform(request_item: PoseTransformModel, background_tasks: Backgroun { "tasks_id": "123-89", "image_url": "aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png", - "pose_id": "1", - "result_type" : "gif" + "pose_id": "1" } """ try: diff --git a/app/schemas/pose_transform.py b/app/schemas/pose_transform.py index 05db63f..045d8b9 100644 --- a/app/schemas/pose_transform.py +++ b/app/schemas/pose_transform.py @@ -5,4 +5,3 @@ class PoseTransformModel(BaseModel): image_url: str tasks_id: str pose_id: str - result_type: str diff --git a/app/service/generate_image/service_pose_transform.py b/app/service/generate_image/service_pose_transform.py index f22cdfb..4892d9a 100644 --- a/app/service/generate_image/service_pose_transform.py +++ b/app/service/generate_image/service_pose_transform.py @@ -39,12 +39,7 @@ class PoseTransformService: self.tasks_id = request_data.tasks_id self.user_id = self.tasks_id[self.tasks_id.rfind('-') + 1:] 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': '', '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.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.expire(self.tasks_id, 600)