Update batch_preview_generator/flux/4_batch_modify_json.py

batch_preview_generator/flux/4_batch_modify_json.py

@@ -6,6 +6,9 @@ from pathlib import Path
 # ================= 配置区 =================
 JSON_FILE = "models.json"
 REMOTE_SHOWCASE_BASE = "https://huggingface.co/waveshaker/loras/tree/main/comfyui_showcase/flux1_dev"
+
+TARGET_MODEL_NAME = REMOTE_SHOWCASE_BASE.rstrip("/").split("/")[-1].lower() 
+
 # ==========================================
 
 def main():
@@ -58,7 +61,7 @@ def main():
         for key, meta in items.items():
             key_filename = os.path.basename(key).replace(".safetensors", "")
             
-            if "flux1_dev" in key.lower() and key_filename.lower() == exact_lora_name.lower():
+            if TARGET_MODEL_NAME in key.lower() and key_filename.lower() == exact_lora_name.lower():
                 new_url = f"{REMOTE_SHOWCASE_BASE}/{folder_name}"
                 
                 if meta.get("showcase") != new_url:
@@ -88,7 +91,7 @@ def main():
     # 🌟 核心新增：统一打印未匹配的清单
     if unmatched_list:
         print("\n" + "🚨" * 20)
-        print(f"⚠️ 发现 {len(unmatched_list)} 个文件夹未能匹配到对应的 flux1_dev 模型：")
+        print(f"⚠️ 发现 {len(unmatched_list)} 个文件夹未能匹配到对应的 {TARGET_MODEL_NAME} 模型：")
         for item in unmatched_list:
             print(f"  ❌ 文件夹: {item['folder']}")
             print(f"     (提取出的名字: {item['extracted_name']})")