From aa3347ad13e4fc29273f94dbb55685c9eb28b736 Mon Sep 17 00:00:00 2001
From: ManInDark <61268856+ManInDark@users.noreply.github.com>
Date: Mon, 10 Jul 2023 00:55:21 +0200
Subject: [PATCH] Fixed miscounting problem by implementing suggestion by
 @madrang

---
 ui/media/js/main.js | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index 7bef003f..31ad30f5 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -1292,7 +1292,13 @@ function onTaskCompleted(task, reqBody, instance, outputContainer, stepUpdate) {
 function updateTitle() {
     let all_tasks = [...document.querySelectorAll("div .imageTaskContainer").entries()].map(c => htmlTaskMap.get(c[1]))
     let tasks_to_be_run = all_tasks.filter(task => task.isProcessing)
-    let img_remaining_per_task = tasks_to_be_run.map(task => task.numOutputsTotal - Math.max(0, (task.batchesDone - 1) * task.reqBody.num_outputs))
+    let img_remaining_per_task = tasks_to_be_run.map(task => { return task.numOutputsTotal - Math.max(0, (task.instances || []).reduce((total, value) => {
+        if (value.status === 'completed') {
+            return total + 1
+        } else {
+            return total
+        }
+    }, 0) * task.reqBody.num_outputs) } )
     let img_remaining = img_remaining_per_task.reduce((total, value) => total + value, 0);
     if (img_remaining > 0) {
         document.title = `${img_remaining} - Easy Diffusion`;