Преглед на файлове

fix(api): 修复图片路径获取逻辑

- 在703行附近添加条件判断,当图片字典只有一项时使用俯视图,否则使用侧视图
- 在841行附近添加相同的条件判断逻辑,确保一致的图片路径获取行为
- 修复了因图片字典结构变化导致的路径获取错误问题
rambo преди 1 месец
родител
ревизия
617d6bfe89
променени са 1 файла, в които са добавени 8 реда и са изтрити 2 реда
  1. 8 2
      python/api.py

+ 8 - 2
python/api.py

@@ -703,7 +703,10 @@ async def _process_scene_images(aigc_clazz, run_main, return_data_check_before_d
                 progress=product_scene_progress
             )
             first_pics = goods_art_dict_info_item.get("pics")
-            ceshi_image_path = first_pics.get("侧视-抠图")
+            if len(first_pics.items()) == 1:
+                ceshi_image_path = first_pics.get("俯视-抠图")
+            else:
+                ceshi_image_path = first_pics.get("侧视-抠图")
             save_root_path = ceshi_image_path.split("阴影图处理")[0]
             save_image_path = f"{save_root_path}场景图.jpg"
             # 【修复】异步等待单个货号的 AI 处理完成
@@ -838,7 +841,10 @@ async def _process_model_images(aigc_clazz, run_main, return_data_check_before_d
             gender = goods_art_dict_info.get("性别", "女")
             model_id = man_id if "男" in gender else women_id
             first_pics = goods_art_dict_info_item.get("pics")
-            ceshi_image_path = first_pics.get("侧视-抠图")
+            if len(first_pics.items()) == 1:
+                ceshi_image_path = first_pics.get("俯视-抠图")
+            else:
+                ceshi_image_path = first_pics.get("侧视-抠图")
             save_root_path = ceshi_image_path.split("阴影图处理")[0]
             save_image_path = f"{save_root_path}模特图.jpg"