Update app.py

app.py

@@ -173,6 +173,9 @@ def save_preferences(lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, g
     
     return gr.update(visible=False), gr.update(visible=True), gr.update(interactive=False)
 
+def hide_post_gen_info():
+    return gr.update(visible=False)
+
 with gr.Blocks(css=css) as demo:
   shuffled_items = gr.State()
   title = gr.HTML(
@@ -229,10 +232,10 @@ with gr.Blocks(css=css) as demo:
   demo.load(shuffle_images, inputs=[], outputs=[lora_1_link, lora_1, lora_1_prompt, lora_1_id, lora_2_link, lora_2, lora_2_prompt, lora_2_id, prompt, shuffled_items, lora_1_scale, lora_2_scale], queue=False, show_progress="hidden")
   shuffle_button.click(shuffle_images, outputs=[lora_1_link, lora_1, lora_1_prompt, lora_1_id, lora_2_link, lora_2, lora_2_prompt, lora_2_id, prompt, shuffled_items, lora_1_scale, lora_2_scale], queue=False, show_progress="hidden")
 
-  run_btn.click(merge_and_run,
+  run_btn.click(hide_post_gen_info, outputs=[post_gen_info], queue=False).then(merge_and_run,
                 inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale, seed],
                 outputs=[output_image, post_gen_info, last_used_seed, thumbs_up, thumbs_up_clicked, thumbs_down, thumbs_down_clicked])
-  prompt.submit(merge_and_run,
+  prompt.submit(hide_post_gen_info, outputs=[post_gen_info], queue=False).then(merge_and_run,
                 inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale, seed],
                 outputs=[output_image, post_gen_info, last_used_seed, thumbs_up, thumbs_up_clicked, thumbs_down, thumbs_down_clicked])