From 1f6861d55cc34c8446927b8c7c0855c870f3b4a0 Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Sun, 25 Sep 2022 13:49:02 -0400
Subject: [PATCH] Update for horde selection to pull models automatically (or
 on typing with a 1 second delay

---
 aiserver.py           | 10 +++++++++-
 static/application.js |  9 +++++++++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/aiserver.py b/aiserver.py
index 12659a1c..d5c010aa 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -1479,6 +1479,7 @@ def get_model_info(model, directory=""):
     multi_online_models = False
     gpu_count = torch.cuda.device_count()
     gpu_names = []
+    send_horde_models = False
     for i in range(gpu_count):
         gpu_names.append(torch.cuda.get_device_name(i))
     if model in ['Colab', 'API']:
@@ -1498,6 +1499,10 @@ def get_model_info(model, directory=""):
                     key_value = js["apikey"]
                 elif 'oaiapikey' in js and js['oaiapikey'] != "":
                     key_value = js["oaiapikey"]
+                if 'url' in js and js['url'] != "":
+                    url = js['url']
+            if key_value != "":
+                send_horde_models = True
     elif model in [x[1] for x in model_menu['apilist']]:
         if path.exists(get_config_filename(model)):
             with open(get_config_filename(model), "r") as file:
@@ -1544,7 +1549,9 @@ def get_model_info(model, directory=""):
                          'disk_break_value': disk_blocks, 'accelerate': utils.HAS_ACCELERATE,
                          'break_values': break_values, 'gpu_count': gpu_count, 'multi_online_models': multi_online_models,
                          'url': url, 'default_url': default_url, 'gpu_names': gpu_names, 'models_on_url': models_on_url}, broadcast=True)
-    if key_value != "":
+    if send_horde_models:
+        get_cluster_models({'key': key_value, 'url': default_url})
+    elif key_value != "" and model in [x[1] for x in model_menu['apilist']] and model != 'CLUSTER':
         get_oai_models(key_value)
     
 
@@ -1678,6 +1685,7 @@ def get_cluster_models(msg):
         js={}
         with open(get_config_filename(vars.model_selected), "w") as file:
             js["apikey"] = vars.oaiapikey
+            js["url"] = url
             file.write(json.dumps(js, indent=3))
         
     logger.init_ok("KAI Horde Models", status="OK")
diff --git a/static/application.js b/static/application.js
index 1ebe5424..b6937f84 100644
--- a/static/application.js
+++ b/static/application.js
@@ -107,6 +107,9 @@ var modelname = null;
 var model = "";
 var ignore_stream = false;
 
+//timer for loading CLUSTER models
+var online_model_timmer;
+
 // This is true iff [we're in macOS and the browser is Safari] or [we're in iOS]
 var using_webkit_patch = true;
 
@@ -2923,6 +2926,12 @@ $(document).ready(function(){
 				$("#modelkey").removeClass("hidden");
 				$("#modelkey")[0].value = msg.key_value;
 				if (msg.models_on_url) {
+					$("#modelkey")[0].oninput = function() {clearTimeout(online_model_timmer);
+																online_model_timmer = setTimeout(function() {
+																	socket.send({'cmd': 'Cluster_Key_Update', 'key': document.getElementById("modelkey").value, 
+																											  'url': document.getElementById("modelurl").value});
+																}, 1000);
+															}
 					$("#modelkey")[0].onblur = function () {socket.send({'cmd': 'Cluster_Key_Update', 'key': this.value, 'url': document.getElementById("modelurl").value});};
 					$("#modelurl")[0].onblur = function () {socket.send({'cmd': 'Cluster_Key_Update', 'key': document.getElementById("modelkey").value, 'url': this.value});};
 				} else {