feat: 新增design模式 merge,前端CV python 合成

app/service/design_fast/design_generate.py

@@ -6,10 +6,10 @@ import requests
 from minio import Minio
 
 from app.core.config import settings
-from app.service.design_fast.item import BodyItem, TopItem, BottomItem, OthersItem
+from app.service.design_fast.item import BodyItem, TopItem, BottomItem, OthersItem, TopMergeItem, BottomMergeItem, OthersMergeItem
 from app.service.design_fast.utils.organize import organize_body, organize_clothing, organize_others
 from app.service.design_fast.utils.progress import final_progress, update_progress
-from app.service.design_fast.utils.synthesis_item import synthesis, synthesis_single, update_base_size_priority
+from app.service.design_fast.utils.synthesis_item import synthesis, synthesis_single, update_base_size_priority, merge
 from app.service.utils.decorator import RunTime
 
 id_lock = threading.Lock()
@@ -19,22 +19,46 @@ logger = logging.getLogger()
 minio_client = Minio(settings.MINIO_URL, access_key=settings.MINIO_ACCESS, secret_key=settings.MINIO_SECRET, secure=settings.MINIO_SECURE)
 
 
-def process_item(item, basic):
-    # 处理project中单个item
-    if item['type'] == "Body":
-        body_server = BodyItem(data=item, basic=basic, minio_client=minio_client)
-        item_data = body_server.process()
-    elif item['type'].lower() in ['blouse', 'outwear', 'dress', 'tops']:
-        top_server = TopItem(data=item, basic=basic, minio_client=minio_client)
-        item_data = top_server.process()
-    elif item['type'].lower() in ['skirt', 'trousers', 'bottoms']:
-        bottom_server = BottomItem(data=item, basic=basic, minio_client=minio_client)
-        item_data = bottom_server.process()
-    elif item['type'].lower() in ['others']:
-        bottom_server = OthersItem(data=item, basic=basic, minio_client=minio_client)
-        item_data = bottom_server.process()
+def process_item(item, basic, design_type):
+    # 1. 定义映射配置
+    # key 为 item_type 的小写，value 为对应的处理类
+    DESIGN_MAP = {
+        'body': BodyItem,
+        'blouse': TopItem, 'outwear': TopItem,
+        'dress': TopItem, 'tops': TopItem,
+        'skirt': BottomItem, 'trousers': BottomItem,
+        'bottoms': BottomItem,
+        'others': OthersItem
+    }
+
+    MERGE_MAP = {
+        'body_merge': BodyItem,
+        'blouse_merge': TopMergeItem, 'outwear_merge': TopMergeItem,
+        'dress_merge': TopMergeItem, 'tops_merge': TopMergeItem,
+        'skirt_merge': BottomMergeItem, 'trousers_merge': BottomMergeItem,
+        'bottoms_merge': BottomMergeItem,
+        'others_merge': OthersMergeItem
+    }
+
+    # 2. 根据 design_type 选择映射表
+    mapping = MERGE_MAP if design_type == 'merge' else DESIGN_MAP
+
+    if design_type == 'merge':
+        item_type_key = f"{item['type'].lower()}_merge"
+    elif design_type == 'default':
+        item_type_key = item['type'].lower()
     else:
-        raise NotImplementedError(f"Item type {item['type']} not implemented")
+        item_type_key = item['type'].lower()
+
+    handler_class = mapping.get(item_type_key)
+
+    if not handler_class:
+        raise NotImplementedError(f"Item type {item['type']} not implemented for design_type={design_type}")
+
+    # 4. 统一实例化并执行
+    # 注意：这里假设所有 Item 类构造函数签名一致
+    server = handler_class(data=item, basic=basic, minio_client=minio_client)
+    item_data = server.process()
     return item_data
 
 
@@ -44,7 +68,7 @@ def process_layer(item, layers):
         body_layer = organize_body(item)
         layers.append(body_layer)
         return item['body_image'].size
-    elif item['name'] == 'others':
+    elif item['name'] in ['others', 'others_merge']:
         front_layer, back_layer = organize_others(item)
         layers.append(front_layer)
         layers.append(back_layer)
@@ -70,10 +94,11 @@ def design_generate(request_data):
         nonlocal active_threads
         basic = object['basic']
         items_response = {'layers': [], 'objectSign': object['objectSign'] if 'objectSign' in object.keys() else ""}
+        design_type = basic.get('design_type', "default")
         if basic['single_overall'] == "overall":
             item_results = []
             for item in object['items']:
-                item_results.append(process_item(item, basic))
+                item_results.append(process_item(item, basic, design_type))
             layers = []
             for item in item_results:
                 process_layer(item, layers)
@@ -97,7 +122,13 @@ def design_generate(request_data):
                     'rotate': lay.get('rotate', None),
                     # 'back_perspective_url': lay['back_perspective_url'] if 'back_perspective_url' in lay.keys() else None,
                 })
-            items_response['synthesis_url'] = synthesis(layers, new_size, basic)
+            if basic.get('design_type') == 'default':
+                items_response['synthesis_url'] = synthesis(layers, new_size, basic)
+            elif basic.get('design_type') == 'merge':
+                items_response['synthesis_url'] = merge(layers, new_size, basic)
+            else:
+                items_response['synthesis_url'] = synthesis(layers, new_size, basic)
+
         else:
             item_result = process_item(object['items'][0], basic)
             items_response['layers'].append({