command_to_image

python/service/online_request/module_online_data.py

@@ -126,19 +126,22 @@ class AIGCDataRequest(object):
             "gemini_model": "gemini-2.5-flash-image-preview",
         }
         """生成场景图"""
-        url = settings.DOMAIN + "/api/ai_image/inspired/command_to_image"
-        resultData = self.s.post(url, data=data, headers=self.post_headers).json()
+        try:
+            url = settings.DOMAIN + "/api/ai_image/inspired/command_to_image"
+            resultData = self.s.post(url, data=data, headers=self.post_headers, timeout=70).json()
 
-        code = resultData.get("code", 0)
-        message = resultData.get("message", "")
-        if code != 0:
-            raise UnicornException(message)
-        image_arr = resultData.get("data", None).get("image", [])
-        if len(image_arr) == 0:
-            raise UnicornException("场景图生成失败")
-        image_url = image_arr[0]
-        save_image_path = download_image_with_pil(image_url, save_path)
-        return save_image_path
+            code = resultData.get("code", 0)
+            message = resultData.get("message", "")
+            if code != 0:
+                raise UnicornException(message)
+            image_arr = resultData.get("data", None).get("image", [])
+            if len(image_arr) == 0:
+                raise UnicornException("场景图生成失败")
+            image_url = image_arr[0]
+            save_image_path = download_image_with_pil(image_url, save_path)
+            return save_image_path
+        except:
+          raise UnicornException("场景图生成失败")
 
     def generateProductSceneQW(self, local_path, prompt, save_path):
         '''千问生成场景图'''