From 82016a163a94272133c23387558e44ae338743fc Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Tue, 21 Feb 2023 08:14:04 -0600
Subject: [PATCH] restore single output in chain pipeline stages

---
 api/onnx_web/chain/persist_disk.py | 4 ++--
 api/onnx_web/output.py             | 4 ++--
 api/onnx_web/serve.py              | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/api/onnx_web/chain/persist_disk.py b/api/onnx_web/chain/persist_disk.py
index 041dabae..3be8a0d0 100644
--- a/api/onnx_web/chain/persist_disk.py
+++ b/api/onnx_web/chain/persist_disk.py
@@ -17,12 +17,12 @@ def persist_disk(
     _params: ImageParams,
     source: Image.Image,
     *,
-    output: List[str],
+    output: str,
     stage_source: Image.Image,
     **kwargs,
 ) -> Image.Image:
     source = stage_source or source
 
-    dest = save_image(server, output[0], source)
+    dest = save_image(server, output, source)
     logger.info("saved image to %s", dest)
     return source
diff --git a/api/onnx_web/output.py b/api/onnx_web/output.py
index 446cafb6..d793d262 100644
--- a/api/onnx_web/output.py
+++ b/api/onnx_web/output.py
@@ -30,14 +30,14 @@ def hash_value(sha, param: Param):
 
 
 def json_params(
-    output: List[str],
+    outputs: List[str],
     params: ImageParams,
     size: Size,
     upscale: Optional[UpscaleParams] = None,
     border: Optional[Border] = None,
 ) -> Any:
     json = {
-        "output": output,
+        "outputs": outputs,
         "params": params.tojson(),
     }
 
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
index befd1fda..3f2dc96b 100644
--- a/api/onnx_web/serve.py
+++ b/api/onnx_web/serve.py
@@ -760,7 +760,7 @@ def chain():
         context,
         params,
         empty_source,
-        output=output,
+        output=output[0],
         size=size,
         needs_device=device,
     )