模特图-场景图

python/api.py

@@ -558,7 +558,10 @@ async def process_handle_detail(request: Request, params: HandlerDetail):
                     continue
                 aigc_clazz.center_paste_image(ceshi_image_path, save_image_path)
                 try:
-                    image_path = aigc_clazz.generateProductScene(
+                    # image_path = aigc_clazz.generateProductScene(
+                    #     save_image_path, product_scene_prompt, save_image_path
+                    # )
+                    image_path = aigc_clazz.generateProductSceneQW(
                         save_image_path, product_scene_prompt, save_image_path
                     )
                     goods_art_dict_info["场景图"] = image_path
@@ -619,8 +622,11 @@ async def process_handle_detail(request: Request, params: HandlerDetail):
                     continue
                 shutil.copy(ceshi_image_path, save_image_path)
                 try:
-                    image_path = aigc_clazz.generateModelFitting(
-                        save_image_path, model_id, face_type, save_image_path
+                    # image_path = aigc_clazz.generateModelFitting(
+                    #     save_image_path, model_id, face_type, save_image_path
+                    # )
+                    image_path = aigc_clazz.generateModelclothingQW(
+                        save_image_path, model_id, save_image_path
                     )
                     goods_art_dict_info["模特图"] = image_path
                     new_goods_dict[goods_art_no] = goods_art_dict_info

python/service/online_request/module_online_data.py

@@ -243,6 +243,52 @@ class AIGCDataRequest(object):
         print("上脚图save_image_path", result_image, save_image_path)
         return save_image_path
 
+    def generateProductSceneQW(self, local_path, prompt, save_path):
+        """千问生成场景图"""
+        imageUrl = self.uploadImage(local_path)
+        data = {
+            "machine_type": 1,  # 0鞋；1服装
+            "generate_type": 0,  # 生成类型，这里指代得是场景图还是模特图；0场景图；1模特图
+            "base_image": imageUrl,
+            "prompt": prompt,
+        }
+        """生成场景图"""
+        url = settings.DOMAIN + "/api/ai_image/main/image_edit_generate"
+        resultData = self.s.post(url, data=data, headers=self.post_headers).json()
+
+        code = resultData.get("code", 0)
+        message = resultData.get("message", "")
+        if code != 0:
+            raise UnicornException(message)
+        image_url = resultData.get("data", None).get("image_url", "")
+        if image_url == "" or image_url is None:
+            raise UnicornException("场景图生成失败")
+        save_image_path = download_image_with_pil(image_url, save_path)
+        return save_image_path
+
+    def generateModelclothingQW(self, local_path, model_id, save_path):
+        """千问生成场景图"""
+        imageUrl = self.uploadImage(local_path)
+        data = {
+            "machine_type": 1,  # 0鞋；1服装
+            "generate_type": 1,  # 生成类型，这里指代得是场景图还是模特图；0场景图；1模特图
+            "base_image": imageUrl,
+            "model_template_id": model_id,
+        }
+        """生成场景图"""
+        url = settings.DOMAIN + "/api/ai_image/main/image_edit_generate"
+        resultData = self.s.post(url, data=data, headers=self.post_headers).json()
+
+        code = resultData.get("code", 0)
+        message = resultData.get("message", "")
+        if code != 0:
+            raise UnicornException(message)
+        image_url = resultData.get("data", None).get("image_url", "")
+        if image_url == "" or image_url is None:
+            raise UnicornException("模特图生成失败")
+        save_image_path = download_image_with_pil(image_url, save_path)
+        return save_image_path
+
 
 class OnlineDataRequest(object):
     def __init__(self, token):