From fcea052dde1c6a3202a84c6c5b33fa865eace3d5 Mon Sep 17 00:00:00 2001
From: Hayden <48267247+hayden-fr@users.noreply.github.com>
Date: Mon, 10 Feb 2025 17:00:08 +0800
Subject: [PATCH] fix: resolve path (#132)

---
 py/manager.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/py/manager.py b/py/manager.py
index 785a5ab..0325105 100644
--- a/py/manager.py
+++ b/py/manager.py
@@ -120,7 +120,11 @@ class ModelManager:
         folders, *others = folder_paths.folder_names_and_paths[folder]
 
         def get_file_info(entry: os.DirEntry[str], base_path: str, path_index: int):
-            fullname = utils.normalize_path(entry.path).replace(f"{base_path}/", "")
+            prefix_path = utils.normalize_path(base_path)
+            if not prefix_path.endswith("/"):
+                prefix_path = f"{prefix_path}/"
+
+            fullname = utils.normalize_path(entry.path).replace(prefix_path, "")
             basename = os.path.splitext(fullname)[0]
             extension = os.path.splitext(fullname)[1]