Update app.py

app.py

@@ -133,7 +133,7 @@ def save_preferences(lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, g
             json.dump({"prompt": prompt, "file_name":image_path.name, "lora_1_id": lora_2_id, "lora_1_scale": lora_1_scale, "lora_2_id": lora_2_id, "lora_2_scale": lora_2_scale, "thumbs_direction": thumbs_direction, "seed": seed}, f)
             f.write("\n")
     
-    return gr.update(visible=True), gr.update(interactive=False, elem_id=f"thumbs-{thumbs_direction}-clicked"), gr.update(interactive=False)
+    return gr.update(visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(interactive=False)
 
 with gr.Blocks(css=css) as demo:
   shuffled_items = gr.State()
@@ -167,8 +167,10 @@ with gr.Blocks(css=css) as demo:
             with gr.Row(visible=False, elem_id="post_gen_info") as post_gen_info:
                 with gr.Column(min_width=10):
                     thumbs_up = gr.Button("👍", elem_id="thumbs_up_unclicked")
+                    thumbs_up_clicked = gr.Button("👍", elem_id="thumbs_up_clicked", interactive=False, visible=False)
                 with gr.Column(min_width=10):
                     thumbs_down = gr.Button("👎", elem_id="thumbs_down_unclicked")
+                    thumbs_down_clicked = gr.Button("👎", elem_id="thumbs_down_clicked", interactive=False, visible=False)
                 with gr.Column(min_width=10):
                     with gr.Group(elem_id="share-btn-container") as share_group:
                         community_icon = gr.HTML(community_icon_html)
@@ -190,8 +192,8 @@ with gr.Blocks(css=css) as demo:
   run_btn.click(merge_and_run, inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale], 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], outputs=[output_image, post_gen_info, last_used_seed])
   
-  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=[post_eval, thumbs_up, 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=[post_eval, thumbs_down, thumbs_up])
+  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=[post_eval, 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=[post_eval, thumbs_down, thumbs_down_clicked, thumbs_up])
   share_button.click(None, [], [], _js=share_js)
 demo.queue()
 demo.launch()