Update app.py

app.py

@@ -27,8 +27,10 @@ def empty_input_image():
              'layers': [Image.new("L", (512, 512), 255),Image.new("L", (512, 512), 255)],
              'composite': Image.new("L", (512, 512), 255)}
 
-def clear_sketchbox():
-    return [empty_input_image(), random.choice(COLORS)]
+def clear_sketchbox(color_name):
+    temp_color_list = COLORS.copy()
+    temp_color_list..remove(color_name)
+    return [empty_input_image(), random.choice(temp_color_list)]
     
 def pil_image_to_data_uri(img, format='PNG'):
     buffered = BytesIO()
@@ -207,6 +209,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=clear_sketchbox, outputs=[image , color])
+    image.clear(fn=clear_sketchbox, input=color, outputs=[image , color])
 if __name__ == "__main__":
     demo.queue().launch(debug=True, allowed_paths=["."])