Add share_btn

app.py

@@ -8,6 +8,8 @@ import cv2
 from io import BytesIO
 from PIL import Image, ImageFilter
 
+from share_btn import community_icon_html, loading_icon_html, share_js, share_btn_css
+
 # Constants
 low_threshold = 100
 high_threshold = 200
@@ -30,7 +32,7 @@ async () => {
 get_js_image = """
 async (canvas, prompt) => {
   const poseMakerEl = document.querySelector("pose-maker");
-  const imgBase64 = poseMakerEl.captureScreenshot();
+  const imgBase64 = poseMakerEl.captureScreenshotDepthMap();
   return [imgBase64, prompt]
 }
 """
@@ -80,7 +82,7 @@ def generate_images(canvas, prompt):
             num_inference_steps=20,
             negative_prompt="longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
         )
-        all_outputs = [input_img]
+        all_outputs = []
         for image in output.images:
             all_outputs.append(image)
         return all_outputs
@@ -104,7 +106,7 @@ async (pose) => {
 }
 """
 
-with gr.Blocks() as blocks:
+with gr.Blocks(css=share_btn_css) as blocks:
     gr.HTML(
         """
             <div style="text-align: center; margin: 0 auto;">
@@ -137,10 +139,15 @@ with gr.Blocks() as blocks:
                 label="Enter your prompt",
                 max_lines=1,
                 placeholder="best quality, extremely detailed",
+                elem_id="prompt",
             )
             run_button = gr.Button("Generate")
+            with gr.Group(elem_id="share-btn-container"):
+                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.Column():
-            gallery = gr.Gallery().style(grid=[2], height="auto")
+            gallery = gr.Gallery(elem_id="gallery").style(grid=[2], height="auto")
     rotation_axis.change(fn=placeholder_fn,
                             inputs=[rotation_axis],
                             outputs=[],
@@ -155,6 +162,7 @@ with gr.Blocks() as blocks:
                      inputs=[canvas, prompt],
                      outputs=[gallery],
                      _js=get_js_image)
+    share_button.click(None, [], [], _js=share_js)
     blocks.load(None, None, None, _js=load_js)
 
 blocks.launch(debug=True)