From d56cbf572d5239d5846a3e33ebd770477bcaf6d6 Mon Sep 17 00:00:00 2001
From: yolain <yolain@163.com>
Date: Fri, 13 Dec 2024 12:40:07 +0800
Subject: [PATCH] Fix image chooer can not using in a loop #574

---
 __init__.py | 2 +-
 py/image.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/__init__.py b/__init__.py
index 5dc003b..7538d55 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "1.2.5"
+__version__ = "1.2.6"
 
 import yaml
 import os
diff --git a/py/image.py b/py/image.py
index 5625917..fa64ac7 100644
--- a/py/image.py
+++ b/py/image.py
@@ -956,6 +956,7 @@ class imageChooser(PreviewImage):
 
   def chooser(self, prompt=None, my_unique_id=None, extra_pnginfo=None, **kwargs):
     id = my_unique_id[0]
+    id = id.split('.')[len(id.split('.')) - 1] if "." in id else id
     if id not in ChooserMessage.stash:
       ChooserMessage.stash[id] = {}
     my_stash = ChooserMessage.stash[id]