Commit 93d84e3b authored by comfyanonymous's avatar comfyanonymous
Browse files

Lower timeout and fix for colab.

parent 68a1ac77
......@@ -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...");
}
});
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment