Update app.py

app.py

@@ -103,19 +103,39 @@ def infer(
             api_name="/infer",
         )
         print(f"[调用API] space_client.predict 返回值: {result}")
-        image_info, seed_used = result
-        # API may return url, path, or other keys for the resulting image
-        img_url = image_info.get("url") or image_info.get("path") or None
-        return img_url, seed_used
+        # result可能不是元组形式，需增加健壮性
+        if isinstance(result, dict):
+            # 新API只返回dict
+            image_info = result
+            seed_used = result.get("seed", seed)
+        elif isinstance(result, (tuple, list)) and len(result) == 2:
+            image_info, seed_used = result
+        else:
+            print(f"[错误] space_client.predict 返回类型未知: {type(result)}，内容: {result}")
+            return None, seed
+
+        # 检查image_info是否包含url/path
+        if isinstance(image_info, dict):
+            img_url = image_info.get("url") or image_info.get("path") or None
+            # 获取base64图片（如果有）
+            if img_url is None and "data" in image_info:
+                # 返回base64图片（data格式）
+                return f"data:image/png;base64,{image_info['data']}", seed_used
+            return img_url, seed_used
+        else:
+            print(f"[错误] image_info 格式异常: {image_info}")
+            return None, seed
     except Exception as e:
         import traceback
         traceback.print_exc()
         print(f"[调用API] 调用接口异常: {e}")
-        if hasattr(e, 'message') and "Could not process uploaded images" in str(e):
+        if "Could not process uploaded images" in str(e):
             print(
                 "\n[错误] 上传图片处理失败。请确保图片文件有效且未损坏。\n"
                 "可以尝试重新选择图片或修改图片格式。"
             )
+        else:
+            print("\n[错误] 调用推理服务失败，请稍后重试或联系开发者。")
         return None, seed
 
 examples = [