fix js func signature and not forget to initialize confirmation var to prevent exception upon cancelling confirmation

2 files changed, 7 insertions, 4 deletions

diff --git a/javascript/ui.js b/javascript/ui.js
index 6c99824b..39011079 100644
--- a/javascript/ui.js
+++ b/javascript/ui.js
@@ -162,12 +162,13 @@ function selected_tab_id() {
 
 }
 
-function clear_prompt(_, _prompt_neg, confirmed,_steps) {
+function clear_prompt(_, _prompt_neg, confirmed, _token_counter) {
+confirmed = false
 
 if(confirm("Delete prompt?")) {
     confirmed = true
 } else {
-return [_, confirmed]
+return [_, _prompt_neg, confirmed, _token_counter]
 }
 
     if(selected_tab_id() == "tab_txt2img") {
@@ -176,7 +177,7 @@ return [_, confirmed]
         update_token_counter("txt2img_token_button")
     }
 
-    return [_, _prompt_neg, confirmed,_steps]
+    return [_, _prompt_neg, confirmed, _token_counter]
 }
 
 
diff --git a/modules/ui.py b/modules/ui.py
index 25aeba3b..e58f040e 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -429,10 +429,12 @@ def create_seed_inputs():
     return seed, reuse_seed, subseed, reuse_subseed, subseed_strength, seed_resize_from_h, seed_resize_from_w, seed_checkbox

 

 

-def clear_prompt(_prompt, _prompt_neg, confirmed, _token_counter):

+def clear_prompt(prompt, _prompt_neg, confirmed, _token_counter):

     """Given confirmation from a user on the client-side, go ahead with clearing prompt"""

     if confirmed:

         return ["", "", confirmed, update_token_counter("", 1)]

+    else:

+        return [prompt, _prompt_neg, confirmed, _token_counter]

 

 

 def connect_clear_prompt(button, prompt, prompt_neg, _dummy_confirmed, token_counter):