cancel button test

app.py

@@ -109,12 +109,13 @@ def control_ui():
         with gr.Column():
             save = gr.Checkbox(value=True, label='云端保存图片')
     gen_btn = gr.Button(value="生成", variant="primary")
+    stop_btn = gr.Button(value="取消", variant="stop", visible=False)
     rand_seed.click(fn=lambda: -1, inputs=None, outputs=seed)
     width.change(lambda w, h: h if w*h<=1024*1024 else (1024*1024//w//64)*64, [width, height], height)
     height.change(lambda w, h: w if w*h<=1024*1024 else (1024*1024//h//64)*64, [width, height], width)
     i2i.select(lambda: 'i2i', inputs=None, outputs=selection)
     inp.select(lambda: 'inp', inputs=None, outputs=selection)
-    return gen_btn, [prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_images, info_extracts, ref_strs, i2i_image, i2i_str, i2i_noise, overlay, inp_img, selection], [save, rand_seed, reuse_seed, reuse_img_i2i, reuse_img_inp]
+    return gen_btn, stop_btn, [prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_images, info_extracts, ref_strs, i2i_image, i2i_str, i2i_noise, overlay, inp_img, selection], [save, rand_seed, reuse_seed, reuse_img_i2i, reuse_img_inp]
 
 def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_images, info_extracts, ref_strs, i2i_image, i2i_str, i2i_noise, overlay, inp_img, selection):
     global today_count
@@ -159,14 +160,17 @@ def main_ui():
     with gr.Blocks() as page:
         with gr.Row(variant="panel"):
             with gr.Column():
-                gen_btn, paras, others = control_ui()
+                gen_btn, stop_btn, paras, others = control_ui()
             with gr.Column():
                 image, info = preview_ui()
-    gen_btn.click(generate, paras, [image, info], concurrency_limit=1)
+    gen_btn.click(lambda: (gr.Button(visible=False), gr.Button(visible=True)), inputs=None, outputs=[gen_btn, stop_btn])
+    gen = gen_btn.click(generate, paras, [image, info], concurrency_limit=1)
     others[2].click(lambda o, s: o if len(s) == 0 else s['parameters']['seed'], inputs=[paras[3], info], outputs=paras[3])
     others[3].click(lambda i: i, inputs=image, outputs=paras[17])
     others[4].click(lambda i: gr.ImageEditor(value=i), inputs=image, outputs=paras[21])
-    image.change(rename_save_img, inputs=[image, info, others[0]], outputs=image, trigger_mode="once")
+    ren = image.change(rename_save_img, inputs=[image, info, others[0]], outputs=image, trigger_mode="once")
+    image.change(lambda: (gr.Button(visible=True), gr.Button(visible=False)), inputs=None, outputs=[gen_btn, stop_btn])
+    stop_btn.click(lambda: (gr.Button(visible=True), gr.Button(visible=False)), inputs=None, outputs=[gen_btn, stop_btn], cancels=[gen, ren])
     return page, paras[:14]
 
 def util_ui():