Update app.py

app.py

@@ -29,6 +29,8 @@ import json
 import os
 import glob
 import pathlib
+import base64
+from io import BytesIO
 from functools import partial
 from pprint import pprint
 
@@ -357,7 +359,7 @@ def run(state, drawpad):
     prompts, negative_prompts = preprocess_prompts(
         prompts, negative_prompts, style_name=state.style_name, quality_name=state.quality_name)
 
-    return generate(
+    image = generate(
         state,
         prompts,
         negative_prompts,
@@ -374,6 +376,11 @@ def run(state, drawpad):
         guidance_scale=0,
     )
 
+    buffered = BytesIO()
+    image.save(buffered, format="JPEG")
+    img_str = base64.b64encode(buffered.getvalue())
+    return image, img_str
+
 
 
 ### Load examples
@@ -704,10 +711,11 @@ 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"):
+                    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.btn_share = gr.Button('Share with Community', elem_id='share-btn')
+                        iface.image_b64 = gr.Textbox(visible=False, elem_id='output-b64')
 
                     iface.model_select = gr.Radio(
                         list(model_dict.keys()),
@@ -820,7 +828,7 @@ 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,
+        outputs=[iface.image_slot, iface.image_b64],
         api_name='run',
     )