From 292bfb82b6abee459b69a201cbf7a3edf830d158 Mon Sep 17 00:00:00 2001
From: qwertzman-png <125167389+qwertzman-png@users.noreply.github.com>
Date: Fri, 21 Apr 2023 01:41:18 +0200
Subject: [PATCH] .

---
 web/scripts/api.js | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/web/scripts/api.js b/web/scripts/api.js
index 2b90c2abc..2d3075e3d 100644
--- a/web/scripts/api.js
+++ b/web/scripts/api.js
@@ -21,9 +21,37 @@ class ComfyApi extends EventTarget {
 				this.dispatchEvent(new CustomEvent("status", { detail: status }));
 			} catch (error) {
 				this.dispatchEvent(new CustomEvent("status", { detail: null }));
+				
 			}
 		}, 1000);
 	}
+	
+	#pollMessageQueue() {
+		let lastMessage = -1;
+		setInterval(async () => {
+			const poll_messages = await (await fetch("/poll_messages")).json();
+			if(lastMessage === -1) lastMessage = poll_messages.length -1;
+			for(let i = lastMessage; i<poll_messages.length; i++) {
+				const msg = poll_messages[i];
+				console.log('MSG', msg);
+				switch (msg.type) {
+					case "status":
+						this.dispatchEvent(new CustomEvent("status", { detail: msg.data.status }));
+						break;
+					case "progress":
+						this.dispatchEvent(new CustomEvent("progress", { detail: msg.data }));
+						break;
+					case "executing":
+						this.dispatchEvent(new CustomEvent("executing", { detail: msg.data.node }));
+						break;
+					case "executed":
+						this.dispatchEvent(new CustomEvent("executed", { detail: msg.data }));
+						break;
+				}
+				lastMessage = i;
+			}
+		}, 3000)
+	}
 
 	/**
 	 * Creates and connects a WebSocket for realtime updates
@@ -106,6 +134,7 @@ class ComfyApi extends EventTarget {
 	 */
 	init() {
 		this.#createSocket();
+		
 	}
 
 	/**