Lower timeout and fix for colab.

This commit is contained in:
comfyanonymous 2023-02-13 15:52:34 -05:00
parent 68a1ac7746
commit 93d84e3b9c

View File

@ -480,11 +480,21 @@ document.addEventListener('drop', (event) => {
prompt_file_load(file);
});
(() => {
function updateStatus(data) {
document.getElementById("queuesize").innerHTML = "Queue size: " + (data ? data.exec_info.queue_remaining : "ERR");
}
//fix for colab and other things that don't support websockets.
function manually_fetch_queue() {
fetch('/prompt')
.then(response => response.json())
.then(data => {
updateStatus(data.exec_info.queue_remaining);
}).catch((response) => {updateStatus(null)});
}
let ws;
function createSocket(isReconnect) {
if(ws) return;
@ -501,14 +511,16 @@ document.addEventListener('drop', (event) => {
ws.addEventListener("error", () => {
if(ws) ws.close();
manually_fetch_queue();
});
ws.addEventListener("close", () => {
setTimeout(() => {
ws = null;
createSocket(true);
}, 1000);
}, 300);
if(opened) {
updateStatus(null);
showModal("Reconnecting...");
}
});