Add share to community

app.py

@@ -11,6 +11,7 @@ from diffusers import (
     DPMSolverMultistepScheduler,  # <-- Added import
     EulerDiscreteScheduler  # <-- Added import
 )
+from share_btn import community_icon_html, loading_icon_html, share_js
 
 BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
 
@@ -87,7 +88,7 @@ def inference(
         #output_type="latent"
     ).images[0]
     
-    return out
+    return out, gr.update(visible=True)
 
 with gr.Blocks() as app:
     gr.Markdown(
@@ -119,13 +120,17 @@ with gr.Blocks() as app:
             run_btn = gr.Button("Run")
         with gr.Column():
             result_image = gr.Image(label="Illusion Diffusion Output")
+            with gr.Group(elem_id="share-btn-container", visible=False) 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")
             
     run_btn.click(
         inference,
         inputs=[control_image, prompt, negative_prompt, guidance_scale, controlnet_conditioning_scale, seed, sampler],
-        outputs=[result_image]
+        outputs=[result_image, share_button]
     )
-
+    share_button.click(None, [], [], _js=share_js)
 app.queue(max_size=20)
 
 if __name__ == "__main__":