diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index 59d474f8..494fdc5d 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -288,7 +288,10 @@ function showImages(reqBody, res, outputContainer, livePreview) {
                             allHidden = false;
                         }
                     }
-                    if(allHidden === true) {parentTaskContainer.classList.add("displayNone")}
+                    if(allHidden === true) {
+                        const req = htmlTaskMap.get(parentTaskContainer)
+                        if(!req.isProcessing || req.batchesDone == req.batchCount) {parentTaskContainer.classList.add("displayNone")}
+                    }
                 })
             })
         }