From 012c53367db8551ab2289405289c03a84a7dedf4 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Fri, 20 Oct 2023 18:41:17 +0800
Subject: [PATCH] minor lite fixes

---
 klite.embd | 35 ++++++++++++++++++++++++-----------
 1 file changed, 24 insertions(+), 11 deletions(-)

diff --git a/klite.embd b/klite.embd
index c018dc24c..6bb441334 100644
--- a/klite.embd
+++ b/klite.embd
@@ -2978,6 +2978,11 @@ Current version: 83
 					synchro_polled_response = synchro_pending_stream;
 					synchro_pending_stream = "";
 					poll_pending_response();
+					//handle gen failures
+					if(resp.status==503)
+					{
+						msgbox("Error while submitting prompt: Server appears to be busy.");
+					}
 				},
 				abort(e) {
 					console.error('Error:', error);
@@ -6756,11 +6761,8 @@ Current version: 83
 		toggle_opmode();
 
 		//sd models display
-		let sdmodelshtml = "";
-		for (var i = 0; i < stablemodels.length; ++i) {
-			sdmodelshtml += "<option value=\"" + stablemodels[i].name + " (" + stablemodels[i].count + ")\">";
-		}
-		document.getElementById("sdmodels").innerHTML = sdmodelshtml;
+		update_horde_sdmodels();
+
 		document.getElementById("tokenstreammode").value = localsettings.tokenstreammode;
 		document.getElementById("img_autogen").checked = localsettings.img_autogen;
 		document.getElementById("save_images").checked = localsettings.save_images;
@@ -6769,6 +6771,15 @@ Current version: 83
 
 	}
 
+	function update_horde_sdmodels()
+	{
+		let sdmodelshtml = "";
+		for (var i = 0; i < stablemodels.length; ++i) {
+			sdmodelshtml += "<option value=\"" + stablemodels[i].name + " (" + stablemodels[i].count + ")\">";
+		}
+		document.getElementById("sdmodels").innerHTML = sdmodelshtml;
+	}
+
 	function toggle_preset() {
 		let selp = document.getElementById("presets").value;
 		let found = presets[selp];
@@ -6806,12 +6817,6 @@ Current version: 83
 	}
 	function clear_sd_model() {
 		document.getElementById("generate_images_model").value = "";
-		if(!image_models_fetched)
-		{
-			//doing it this way will be more buggy,
-			//but since some anons are paranoid about privacy then whatever, only fetch it manually
-			fetch_image_models(display_settings);
-		}
 	}
 
 	function validate_samplers(savesetting = false) {
@@ -7044,6 +7049,14 @@ Current version: 83
 		}else if(document.getElementById("generate_images_mode").value==1){
 			document.getElementById("generate_images_model").classList.remove("hidden");
 			document.getElementById("generate_images_local_model_container").classList.add("hidden");
+			if(!image_models_fetched)
+			{
+				//doing it this way will be more buggy,
+				//but since some anons are paranoid about privacy then whatever, only fetch it manually
+				fetch_image_models(()=>{
+					update_horde_sdmodels();
+				});
+			}
 		}else{
 			document.getElementById("generate_images_model").classList.add("hidden");
 			document.getElementById("generate_images_local_model_container").classList.remove("hidden");