Update app.py

app.py

@@ -564,9 +564,9 @@ def draw(state, drawpad):
     #     mask_stds = [0] + [state.mask_stds[v] for v in has_masks]
 
     model.update_masks(
-        mask=masks,
-        mask_strength=mask_strengths,
-        mask_std=mask_stds,
+        masks=masks,
+        mask_strengths=mask_strengths,
+        mask_stds=mask_stds,
     )
     # for i in range(len(has_masks)):
     #     model.update(
@@ -627,6 +627,20 @@ css = f"""
     width: 100%;
     aspect-ratio: {opt.width} / {opt.height};
 }}
+
+#share-btn {{
+    color: #ffffff;font-weight: 600;
+    background-color: #000000;
+    font-family: 'IBM Plex Sans', sans-serif;
+    border-radius: 9999px !important;
+}}
+#share-btn:hover {{
+    color: #ffffff;font-weight: 600;
+    background-color: #000000;
+    font-family: 'IBM Plex Sans', sans-serif;
+    border-radius: 9999px !important;
+    box-shadow: 0 0 20px #eee;
+}}
 """
 
 for i in range(opt.max_palettes + 1):
@@ -908,6 +922,19 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, head=head) as demo:
                         elem_id='output-screen',
                         value=lambda: random.choice(example_images),
                     )
+                    
+                    with gr.Accordion(label='History', open=False):
+                        iface.history = gr.Gallery(
+                            label="My creation",
+                            elem_id='output-history',
+                            object_fit='contain',
+                            columns=3,
+                            height='auto',
+                            preview=True,
+                            interactive=False,
+                            visible=False,
+                        )
+                        iface.btn_share = gr.Button('🤗 Share with Community', elem_id='share-btn')
 
                     iface.btn_generate = gr.Button(
                         value=f'Lemme try! ({int(opt.run_time // 60)} min)',
@@ -1378,6 +1405,8 @@ async () => {{
         api_name='draw',
     )
 
+    # iface.btn_share.click(None, [], [], js=share_js)
+
 
 
 if __name__ == '__main__':