Update app.py

app.py

@@ -29,8 +29,6 @@ import json
 import os
 import glob
 import pathlib
-import base64
-from io import BytesIO
 from functools import partial
 from pprint import pprint
 
@@ -45,7 +43,7 @@ import spaces
 from model import StableMultiDiffusionSDXLPipeline
 from util import seed_everything
 from prompt_util import preprocess_prompts, _quality_dict, _style_dict
-from share_btn import community_icon_html, loading_icon_html, share_js
+from share_btn import share_js
 
 
 ### Utils
@@ -377,12 +375,6 @@ def run(state, drawpad):
     )
     return image
 
-    # buffered = BytesIO()
-    # image.save(buffered, format="JPEG")
-    # img_str = base64.b64encode(buffered.getvalue())
-    # return image, img_str
-
-
 
 ### Load examples
 
@@ -456,44 +448,19 @@ css = f"""
     display: none;
 }}
 
-#share-btn-container {{
-    display: flex;
-    padding-left: 0.5rem !important;
-    padding-right: 0.5rem !important;
+#share-btn {{
+    color: #ffffff;font-weight: 600;
     background-color: #000000;
-    justify-content: center;
-    align-items: center;
+    font-family: 'IBM Plex Sans', sans-serif;
     border-radius: 9999px !important;
-    width: 13rem;
-    margin-top: 10px;
-    margin-left: auto;
 }}
 
-#share-btn {{
-    all: initial;
+#share-btn:hover {{
     color: #ffffff;font-weight: 600;
-    cursor:pointer;
+    background-color: #000000;
     font-family: 'IBM Plex Sans', sans-serif;
-    margin-left: 0.5rem !important;
-    padding-top: 0.25rem !important;
-    padding-bottom: 0.25rem !important;right:0;
-}}
-
-#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;
-}}
-
-#output-b64 {{
-    display: none !important;
+    border-radius: 9999px !important;
+    box-shadow: 0 0 20px #eee;
 }}
 """
 
@@ -716,11 +683,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         # scale=1,
                         elem_id='run-button'
                     )
-                    # with gr.Group(elem_id='share-btn-container'):
-                    #     gr.HTML(community_icon_html)
-                    #     gr.HTML(loading_icon_html)
-                    #     iface.btn_share = gr.Button('Share with Community', elem_id='share-btn')
-                    #     iface.image_b64 = gr.Textbox(elem_id='output-b64')
+                    iface.btn_share = gr.Button('🤗 Share with Community', elem_id='share-btn')
 
                     iface.model_select = gr.Radio(
                         list(model_dict.keys()),
@@ -833,7 +796,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     iface.btn_generate.click(
         fn=run,
         inputs=[state, iface.ctrl_semantic],
-        # outputs=[iface.image_slot, iface.image_b64],
         outputs=iface.image_slot,
         api_name='run',
     )
@@ -901,7 +863,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         api_name='quality_select',
     )
 
-    # iface.btn_share.click(None, [], [], js=share_js)
+    iface.btn_share.click(None, [], [], js=share_js)
 
     iface.btn_export_state.click(lambda x: vars(x), state, iface.json_state_export)
     iface.btn_import_state.click(import_state, [state, iface.tbox_state_import], [