Merge pull request #69 from AIGODLIKE/develop

-
AIGODLIKE · Jan 2, 2025 · 43839ce · 43839ce
2 parents b2b3072 + 6c6b30f
commit 43839ce
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/__init__.py b/__init__.py
@@ -213,6 +213,7 @@ def find_model(mtype, model_name) -> Path:
 
 class ThumbnailManager:
     _code_map = {}
+    _path_code_map = {}
     _image_map = {}
     _image_time = {}
 
@@ -221,7 +222,8 @@ def add_code(cls, path: str):
         if path not in cls._image_map:
             code = md5(path.encode()).hexdigest()
             cls._code_map[code] = path
-        return cls._code_map.get(code, "")
+            cls._path_code_map[path] = code
+        return cls._path_code_map.get(path, "")
 
     @classmethod
     def get_path(cls, code: str) -> str: