* 缺失LoRA模型提示

 * 插件增加Core和ALL两种运行模式

Author: legionfu

ChangeLog.md

@@ -1,5 +1,16 @@
 ## 2.1.2
 
+### Features:
+ * 缺失LoRA模型提示。
+ * 插件增加Core和ALL两种运行模式，Core模式仅支持WebUI的基本参数+ControlNet参数。All模式则会尽可能多的支持页面上的参数，但是缺点是有概率导致UI卡住，无限转圈。
+
+### Bug Fixes:
+ * 修复了一些已知的问题。
+ * 修复了一些img2img模式丢失的参数。
+
+
+## 2.1.2
+
 ### Features:
  * UI组件增加elem_id。
  * flow文件导出时，支持自定义名称。

javascript/state.core.js

@@ -39,7 +39,13 @@ state.core = (function () {
     'resize_mode': 'resize_mode',
     'setting_inpainting_mask_weight': 'setting_inpainting_mask_weight',
     'setting_CLIP_stop_at_last_layers': 'setting_CLIP_stop_at_last_layers',
-    'setting_eta_noise_seed_delta': 'setting_eta_noise_seed_delta'
+    'setting_eta_noise_seed_delta': 'setting_eta_noise_seed_delta',
+    'img2img_mask_blur': 'img2img_mask_blur',
+    'img2img_mask_mode': 'img2img_mask_mode',
+    'img2img_inpainting_fill': 'img2img_inpainting_fill',
+    'img2img_inpaint_full_res_padding': 'img2img_inpaint_full_res_padding',
+    'img2img_inpaint_full_res': 'img2img_inpaint_full_res',
+    'img2img_mask_alpha': 'img2img_mask_alpha'
     //'generation_info_txt2img': 'generation_info_txt2img' // 可能因为是visible=false 所以触发不了onchange事件？
   };
 
@@ -143,6 +149,7 @@ state.core = (function () {
 
   let img_elem_keys=[];
   let ext_list=[];
+  let flow_save_mode = "Core"
 
   function get_imgs_elem_key(){
 
@@ -173,6 +180,13 @@ state.core = (function () {
             .then(response => response.json())
             .then(config => {
               try {
+                
+                try{
+                  flow_save_mode = config['lightdiffusionflow-mode']
+                }catch (error) {
+                  flow_save_mode = "Core"
+                }
+
                 store = new state.Store();
                 store.clearAll();
                 load(config);
@@ -469,7 +483,7 @@ state.core = (function () {
     // }
 
     for (const [name, obj] of Object.entries(state.extensions)) {
-      obj.init();
+      obj.init(flow_save_mode == "Core");
     }
 
   }
@@ -546,6 +560,12 @@ state.core = (function () {
         .then(config => {
           try {
             config.hasSetting = hasSetting
+            
+            try{
+              flow_save_mode = config['lightdiffusionflow-mode']
+            }catch (error) {
+              flow_save_mode = "Core"
+            }
             //console.log(config)
             //restoreTabs(config); // 恢复到最后点击的tab页面
             load(config);

javascript/state.ext.general.js

@@ -411,7 +411,7 @@ function general_ext_main(tab){
     return res
   }
 
-  function init() {
+  function init(core_mode = true) {
     console.log(`------------${cur_tab_name}----init-------`)
 
     let container = gradioApp().getElementById(cur_tab_name+'_script_container'); // main container
@@ -446,6 +446,9 @@ function general_ext_main(tab){
       if(reg.test(title)){title = RegExp.$1} // 匹配 xxx v0.0.0 格式的标题，把后半部分的版本号去掉
 
       if(title == "ControlNet"){title = "Control Net"} // 兼容旧命名
+      else{
+        if(core_mode){continue}
+      }
 
       let ext_name = title.replace(" ","-").toLowerCase()
       console.log(ext_name)

javascript/state.utils.js

@@ -8,6 +8,42 @@ state.utils = {
     //console.log(state.extensions)
     // const button = gradioApp().getElementById("lightdiffusionflow_set_elements");
     // button.click();
+    json_obj = {
+      "state-ext-control-net-txt2img_0-enable": "true",
+      "state-ext-control-net-txt2img_0-canny-high-threshold": "255",
+      "state-ext-control-net-txt2img_3-enable": "true",
+      "state-ext-control-net-txt2img_0-starting-control-step": "1",
+      "state-ext-control-net-txt2img_0-canny-low-threshold": "255",
+      "state-ext-control-net-txt2img_0-control-weight": "2",
+      "state-ext-control-net-txt2img_3-control-type": "OpenPose",
+      "state-ext-control-net-txt2img_0-control-mode": "ControlNet is more important",
+      "state-ext-control-net-txt2img_0-control-type": "Canny",
+      "state-setting_sd_model_checkpoint": "majicmixRealistic_betterV2V25.safetensors [d7e2ac2f4a]",
+      "state-ext-control-net-txt2img_0-preprocessor-resolution": "2048",
+      "state-ext-control-net-txt2img-tab": "ControlNet Unit 3 ",
+      "state-ext-control-net-txt2img_1-enable": "false",
+      "state-tab": "txt2img ",
+      "state-ext-control-net-txt2img-toggled": "true",
+      "txt2img_controlnet_ControlNet-1_input_image": "data:image",
+      "state-ext-control-net-txt2img_0-preprocessor": "canny",
+      "state-ext-control-net-txt2img_0-model": "control_v11p_sd15_canny [d14c016b]",
+      "state-ext-control-net-txt2img_3-preprocessor": "openpose_full",
+      "state-ext-control-net-txt2img_3-model": "control_v11p_sd15_openpose [cab727d4]"
+  }
+
+  console.log(json_obj["state-ext-control-net-txt2img_0-enable"])
+  console.log(json_obj["state-ext-control-net-txt2img_0-enabl1e"])
+
+  },
+
+  fix_old_json_error: function fix_old_json_error(json_obj){
+    
+    cn_index = 1
+    for (let cn_index=1; cn_index < 10; cn_index++){
+      
+    }
+
+    return json_obj
   },
 
   target_is_newer_version: function(cur_version, target_version){

scripts/lightdiffusionflow_config.py

@@ -3,6 +3,7 @@
 PNGINFO_2_LIGHTDIFFUSIONFLOW = {}
 PNGINFO_CN_2_LIGHTDIFFUSIONFLOW = {}
 Image_Components_Key = {}
+LoRAs_In_Use = "loras_in_use"
 
 class OutputPrompt_English:
 
@@ -33,11 +34,22 @@ def missing_extensions(ext_list:[]):
     error_str+="<b style='color:Orange;'>The above Extension Missing Reminder is for reference only. Please determine the necessary plugins based on your actual needs and specific conditions.</b></p> "
     return error_str
 
+  def missing_loras(loras_list:[]):
+    error_str = "Note: <b style='color:Orange;'>Found missing LoRAs.</b></p>"
+    for lora in loras_list:
+      error_str+="<p>- <b style='color:Orange;'>"+lora+"</b></p> "
+    return error_str
+
   def click_to_download(file_name, file_url):
     return f'''<p style="color:Orange;">Click to download \
     <a style ='text-decoration:underline;color:cornflowerblue;', target="_blank", href='{file_url}'> {file_name} </a>
     '''
 
+  def note_for_save_mode():
+    return "Core mode only supports basic parameters for Text-to-Image and Image-to-Image, along with ControlNet parameters. \
+The All mode, on the other hand, aims to support as many parameters as possible on the page, \
+but the downside is that it may occasionally cause the UI to freeze with an infinite loading circle."
+
 class OutputPrompt_Chinese:
 
   def startimport():
@@ -70,6 +82,12 @@ def missing_extensions(ext_list:[]):
     error_str+="<b style='color:Orange;'>以上插件缺失提示仅供参考，请注意辨别实际情况下您所需要安装的插件。</b></p> "
     return error_str
 
+  def missing_loras(loras_list:[]):
+    error_str = "注意, <b style='color:Orange;'>发现缺失的LoRA模型:</b></p>"
+    for lora in loras_list:
+      error_str+="<p>- <b style='color:Orange;'>"+lora+"</b></p> "
+    return error_str
+
   def click_to_download(file_name, file_url):
     name = file_name
     if(name == "ControlNet Models"):
@@ -82,12 +100,17 @@ def click_to_download(file_name, file_url):
     <a style ='text-decoration:underline;color:cornflowerblue;', target="_blank", href='{file_url}'> {name} </a>
     '''
 
+  def note_for_save_mode():
+    return "Core模式仅支持文生图和图生图的基本参数+ControlNet参数。All模式则会尽可能多的支持页面上的参数，但是缺点是有概率导致UI卡住，无限转圈。"
+
+
 OutputPrompt = OutputPrompt_English
+Flow_Save_mode = "Core"
 
 # 改成函数调用，修改配置之后能及时刷新
 def init():
   global PNGINFO_2_LIGHTDIFFUSIONFLOW,PNGINFO_CN_2_LIGHTDIFFUSIONFLOW
-  global OutputPrompt,Image_Components_Key
+  global OutputPrompt,Flow_Save_mode,Image_Components_Key
   # PNG Info的功能除了主要的选项以外其他的都靠第三方插件的主动支持，后续再考虑能否有优化的办法
   #print(parameters_copypaste.paste_fields) 
   PNGINFO_2_LIGHTDIFFUSIONFLOW = {
@@ -133,40 +156,46 @@ def init():
       json_str = f.read()
       webui_settings = json.loads(json_str)
 
-      successed = False
+      # 保存模式
+      try:
+        Flow_Save_mode = webui_settings["lightdiffusionflow-mode"]
+      except:
+        Flow_Save_mode = "Core"
+
+      language_successed = False
       auto_language = False
       try:
         # 优先读取自己的设置
         if(webui_settings['lightdiffusionflow-language'] == "default"):
           auto_language = True
         elif(webui_settings['lightdiffusionflow-language'] == "english"):
           OutputPrompt = OutputPrompt_English
-          successed = True
+          language_successed = True
         else:
           OutputPrompt = OutputPrompt_Chinese
-          successed = True
+          language_successed = True
       except:
         OutputPrompt = OutputPrompt_English
 
       # 如果是default就读取其他设置配合
-      if(auto_language and not successed):
+      if(auto_language and not language_successed):
         # 自带的本地化文件
         localization_files = ["zh_CN", "zh-Hans (Stable) [vladmandic]", "zh-Hans (Stable)",
           "zh-Hans (Testing) [vladmandic]", "zh-Hans (Testing)","chinese-all-1024","chinese-english-1024"]
         try:
           # 如果用户使用了中文汉化文件，插件也默认显示中文
           localization_files.index(webui_settings["localization"])
           OutputPrompt = OutputPrompt_Chinese
-          successed = True
+          language_successed = True
         except:
           pass
 
         # 第三方翻译插件bilingual-localization
-        if(not successed):
+        if(not language_successed):
           try:
             if(webui_settings["bilingual_localization_enabled"] and webui_settings["bilingual_localization_file"] != "None"):
               OutputPrompt = OutputPrompt_Chinese
-              successed = True
+              language_successed = True
           except:
             OutputPrompt = OutputPrompt_English
   except: