diff --git a/app/service/design/service.py b/app/service/design/service.py index 54cb45b..ac17351 100644 --- a/app/service/design/service.py +++ b/app/service/design/service.py @@ -115,8 +115,8 @@ def process_object(cfg, process_id, total): 'priority': 0, 'image_url': item.result['front_image_url'], 'mask_url': item.result['front_mask_url'], - "gradient_string": item.result['gradient_string'] if 'gradient_string' in item.result.keys() else "" - + "gradient_string": item.result['gradient_string'] if 'gradient_string' in item.result.keys() else "", + 'pattern_image_url': item.result['pattern_image_url'] if 'pattern_image_url' in item.result.keys() else None, }) items_response['layers'].append({ 'image_category': f"{item.result['name']}_back", @@ -125,8 +125,8 @@ def process_object(cfg, process_id, total): 'priority': 0, 'image_url': item.result['back_image_url'], 'mask_url': item.result['back_mask_url'], - "gradient_string": item.result['gradient_string'] if 'gradient_string' in item.result.keys() else "" - + "gradient_string": item.result['gradient_string'] if 'gradient_string' in item.result.keys() else "", + 'pattern_image_url': item.result['pattern_image_url'] if 'pattern_image_url' in item.result.keys() else None, }) items_response['synthesis_url'] = synthesis_single(item.result['front_image'], item.result['back_image']) break