Avoid conflicting btn

app.py

@@ -275,14 +275,19 @@ TITPOP
                 input_prompt = gr.Textbox(
                     label="Input Prompt", lines=1, interactive=False, visible=False
                 )
-                gen_img = gr.Button("Generate Image from Result", variant="primary")
+                gen_img = gr.Button("Generate Image from Result", variant="primary", interactive=False)
                 with gr.Row():
                     with gr.Column():
                         img1 = gr.Image(label="Original Propmt", interactive=False)
                     with gr.Column():
                         img2 = gr.Image(label="Generated Prompt", interactive=False)
+        def generate_wrapper(*args):
+            yield "", "", "", gr.update(interactive=False),
+            for i in generate(*args):
+                yield *i, gr.update(interactive=False)
+            yield *i, gr.update(interactive=True)
         submit.click(
-            generate,
+            generate_wrapper,
             [
                 tags_input,
                 nl_prompt_input,
@@ -299,14 +304,21 @@ TITPOP
                 result,
                 input_prompt,
                 cost_time,
+                gen_img,
             ],
             queue=True,
         )
         gen_img.click(
-            generate_image,
+            lambda *args: (*generate_image(*args), gr.update(interactive=True)),
             [seed, result, input_prompt],
-            [img1, img2],
+            [img1, img2, submit],
             queue=True,
         )
+        gen_img.click(
+            lambda *args: gr.update(interactive=False),
+            None,
+            [submit],
+            queue=False,
+        )
 
     demo.launch()