From f6d3349b56b3e79be38a5686b7a9491c7243ad20 Mon Sep 17 00:00:00 2001 From: somebody Date: Fri, 25 Nov 2022 22:26:28 -0600 Subject: [PATCH] Add ability to remove image from wi --- aiserver.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/aiserver.py b/aiserver.py index 7221a3da..b07f8ee0 100644 --- a/aiserver.py +++ b/aiserver.py @@ -8823,7 +8823,14 @@ def UI_2_set_wi_image(uid): "", ) - koboldai_vars.worldinfo_v2.image_store[str(uid)] = request.get_data(as_text=True) + uid = str(uid) + data = request.get_data(as_text=True) + if not data and uid in koboldai_vars.worldinfo_v2.image_store: + # Delete if sent null image + del koboldai_vars.worldinfo_v2.image_store[uid] + else: + # Otherwise assign image + koboldai_vars.worldinfo_v2.image_store[uid] = data return ":)" @app.route("/get_wi_image/", methods=["GET"])