Update app.py

app.py

@@ -114,7 +114,7 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   image = pipe(prompt=prompt, negative_prompt=negative_prompt, num_inference_steps=20, width=768, height=768, generator=generator).images[0]
   del pipe
   gc.collect()
-  return image, gr.update(visible=True), seed
+  return image, gr.update(visible=True), seed, gr.update(visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
 
 def get_description(item):
       trigger_word = item["trigger_word"]
@@ -210,8 +210,12 @@ 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, inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale, seed], outputs=[output_image, post_gen_info, last_used_seed])
-  prompt.submit(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])
+  run_btn.click(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,
+                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])
   
   thumbs_up.click(save_preferences, inputs=[lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, output_image, gr.State("up"), seed], outputs=[thumbs_up, thumbs_up_clicked, thumbs_down])
   thumbs_down.click(save_preferences, inputs=[lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, output_image, gr.State("down"), seed], outputs=[thumbs_down, thumbs_down_clicked, thumbs_up])