更新版本V0.0.52

更新版本V0.0.52

Author: weilin9999

README.md

@@ -30,14 +30,16 @@
 
 > 最新更新：2025-04-28
 
-> 0.0.51 公测版本介绍 如果你要使用本插件请务必进我们的官方 QQ 群（1018231382）！
+> 0.0.52 公测版本介绍 如果你要使用本插件请务必进我们的官方 QQ 群（1018231382）！
 >
 > 1. 新增功能：Lora详细界面内可以自定义自己的字段
 >
 > 2. 新增功能：在插件（非节点）UI内鼠标悬浮在Lora卡片上能显示Lora的基本信息
 >
 > 3. 修复了已知的BUG
 >
+> 4. Lora获取C站数据的时候可以一并获取封面了
+>
 
 <details>
 <summary>点击查看往期更多更新内容</summary>

README_EN.md

@@ -26,14 +26,16 @@ Due to limited personal time, the frequency of updating plug-ins will not be ver
 
 > Last updated: 2025-04-28
 
-> 0.0.51 Public Beta Version Introduction: If you want to use this plugin, please be sure to join our official QQ group (1018231382)! 
+> 0.0.52 Public Beta Version Introduction: If you want to use this plugin, please be sure to join our official QQ group (1018231382)! 
 >
 > 1. New feature: Within the Lora detailed interface, you can customize your own fields
 >
 > 2. New feature: When the mouse hoses over a Lora card within the plugin (non-node) UI, the basic information of Lora can be displayed
 >
 > 3. Fixed the known bugs
 >
+> 4. When Lora acquires data from C Station, it can also obtain the cover at the same time
+>
 
 <details>
 <summary>Click here for more updates from the past</summary>

app/server/prompt_api/lora_info.py

@@ -252,6 +252,11 @@ async def get_model_info(file: str,
     if file_hash is not None:
       info_data['sha256'] = file_hash
       should_save = True
+  
+  if len(info_data['images']) != 0 and file not in info_data['images'][0]['url']: # 未设置封面
+    file_name = os.path.basename(file)
+    url = next(filter(lambda x: x['type'] == 'image', info_data['images']), {}).get('url')
+    download_image(url=url, filename=file_name, directory=os.path.dirname(file_path))
 
   if should_save:
     if 'trainedWords' in info_data:
@@ -263,9 +268,32 @@ async def get_model_info(file: str,
 
     # If we're saving, then the UI is likely waiting to see if the refreshed data is coming in.
     # await PromptServer.instance.send("weilin-refreshed-lora-info", {"data": info_data})
+  
 
   return info_data
 
+def download_image(url, filename, directory):
+    try:
+        # 安全处理文件名
+        filename = filename.encode('utf-8', 'ignore').decode('utf-8')
+        _, ext = os.path.splitext(url)
+        filename, _ = os.path.splitext(filename)
+        filepath = os.path.join(directory, f"{filename}{ext}")
+
+        try:
+            resp = requests.get(url, stream=True)
+            resp.raise_for_status()
+            with open(filepath, 'wb') as f:
+                for chunk in resp.iter_content(chunk_size=4096):
+                    f.write(chunk)
+        except Exception as e:
+            print(e)
+            if os.path.exists(filepath):
+                os.remove(filepath)
+
+    except Exception as e:
+        print(f"文件名处理错误: {e}")
+
 
 def _merge_metadata(info_data: dict, data_meta: dict) -> bool:
   """Returns true if data was saved."""

app/server/prompt_api/lora_networks.py

@@ -4,7 +4,6 @@
 from PIL import Image
 import base64
 from io import BytesIO
-import requests
 import asyncio
 import concurrent.futures
 from tqdm import tqdm
@@ -26,27 +25,6 @@
     'metadata',
 ]
 
-def download_image(url, filename, directory):
-    try:
-        # 安全处理文件名
-        filename = filename.encode('utf-8', 'ignore').decode('utf-8')
-        _, ext = os.path.splitext(url)
-        filename, _ = os.path.splitext(filename)
-        filepath = os.path.join(directory, f"{filename}{ext}")
-
-        try:
-            resp = requests.get(url, stream=True)
-            resp.raise_for_status()
-            with open(filepath, 'wb') as f:
-                for chunk in resp.iter_content(chunk_size=4096):
-                    f.write(chunk)
-        except Exception as e:
-            print(e)
-            if os.path.exists(filepath):
-                os.remove(filepath)
-
-    except Exception as e:
-        print(f"文件名处理错误: {e}")
 
 def prepare_lora_item_data(item_path, auto_fetch=False):
     lora_path = folder_paths.get_full_path("loras", item_path)
@@ -65,9 +43,9 @@ def prepare_lora_item_data(item_path, auto_fetch=False):
     if auto_fetch:
         if len(info_data['images']) == 0: # 无数据
             info_data = asyncio.run(get_model_info(item_path, maybe_fetch_civitai=True, maybe_fetch_metadata=True, light=False))
-        if len(info_data['images']) != 0 and item_path not in info_data['images'][0]['url']: # 未设置封面
-            url = next(filter(lambda x: x['type'] == 'image', info_data['images']), {}).get('url')
-            download_image(url=url, filename=file_name, directory=os.path.dirname(lora_path))
+        # if len(info_data['images']) != 0 and item_path not in info_data['images'][0]['url']: # 未设置封面
+        #     url = next(filter(lambda x: x['type'] == 'image', info_data['images']), {}).get('url')
+        #     download_image(url=url, filename=file_name, directory=os.path.dirname(lora_path))
 
     item = {
             "basename": item_path,

dist/javascript/main.entry.js

@@ -1,7 +1,7 @@
 [project]
 name = "weilin-comfyui-tools"
 description = "让你在 ComfyUI 中快捷的使用提示词工具 Quickly use the prompt word tool in ComfyUI"
-version = "0.0.51"
+version = "0.0.52"
 license = {file = "LICENSE"}
 dependencies = []