From 35245288da69cc184edbb84f18ec8361fe7ce645 Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Fri, 19 Aug 2022 19:08:09 -0400
Subject: [PATCH] Abort fix for new UI

---
 koboldai_settings.py | 3 +++
 static/koboldai.js   | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/koboldai_settings.py b/koboldai_settings.py
index 0d7cee23..c3af317f 100644
--- a/koboldai_settings.py
+++ b/koboldai_settings.py
@@ -707,6 +707,9 @@ class system_settings(settings):
             serverstarted = value
         if name not in self.local_only_variables and name[0] != "_" and not new_variable:
             process_variable_changes(self.socketio, self.__class__.__name__.replace("_settings", ""), name, value, old_value)
+            
+            if name == "aibusy" and value == False:
+                koboldai_vars.abort = False
         
 class KoboldStoryRegister(object):
     def __init__(self, socketio, story_settings, koboldai_vars, tokenizer=None, sequence=[]):
diff --git a/static/koboldai.js b/static/koboldai.js
index 4d9d907e..89e6878d 100644
--- a/static/koboldai.js
+++ b/static/koboldai.js
@@ -427,6 +427,9 @@ function do_ai_busy(data) {
 		}
 		favicon.stop_swap()
 		document.getElementById('btnsend').textContent = "Submit";
+		for (item of document.getElementsByClassName('statusbar_outer')) {
+			item.classList.add("hidden");
+		}
 	}
 }