Merge pull request #35 from idrirap/dev/issue34

fix: sometimes the preview image extension is prefixed with '.preview'

Author: idrirap

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-lora-auto-trigger-words"
 description = "The aim of these custom nodes is to get an easy access to the tags used to trigger a lora / lycoris. Extract the tags from civitai or from the safetensors metadatas when available."
-version = "1.0.1"
+version = "1.0.2"
 license = "MIT"
 
 [project.urls]

utils.py

@@ -15,11 +15,14 @@ def get_preview_path(name, type):
 
     file_path_no_ext = os.path.splitext(file_path)[0]
     item_image=None
-    for ext in ["png", "jpg", "jpeg", "preview.png"]:
-        has_image = os.path.isfile(file_path_no_ext + "." + ext)
-        if has_image:
-            item_image = f"{file_name}.{ext}"
+    for ext in ["png", "jpg", "jpeg", "gif"]:
+        if item_image is not None:
             break
+        for ext2 in ["", ".preview"]:
+            has_image = os.path.isfile(file_path_no_ext + ext2 + "." + ext)
+            if has_image:
+                item_image = f"{file_name}{ext2}.{ext}"
+                break
 
     return has_image, item_image