Update app.py

app.py

@@ -96,7 +96,8 @@ MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
 os.makedirs(TMP_DIR, exist_ok=True)
 
-
+def reset_canvas():
+    return gr.update({"background":Image.new("RGB", (512, 512), (255, 255, 255)), "layers":[Image.new("RGB", (512, 512), (255, 255, 255))], "composite":Image.new("RGB", (512, 512), (255, 255, 255))})
 
 def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str, str]:
     p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
@@ -132,7 +133,6 @@ def preprocess_image(image: Image.Image,
     """
         
     width, height  = image['composite'].size
-    image["background"] = Image.new("RGB", (width, height), (255, 255, 255))
     ratio = np.sqrt(1024. * 1024. / (width * height))
     new_width, new_height = int(width * ratio), int(height * ratio)
     image = image['composite'].resize((new_width, new_height))
@@ -500,6 +500,10 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         lambda: tuple([True, gr.Row.update(visible=False), gr.Row.update(visible=True)]),
         outputs=[is_multiimage, single_image_example, multiimage_example]
     )
+    image_prompt.clear(
+        fn=reset_canvas,
+        outputs = [image_prompt]
+    )
     
     # image_prompt.upload(
     #     preprocess_image,