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

This commit is contained in:
Dijkstra
2024-10-31 19:14:31 +01:00
parent b02bea5480
commit 9a212d13e7
2 changed files with 8 additions and 5 deletions

View File

@@ -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]

View File

@@ -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