Update app.py

app.py

@@ -1,12 +1,15 @@
 import os
+import io
 import json
 import random
 import string
 import requests
+import base64
 import torch
 
 import gradio as gr
 from PIL import Image
+from datasets import load_dataset
 from diffusers import StableDiffusionPipeline
 
 from pingpong import PingPong
@@ -453,6 +456,32 @@ def fill_with_gen(alt_txt, editor):
       gr.update(visible=False)
   ]
 
+def share(title, editor, cover_art):
+  if title.strip() == '':
+      title = 'Untitled'
+    
+  dataset = load_dataset("chansung/llama2-stories")
+    
+  buffered = io.BytesIO()
+  cover_art.save(buffered, format="PNG")
+  img_str = base64.b64encode(buffered.getvalue())
+
+  dataset['train'] = dataset['train'].add_item({
+      'title': title,
+      'story': story,
+      'image': img_str
+  })
+
+  dataset.push_to_hub("chansung/llama2-stories", token=TOKEN)
+    
+  return  [
+      "       "
+  ]
+    
+      fn=share
+      inputs=[title, editor, cover_art],
+      outputs=[progress_bar]
+
 with gr.Blocks(css=STYLES) as demo:
 
   num_enabled_alts = gr.State(0)
@@ -531,7 +560,7 @@ with gr.Blocks(css=STYLES) as demo:
             clear_btn = gr.Button("clear", interactive=False, elem_classes=['control-label-font', 'control-button'])
             regen_btn = gr.Button("regenerate", interactive=False, elem_classes=['control-label-font', 'control-button'])
 
-      with gr.Tab("Exporting"):
+      with gr.Tab("Exporting/Sharing"):
         with gr.Column(elem_classes=['group-border']):
           gr.Markdown("generate cover art with [`nota-ai/bk-sdm-base`](https://huggingface.co/nota-ai/bk-sdm-base) model. "
                       "design your own prompt in the textbox below, or just hit 'generate prompt for cover art` button. LLaMA2 "
@@ -544,8 +573,21 @@ with gr.Blocks(css=STYLES) as demo:
           gen_pdf_btn = gr.Button("export as PDF", elem_classes=['control-label-font', 'control-button'])
           pdf_file = gr.File(visible=False)
 
+        with gr.Column(elem_classes=['group-border']):
+          gr.Markdown("by clicking the button below, your stories will be shared within community. All shared stories will be "
+                      "archived in [llama2-stories](https://huggingface.co/datasets/chansung/llama2-stories) dataset repository, and "
+                      "they will be displayed in [chansung/LLaMA2-Story-Showcase](https://huggingface.co/spaces/chansung/LLaMA2-Story-Showcase) space.")
+            
+          share_btn = gr.Button("share", elem_classes=['control-label-font', 'control-button'])
+
   progress_bar = gr.Textbox(elem_classes=['no-label'])
 
+  share_btn.click(
+      fn=share
+      inputs=[title, editor, cover_art],
+      outputs=[progress_bar]
+  )
+    
   gen_pdf_btn.click(
       lambda t, e, c: generate_pdf(t, e, c),
       inputs=[title, editor, cover_art],