From 679c854b01fd1cdc4e2ae9f90cd3a08ae21c8166 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Wed, 24 Apr 2024 11:24:38 +0800 Subject: [PATCH] =?UTF-8?q?feat=20generate=20sketch=20=E6=96=B0=E5=A2=9E?= =?UTF-8?q?=20=E6=9C=8D=E8=A3=85=E7=B1=BB=E5=88=AB=E8=AF=86=E5=88=AB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/generate_image/service.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/app/service/generate_image/service.py b/app/service/generate_image/service.py index 831c2ce..04c3734 100644 --- a/app/service/generate_image/service.py +++ b/app/service/generate_image/service.py @@ -52,7 +52,7 @@ class GenerateImage: self.batch_size = 1 self.category = request_data.category self.index = 0 - self.generate_data = {'tasks_id': self.tasks_id, 'status': 'PENDING', 'message': "pending", 'data': {"image_url": "", "category": ""}} + self.generate_data = {'tasks_id': self.tasks_id, 'status': 'PENDING', 'message': "pending", 'image_url': '', 'category': ''} self.redis_client.set(self.tasks_id, json.dumps(self.generate_data)) self.redis_client.expire(self.tasks_id, 600) @@ -85,7 +85,7 @@ class GenerateImage: is_smudge, not_smudge_image = stain_detection(remove_bg_image) # 类型识别 category, scores, not_smudge_image = generate_category_recognition(image_result) - self.generate_data['data']['category'] = str(category) + self.generate_data['category'] = str(category) image_result = not_smudge_image if is_smudge: # 无污点 image_result = adjust_contrast(image_result) @@ -93,12 +93,12 @@ class GenerateImage: # logger.info(f"upload image SUCCESS ： {image_url}") self.generate_data['status'] = "SUCCESS" self.generate_data['message'] = "success" - self.generate_data['data']['image_url'] = str(image_url) + self.generate_data['image_url'] = str(image_url) self.redis_client.set(self.tasks_id, json.dumps(self.generate_data)) else: # 有污点 self.generate_data['status'] = "SUCCESS" self.generate_data['message'] = "success" - self.generate_data['data']['image_url'] = str(GI_SYS_IMAGE_URL) + self.generate_data['image_url'] = str(GI_SYS_IMAGE_URL) self.redis_client.set(self.tasks_id, json.dumps(self.generate_data)) # logger.info(f"stain_detection result : {self.generate_data}")