Update app.py

app.py

@@ -23,10 +23,13 @@ DEFAULT_ITEM_NAME = "💡 Lamp"
 COLORS = ["blue", "red", "black", "yellow", "green", "pink", "grey"]
 DEFAULT_COLOR = "yellow"
 def empty_input_image():
-    return [{ 'background': Image.new("L", (512, 512), 255),
+    return { 'background': Image.new("L", (512, 512), 255),
              'layers': [Image.new("L", (512, 512), 255),Image.new("L", (512, 512), 255)],
-             'composite': Image.new("L", (512, 512), 255)}, random.choice(COLORS)]
+             'composite': Image.new("L", (512, 512), 255)}
 
+def clear_sketchbox():
+    return [empty_input_image(), random.choice(COLORS)]
+    
 def pil_image_to_data_uri(img, format='PNG'):
     buffered = BytesIO()
     img.save(buffered, format=format)
@@ -204,6 +207,6 @@ with gr.Blocks(css="style.css", head = head) as demo:
     item.change(fn=run, inputs=inputs, outputs=outputs)
     color.change(fn=run, inputs=inputs, outputs=outputs)
     image.change(fn=run, inputs=inputs, outputs=outputs, trigger_mode="always_last")
-    image.clear(fn=empty_input_image, outputs=[image , color])
+    image.clear(fn=clear_sketchbox, outputs=[image , color])
 if __name__ == "__main__":
     demo.queue().launch(debug=True, allowed_paths=["."])