From c711aa70186fcd95147139579b49219f1663caea Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Tue, 27 Jun 2023 01:29:46 +0900
Subject: [PATCH] bugfix: invalid path on LatentReceiver

---
 modules/impact/config.py      |  2 +-
 modules/impact/impact_pack.py | 20 +++++++++++++++-----
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/modules/impact/config.py b/modules/impact/config.py
index 340519b..86856b1 100644
--- a/modules/impact/config.py
+++ b/modules/impact/config.py
@@ -1,7 +1,7 @@
 import configparser
 import os
 
-version = "V2.15.1"
+version = "V2.15.2"
 
 dependency_version = 1
 
diff --git a/modules/impact/impact_pack.py b/modules/impact/impact_pack.py
index bf2a508..23dee19 100644
--- a/modules/impact/impact_pack.py
+++ b/modules/impact/impact_pack.py
@@ -8,6 +8,7 @@ from segment_anything import sam_model_registry
 from io import BytesIO
 import piexif
 import zipfile
+import re
 from server import PromptServer
 
 from impact.utils import *
@@ -1539,6 +1540,19 @@ class LatentReceiver:
             return {"samples": tensor['latent_tensor']}
         return None
 
+    def parse_filename(self, filename):
+        pattern = r"^(.*)/(.*?)\[(.*)\]\s*$"
+        match = re.match(pattern, filename)
+        if match:
+            subfolder = match.group(1)
+            filename = match.group(2).rstrip()  # 오른쪽 공백 제거
+            file_type = match.group(3)
+        else:
+            subfolder = ''
+            file_type = self.type
+
+        return {'filename': filename, 'subfolder': subfolder, 'type': file_type}
+
     def doit(self, latent, link_id):
         latent_path = folder_paths.get_annotated_filepath(latent)
 
@@ -1551,11 +1565,7 @@ class LatentReceiver:
         else:
             samples = LatentReceiver.load_preview_latent(latent_path)
 
-        preview = {
-                    'filename': latent_path,
-                    'subfolder': '',
-                    'type': self.type
-                    }
+        preview = self.parse_filename(latent)
 
         return {
                 'ui': {"images": [preview]},