update sth. at 2024-11-15 19:04:11

app.py

@@ -40,10 +40,16 @@ def image_gen(prompt):
     """
     try:
         # 使用服务一 (默认模型)
+        print(f"Generating image from service 1 with prompt: {prompt}")
         image_1 = client_image_1.text_to_image(prompt)
+        if image_1 is None:
+            print("Service 1 returned no image.")
         
         # 使用服务二 (FLUX 模型)
+        print(f"Generating image from service 2 with prompt: {prompt}")
         image_2 = client_image_2.text_to_image(prompt)
+        if image_2 is None:
+            print("Service 2 returned no image.")
         
         return image_1, image_2  # 返回两个生成的图像
     except Exception as e: