From 7e7ef9be413cf408689451e7a87a0bccda4b6e77 Mon Sep 17 00:00:00 2001
From: LAOGOU <164337292+LAOGOU-666@users.noreply.github.com>
Date: Sat, 1 Feb 2025 18:07:28 +0800
Subject: [PATCH] Update __init__.py

---
 __init__.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/__init__.py b/__init__.py
index 82d3b0c..1aaaa3d 100644
--- a/__init__.py
+++ b/__init__.py
@@ -33,8 +33,11 @@ async def window_closed(request):
     try:
         data = await request.json()
         node_id = data.get("node_id")
+        original_image = data.get("original_image")
         
         if node_id:
+            if original_image:
+                image_cache[node_id] = original_image
             if node_id in event_dict:
                 event_dict[node_id].set()
             return web.Response(text=json.dumps({"status": "success"}))