mirror/stable-diffusion-webui
2
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-12-03 07:00:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

initial work on getting prompts cleared on the backend and synchronizing token counter

Browse Source
This commit is contained in:
papuSpartan 2022-10-21 13:55:42 -05:00
parent 4a9ff0891a
commit 9ba372de90
2 changed files with 26 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
javascript/ui.js
View File

@ -162,9 +162,13 @@ function selected_tab_id() {
}
function trash_prompt(_,_, is_img2img) {
function trash_prompt(_, confirmed) {
if(!confirm("Delete prompt?")) return false
if(confirm("Delete prompt?")) {
confirmed = true
} else {
return [_, confirmed]
}
if(selected_tab_id() == "tab_txt2img") {
gradioApp().querySelector("#txt2img_prompt > label > textarea").value = "";
@ -178,7 +182,7 @@ if(!confirm("Delete prompt?")) return false
update_token_counter("txt2img_token_button")
}
return true
return [_, confirmed]
}

29
modules/ui.py
View File

@ -429,15 +429,16 @@ def create_seed_inputs():
return seed, reuse_seed, subseed, reuse_subseed, subseed_strength, seed_resize_from_h, seed_resize_from_w, seed_checkbox
# setup button for clearing prompt input boxes on client side of webui
def connect_trash_prompt(dummy_component, button, is_img2img):
def clear_prompt(prompt):
print(f"type: '{prompt}'")
print(prompt)
# update_token_counter(prompt, steps)
return ""
def connect_trash_prompt(prompt, confirmed):
if(confirmed): clear_prompt(prompt)
button.click(
fn=lambda: print("Clearing prompt"),
_js="trash_prompt",
inputs=[],
outputs=[],
)
def connect_reuse_seed(seed: gr.Number, reuse_seed: gr.Button, generation_info: gr.Textbox, dummy_component, is_subseed):
""" Connects a 'reuse (sub)seed' button's click event so that it copies last used
@ -640,7 +641,16 @@ def create_ui(wrap_gradio_gpu_call):
dummy_component = gr.Label(visible=False)
txt_prompt_img = gr.File(label="", elem_id="txt2img_prompt_image", file_count="single", type="bytes", visible=False)
connect_trash_prompt(dummy_component, trash_prompt_button, False)
trash_prompt_button.click(
# fn=lambda: print("Clearing prompt"),
_js="trash_prompt",
fn=lambda: clear_prompt(),
inputs=[txt2img_prompt, dummy_component],
outputs=[txt2img_prompt, dummy_component],
)
with gr.Row(elem_id='txt2img_progress_row'):
with gr.Column(scale=1):
@ -848,7 +858,6 @@ def create_ui(wrap_gradio_gpu_call):
img2img_interrogate, img2img_deepbooru, img2img_prompt_style_apply, img2img_save_style, img2img_paste,\
token_counter, token_button, trash_prompt_button = create_toprow(is_img2img=True)
connect_trash_prompt(dummy_component,trash_prompt_button, True)
with gr.Row(elem_id='img2img_progress_row'):
img2img_prompt_img = gr.File(label="", elem_id="img2img_prompt_image", file_count="single", type="bytes", visible=False)