Update app.py

app.py

@@ -132,7 +132,8 @@ def save_preferences(lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, g
         with IMAGE_JSONL_PATH.open("a") as f:
             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)
+    
+    return gr.update(visible=True), gr.update(interactive=False, elem_classes=f"thumbs-{thumbs_direction}-clicked"), gr.update(interactive=False)
 
 with gr.Blocks(css=css) as demo:
   shuffled_items = gr.State()
@@ -165,9 +166,9 @@ with gr.Blocks(css=css) as demo:
             output_image = gr.Image(label="Output", height=355, elem_id="output_image")
             with gr.Row(visible=False, elem_id="post_gen_info") as post_gen_info:
                 with gr.Column(min_width=10):
-                    thumbs_up = gr.Button("👍")
+                    thumbs_up = gr.Button("👍", elem_classes="thumbs_up_unclicked")
                 with gr.Column(min_width=10):
-                    thumbs_down = gr.Button("👎")
+                    thumbs_down = gr.Button("👎", elem_classes="thumbs_down_unclicked")
                 with gr.Column(min_width=10):
                     with gr.Group(elem_id="share-btn-container") as share_group:
                         community_icon = gr.HTML(community_icon_html)
@@ -188,8 +189,9 @@ 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_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_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])
   share_button.click(None, [], [], _js=share_js)
 demo.queue()
 demo.launch()