Update app.py

app.py

@@ -85,7 +85,7 @@ def generate(image_editor, prompt, neg_prompt, versions):
         final_gallery = versions
         final_gallery.append(final_image)
     
-    return final_dict, gr.Gallery(value=final_gallery, visible=True), gr.update(visible=True)
+    return final_dict, gr.Gallery(value=final_gallery, visible=True), gr.update(visible=True), gr.update(visible=True)
 
 with gr.Blocks() as demo:
     gr.Markdown("""
@@ -105,18 +105,19 @@ with gr.Blocks() as demo:
         with gr.Column():
             version_gallery = gr.Gallery(label="Versions", type="pil", object_fit='contain', visible=False)
             restore_button = gr.Button("Restore Version", visible=False)
+            clear_button = br.Button('Clear', visible=False)
             selected = gr.Number(show_label=False, visible=False)
             
     gr.Examples(
         [[{'background':'./tony.jpg', 'layers':['./tony-mask.jpg'], 'composite':'./tony.jpg'}, 'black and white tuxedo, bowtie', 'ugly',  None]],
         [sketch_pad, prompt, neg_prompt, version_gallery],
-        [sketch_pad, version_gallery, restore_button],
+        [sketch_pad, version_gallery, restore_button, clear_button],
         generate,
         cache_examples=True,
     )
 
     version_gallery.select(get_select_index, None, selected)
-    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, version_gallery], outputs=[sketch_pad, version_gallery, restore_button])
+    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, version_gallery], outputs=[sketch_pad, version_gallery, restore_button, clear_button])
     restore_button.click(fn=restore_version, inputs=[selected, version_gallery], outputs=sketch_pad)
 
 demo.launch()