Update app.py

app.py

@@ -53,10 +53,10 @@ css = '''
 #run_button{position:absolute;margin-top: 36px;right: 0;margin-right: 1.5em;border-bottom-left-radius: 0px;
     border-top-left-radius: 0px;}
 .random_column{align-self: center; align-items: center}
-#share-btn-container{padding-left: 0.5rem !important; padding-right: 0.5rem !important; background-color: #000000; justify-content: center; align-items: center; border-radius: 9999px !important; max-width: 13rem; margin-left: auto;}
+#share-btn-container{padding-left: 0.5rem !important; padding-right: 0.5rem !important; background-color: #000000; justify-content: center; align-items: center; border-radius: 9999px !important; max-width: 13rem; margin-left: auto;margin-top: 0.35em;}
 div#share-btn-container > div {flex-direction: row;background: black;align-items: center}
 #share-btn-container:hover {background-color: #060606}
-#share-btn {all: initial; color: #ffffff;font-weight: 600; cursor:pointer; font-family: 'IBM Plex Sans', sans-serif; margin-left: 0.5rem !important; padding-top: 0.5rem !important; padding-bottom: 0.5rem !important;right:0;}
+#share-btn {all: initial; color: #ffffff;font-weight: 600; cursor:pointer; font-family: 'IBM Plex Sans', sans-serif; margin-left: 0.5rem !important; padding-top: 0.5rem !important; padding-bottom: 0.5rem !important;right:0;font-size: 15px;}
 #share-btn * {all: unset}
 #share-btn-container div:nth-child(-n+2){width: auto !important;min-height: 0px !important;}
 #share-btn-container .wrap {display: none !important}
@@ -83,7 +83,7 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   del pipe
   gc.collect()
   torch.cuda.empty_cache()
-  return image
+  return image, gr.update(visible=True)
 
 def get_description(item):
       trigger_word = item["trigger_word"]
@@ -133,21 +133,17 @@ with gr.Blocks(css=css) as demo:
                 prompt = gr.Textbox(label="Your prompt", info="rearrange the trigger words into a coherent prompt!", show_label=False, interactive=True, elem_id="prompt")
                 run_btn = gr.Button("Run", elem_id="run_button")
             output_image = gr.Image(label="Output", height=355)
-            with gr.Row():
+            with gr.Row(visible=False) as post_gen_info:
                 with gr.Column(min_width=10):
-                    gr.Button("👍")
+                    thumbs_up = gr.Button("👍")
                 with gr.Column(min_width=10):
-                    gr.Button("👎")
+                    thumbs_down = gr.Button("👎")
                 with gr.Column(min_width=10):
                     with gr.Group(elem_id="share-btn-container") as share_group:
                         community_icon = gr.HTML(community_icon_html)
                         loading_icon = gr.HTML(loading_icon_html)
                         share_button = gr.Button("Share to community", elem_id="share-btn")
-                
-    
-  
-  
-  
+
   with gr.Accordion("Advanced settings", open=False):
     negative_prompt = gr.Textbox(label="Negative prompt")
     with gr.Row():
@@ -158,8 +154,8 @@ with gr.Blocks(css=css) as demo:
   demo.load(shuffle_images, inputs=[], outputs=[lora_1, lora_1_prompt, lora_2, lora_2_prompt, prompt, shuffled_items, lora_1_scale, lora_2_scale], queue=False, show_progress="hidden")
   shuffle_button.click(shuffle_images, outputs=[lora_1, lora_1_prompt, lora_2, lora_2_prompt, 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], outputs=[output_image])
-  prompt.submit(merge_and_run, inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale], outputs=[output_image])
+  run_btn.click(merge_and_run, inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale], outputs=[output_image, post_gen_info])
+  prompt.submit(merge_and_run, inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale], outputs=[output_image, post_gen_info])
   share_button.click(None, [], [], _js=share_js)
 demo.queue()
 demo.launch()