From 74a9c46f085af13d38a79de34a2ce788a01f6479 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Wed, 14 Sep 2022 17:21:00 +0530
Subject: [PATCH] Show the final output, if it was sent in a single streaming
 packet

---
 ui/index.html | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ui/index.html b/ui/index.html
index eb6465cd..5d2c770b 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -669,7 +669,9 @@ async function doMakeImage(reqBody, batchCount) {
                 try {
                     let stepUpdate = JSON.parse(jsonStr)
 
-                    if (stepUpdate.step !== undefined) {
+                    if (stepUpdate.step === undefined) {
+                        finalJSON += jsonStr
+                    } else {
                         let batchSize = parseInt(reqBody['num_inference_steps'])
                         let overallStepCount = stepUpdate.step + batchesDone * batchSize
                         let totalSteps = batchCount * batchSize