From ee68cf222c901f52df6c030eeff097b2be894f26 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Tue, 8 Sep 2026 18:15:18 +0800
Subject: [PATCH] fix genkey image gen status leak
---
embd_res/kcpp_sdui.embd | 8 ++++----
koboldcpp.py | 8 ++++++--
2 files changed, 10 insertions(+), 6 deletions(-)
diff --git a/embd_res/kcpp_sdui.embd b/embd_res/kcpp_sdui.embd
index 4a8609d4a..76be75b43 100644
--- a/embd_res/kcpp_sdui.embd
+++ b/embd_res/kcpp_sdui.embd
@@ -6,7 +6,7 @@
Stable UI for KoboldCpp
diff --git a/koboldcpp.py b/koboldcpp.py
index 6d8545533..0c50d2f62 100644
--- a/koboldcpp.py
+++ b/koboldcpp.py
@@ -6833,9 +6833,13 @@ Change Mode
parsed_dict = urllib.parse.parse_qs(parsed_url.query)
genkey = parsed_dict.get('genkey', [''])[0]
skip_current_image = parse_query_bool(parsed_dict, 'skip_current_image')
- # with no auth, reveal status without preview image
+ # Only expose progress and previews for the requested active generation.
auth = bool(genkey and genkey==currgenimgkey)
- info = a1111_progress_response(auth and not skip_current_image)
+ info = build_a1111_progress_response('idle')
+ if auth:
+ active_info = a1111_progress_response(not skip_current_image)
+ if genkey == currgenimgkey:
+ info = active_info
response_body = json.dumps(info).encode()
elif clean_path=='/history' or clean_path=='/api/history' or clean_path.startswith('/api/history/') or clean_path.startswith('/history/'): #emulate comfyui
modelNameToReturn = friendlysdmodelname