From f91c77bdc660bb8a5ca13ee20574a042d566638f Mon Sep 17 00:00:00 2001
From: Marc-Andre Ferland <madrang@gmail.com>
Date: Fri, 14 Oct 2022 04:47:13 -0400
Subject: [PATCH] Failed task go immediately into the buffer state with the
 error.

---
 ui/media/main.js | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/ui/media/main.js b/ui/media/main.js
index 260ef0a3..345287ef 100644
--- a/ui/media/main.js
+++ b/ui/media/main.js
@@ -541,12 +541,13 @@ async function doMakeImage(task) {
                 throw new Error('Connexion with server lost.')
             }
         } while (serverState.time > (Date.now() - (10 * 1000)) && serverState.task !== renderRequest.task)
-        if (serverState.session !== 'pending' && serverState.session !== 'running') {
+        if (serverState.session !== 'pending' && serverState.session !== 'running' && serverState.session !== 'buffer') {
             throw new Error('Unexpected server task state: ' + serverState.session || 'Undefined')
         }
-        do { // Wait for task to start on server.
+        while (serverState?.session === 'pending') {
+            // Wait for task to start on server.
             await asyncDelay(1500)
-        } while (serverState?.session === 'pending')
+        }
 
         // Task started!
         res = await fetch(renderRequest.stream, {