Upload sd-webui-weight-helper-main using SD-Hub

sd-webui-weight-helper-main/.flake8

@@ -0,0 +1,2 @@
+[flake8]
+max-line-length = 160

sd-webui-weight-helper-main/.github/FUNDING.yml

@@ -0,0 +1,2 @@
+ko_fi: nihedon
+buy_me_a_coffee: nihedon

sd-webui-weight-helper-main/.github/ISSUE_TEMPLATE/bug_report.yml

@@ -0,0 +1,67 @@
+name: Bug Report
+description: Report a problem or unexpected behavior in the application.
+title: "[Bug] Brief description of the issue"
+labels: ["bug"]
+body:
+  - type: markdown
+    attributes:
+      value: |
+        Please fill out the following form to help us reproduce and resolve the issue.
+
+  - type: checkboxes
+    id: environment
+    attributes:
+      label: Operating Environment
+      description: Please select the environment you are using.
+      options:
+        - label: Windows
+        - label: macOS
+        - label: Other (e.g., virtual environment)
+    validations:
+      required: true
+
+  - type: input
+    id: browser
+    attributes:
+      label: Browser
+      description: Please specify the browser you are using (e.g., Chrome, Firefox).
+    validations:
+      required: true
+
+  - type: checkboxes
+    id: webui
+    attributes:
+      label: WebUI
+      description: Please select the WebUI(s) you are using.
+      options:
+        - label: A1111
+        - label: Forge
+        - label: ReForge
+        - label: SD-Next
+        - label: Other
+    validations:
+      required: true
+
+  - type: textarea
+    id: console_log
+    attributes:
+      label: Console Log
+      description: Paste any relevant console logs here.
+    validations:
+      required: false
+
+  - type: textarea
+    id: browser_log
+    attributes:
+      label: Browser Log
+      description: Paste any relevant browser logs here.
+    validations:
+      required: false
+
+  - type: textarea
+    id: description
+    attributes:
+      label: Description of the Issue
+      description: Please describe the issue you encountered in detail.
+    validations:
+      required: true

sd-webui-weight-helper-main/.gitignore

@@ -0,0 +1,164 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+node_modules/
+package-lock.json

sd-webui-weight-helper-main/.prettierrc.json

@@ -0,0 +1,7 @@
+{
+    "semi": true,
+    "singleQuote": true,
+    "tabWidth": 4,
+    "trailingComma": "all",
+    "printWidth": 160
+}

sd-webui-weight-helper-main/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 t.nihei
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

sd-webui-weight-helper-main/README.md

@@ -0,0 +1,69 @@
+[English](README.md) | [日本語](README_JP.md)
+
+<img src="https://github.com/nihedon/sd-webui-weight-helper/assets/66118290/08f8d818-ac5d-4125-b2aa-45c302f626f8" height="600px">
+
+# Weight Helper Extension
+
+## Overview
+
+**Weight Helper** is an extension that allows you to visually adjust the weights of Lora and Lyco with mouse operations.<br>
+You can adjust the magnification of each weight block through the context menu.<br>
+Note: The LoRA Block Weight extension is required to use this extension.
+
+## Installation
+
+1. Open the [Extensions] tab in Stable Diffusion Web UI.
+2. Select [Install from URL].
+3. Enter the following URL and execute the installation.
+```
+https://github.com/nihedon/sd-webui-weight-helper.git
+```
+
+## Usage
+
+1. Right-click on a Lora or Lyco tag to open the context menu.
+   - `<lora:lora_name:1.0>`
+   - `<lyco:lyco_name:1.0>`
+2. Choose "LoRA" or "LyCORIS" according to the tag, and select "SDXL" if it's for the SDXL version.
+3. Move the weight slider to adjust the magnification of each weight block.
+4. UNet, Dyn, Start, and Stop will be hidden by default, but if configured, you can display them by pressing the "show more options" button.
+5. Save your favorite weight settings by clicking the "★" icon at the top left of the context menu.
+   Important: Saved settings are stored in LocalStorage, but any unsaved history will be deleted upon refreshing the screen.
+
+## Features
+
+- Easy to set weight values with mouse operations.
+- Use of preset values from Lora Block Weight is possible.
+- You can check LoRA information and add trigger words from the icons displayed on the preview screen.
+- Information from the readme file attached to LoRA can be displayed on the preview screen. ({loraname}.txt, {loraname}.description.txt)
+- The SD version of LoRA is displayed based on the metadata of LoRA. (If there is metadata text from Civitai, the SD version registered with Civitai will also be displayed.)
+
+## Known Issues
+- The algorithm is deduced from the metadata of LoRA, but it is incomplete. Bug reports and suggestions for the correct algorithm would be appreciated.
+
+## Configuration Options
+
+- Scale adjustment in the context menu.
+- Use of the `execCommand` function for text replacement (Note: execCommand is a deprecated function, but it allows for "undo" and "redo").
+- Display of UNet, Dyn, Start, and Stop sliders.
+- Setting of minimum, maximum, and step values for each weight slider.
+- Detailed settings for LBW's Lora and Lyco blocks.
+- Option to display or hide previews, and settings for preview size and position.
+- Settings for block weight segmentation.
+
+## Guaranteed Operating Environment
+
+- Stable Diffusion AUTOMATIC1111
+- Windows
+- Google Chrome
+
+## Acknowledgments
+
+We would like to express our deep gratitude to **hako-mikan**, the author of **LoRA Block Weight**, for creating this extension.
+```
+https://github.com/hako-mikan/sd-webui-lora-block-weight
+```
+
+## License
+
+This project is published under the [MIT License](LICENSE).

sd-webui-weight-helper-main/README_JP.md

@@ -0,0 +1,69 @@
+[English](README.md) | [日本語](README_JP.md)
+
+<img src="https://github.com/nihedon/sd-webui-weight-helper/assets/66118290/08f8d818-ac5d-4125-b2aa-45c302f626f8" height="600px">
+
+# Weight Helper 拡張機能
+
+## 概要
+
+**Weight Helper** はLoraやLycoのウェイトをマウス操作で視覚的に調整できる拡張機能です。<br>
+コンテキストメニューを通じて各ウェイトブロックの倍率を調整できます。<br>
+注：この拡張機能を利用するにはLoRA Block Weight拡張機能が必要です。
+
+## インストール
+
+1. Stable Diffusion Web UIの[拡張機能]タブを開く
+2. [URLからインストール]を選択
+3. 以下のURLを入力し、インストールを実行
+```
+https://github.com/nihedon/sd-webui-weight-helper.git
+```
+
+## 使い方
+
+1. LoraまたはLycoのタグを右クリックしてコンテキストメニューを開く
+   - `<lora:lora_name:1.0>`
+   - `<lyco:lyco_name:1.0>`
+2. LoRA（またはLyCORIS）に合わせて"LoRA"か"LyCORIS"を選択し、SDXLバージョンの場合は"SDXL"を選択します。
+3. ウェイトスライダーを動かして各ウェイトブロックの倍率を調整します。
+4. UNet、Dyn、Start、Stopは非表示となっているため、設定した場合は"show more options"ボタンを押下して表示できます。
+5. お気に入りのウェイト設定はコンテキストメニュー左上の"★"アイコンをクリックし保存します。
+   重要：保存した場合はLocalStrageに保存されますが、保存していない履歴は画面更新時にすべて削除されます。
+
+## 特徴
+
+- マウス操作でウェイト値を容易に設定できます。
+- Lora Block Weightのプリセット値の利用が可能です。
+- プレビュー画面に表示されるアイコンからLoRA情報の確認、トリガーワードの追加ができます。
+- LoRAに付属するreadmeファイルの情報をプレビュー画面に表示できます。（{loraname}.txt, {loraname}.description.txt）
+- LoRAのメタ情報からLoRAのSDバージョンが表示されます。（Civitaiのメタテキストが存在する場合、civitaiに登録したSDバージョンも表示します。）
+
+## 既知の問題
+- LoRAのメタ情報からアルゴリズムを割り出していますが不完全です。不具合報告や正しいアルゴリズムを提示していただけると助かります(_ _)
+
+## 設定オプション
+
+- コンテキストメニューのスケール調整
+- テキスト置換に `execCommand` 関数の使用（注：execCommandは非推奨関数ですが、「元に戻す」「やり直し」が使えます）
+- UNet、Dyn、Start、およびStopスライダーの表示
+- 各ウェイトスライダーの最小値、最大値、ステップ数の設定
+- LBWのLoraおよびLycoブロックの詳細設定
+- プレビューの表示有無、表示サイズと表示位置の設定
+- ブロックウェイトの区切り設定
+
+## 動作保証環境
+
+- Stable Difusion AUTOMATIC1111
+- Windows
+- Google Chrome
+
+## 謝辞
+
+この拡張機能を作成するにあたり、**LoRA Block Weight** の作者である **hako-mikan** 様に深く感謝申し上げます。
+```
+https://github.com/hako-mikan/sd-webui-lora-block-weight
+```
+
+## ライセンス
+
+このプロジェクトは[MITライセンス](LICENSE)の下で公開されています。

sd-webui-weight-helper-main/backup_before_patch/scripts/analyzer.py

@@ -0,0 +1,189 @@
+import os
+import pickle
+import re
+from typing import Optional
+from scripts import const, env
+
+
+def get_model_type(lora) -> Optional[str]:
+    if env.ui_type == const.UI_TYPE.A1111:
+        sd_version: Optional[str] = lora.sd_version.name
+        if sd_version and env.network:
+            if sd_version == env.network.SdVersion.SD1.name or sd_version == env.network.SdVersion.SD2.name:
+                return "SD"
+            elif sd_version == env.network.SdVersion.SDXL.name:
+                return "SDXL"
+            return sd_version
+    elif env.ui_type == const.UI_TYPE.NEW_FORGE:
+        base_model_version: Optional[str] = lora.metadata.get("ss_base_model_version")
+        if base_model_version:
+            base_model_version = base_model_version.lower()
+            if "sd_" in base_model_version:
+                return "SD"
+            if "sd3" in base_model_version:
+                return "SD"
+            elif "sdxl" in base_model_version:
+                return "SDXL"
+            elif "flux" in base_model_version:
+                return "Flux"
+            return None
+    return None
+
+
+def get_algorithm(metadata, ss_network_args) -> Optional[str]:
+    ss_network_module = metadata.get("ss_network_module")
+    if not ss_network_module or ss_network_module == "Unknown":
+        return None
+
+    conv_dim = float(ss_network_args.get("conv_dim", "-1"))
+    conv_alpha = float(ss_network_args.get("conv_alpha", "-1"))
+    algo = ss_network_args.get("algo", "").lower()
+    unit = ss_network_args.get("unit", "").lower()
+    dora_wd = ss_network_args.get("dora_wd", False)
+
+    if ss_network_module.find("locon.locon_kohya") >= 0:
+        return "LoCon"
+
+    elif ss_network_module.find("lycoris.kohya") >= 0:
+        algoName = ""
+        if dora_wd:
+            algoName = "(DoRA)"
+        elif algo == "lora":
+            algoName += "(LoCon)"
+        elif algo == "locon":
+            if unit:
+                algoName += "(DyLoRA)"
+            else:
+                algoName += "(LoCon)"
+        elif algo == "loha":
+            algoName += "(LoHa)"
+        elif algo == "lokr":
+            algoName += "(Lokr)"
+        elif algo == "ia3":
+            algoName += "(IA3)"
+        elif algo == "full":
+            algoName += "(Full)"
+        elif algo == "glora":
+            algoName += "(GLoRA)"
+
+        return f"LyCORIS{algoName}"
+
+    elif ss_network_module.find("networks.dylora") >= 0:
+        if algo == "dylora" and unit:
+            if conv_dim > 0 or conv_alpha > 0:
+                return "DyLoRA(C3Lier)"
+        return "DyLoRA(LierLa)"
+
+    elif conv_dim > 0 or conv_alpha > 0:
+        return "LoRA(C3Lier)"
+
+    else:
+        return "LoRA(LierLa)"
+
+
+def get_blocks(lora_on_disk, force):
+    baseKeys = __get_lora_state_keys(lora_on_disk.filename, force)
+
+    def formatForDownMidUp(key):
+        blocks = ["BASE"]
+        for key in keys:
+            k, v = key.split("_", 1)
+            if k == "down":
+                k = "IN"
+                v1, v2 = v.split("_")
+                val = int(v1) * 3 + int(v2) + 1
+            elif k == "mid":
+                blocks.append("M00")
+                continue
+            elif k == "up":
+                k = "OUT"
+                v1, v2 = v.split("_")
+                val = int(v1) * 3 + int(v2)
+            blocks.append(f'{k}{int(val):02}')
+        return sorted(blocks)
+
+    def formatForInputMiddleOutput(key):
+        blocks = ["BASE"]
+        for key in keys:
+            k, v = key.split("_")
+            if k == "input":
+                k = "IN"
+            elif k == "middle":
+                blocks.append("M00")
+                continue
+            elif k == "output":
+                k = "OUT"
+            blocks.append(f'{k}{int(v):02}')
+        return sorted(blocks)
+
+    def formatForFlux(key):
+        blocks = ["CLIP", "T5", "IN", "OUT"]
+        for key in keys:
+            k, v = key.split("_")
+            v = int(v)
+            if k == "double":
+                k = "D"
+            if k == "single":
+                k = "S"
+            blocks.append(f'{k}{v:02}')
+        return sorted(blocks)
+
+    keys = [k for k in baseKeys if k.find("lora_unet") == 0]
+    if len(keys) > 0:
+        keys_tmp = [k for k in keys if re.match("^lora_unet_(down|mid|up)_", k)]
+        if len(keys_tmp) > 0:
+            keys = set(["_".join([k.split("_")[2], k.split("_")[4] + "_" + k.split("_")[6]]) for k in keys_tmp if "transformer_blocks" in k])
+            return formatForDownMidUp(keys)
+
+        keys_tmp = [k for k in keys if re.match("^lora_unet_(input|middle|output)_", k)]
+        if len(keys_tmp) > 0:
+            keys = set(["_".join([k.split("_")[2], k.split("_")[4]]) for k in keys_tmp if "transformer_blocks" in k])
+            return formatForInputMiddleOutput(keys)
+
+        keys_tmp = [k for k in keys if re.match("^lora_unet_(single|double)_", k)]
+        if len(keys_tmp) > 0:
+            keys = set(["_".join([k.split("_")[2], k.split("_")[4]]) for k in keys_tmp])
+            return formatForFlux(keys)
+
+        keys_tmp = [k for k in keys if re.match("^lora_unet_joint_blocks_", k)]
+        if len(keys_tmp) > 0:
+            # TODO SD3? 38 blocks
+            return None
+
+    keys = [k for k in baseKeys if "transformer_blocks" in k and "attentions" in k]
+    if len(keys) > 0:
+        # TODO unknown
+        keys = set([".".join(k.split(".")[2:6]) for k in keys])
+        in_blocks = set([f"{k.split('.')[1]}_{k.split('.')[3]}" for k in keys if "down_block" in k])
+        mid_block = set([k.split(".")[2] for k in keys if "mid_block" in k])
+        out_blocks = set([f"{k.split('.')[1]}_{k.split('.')[3]}" for k in keys if "up_block" in k])
+        keys = list(in_blocks) + list(mid_block) + list(out_blocks)
+        return None
+
+    keys = [k for k in baseKeys if "transformer_blocks" in k and k.find("transformer") == 0]
+    if len(keys) > 0:
+        keys = set([".".join(k.split(".")[1:3]) for k in keys])
+        keys = [k.replace("single_transformer_blocks.", "single_") for k in keys]
+        keys = [k.replace("transformer_blocks.", "double_") for k in keys]
+        return formatForFlux(keys)
+
+    return None
+
+
+def __get_lora_state_keys(filename, force):
+    cache_path = f"{os.path.splitext(filename)[0]}.whcache"
+    if not force and os.path.exists(cache_path):
+        try:
+            with open(cache_path, "rb") as f:
+                return pickle.load(f)
+        except Exception:
+            pass
+
+    lora_sd = env.network_lora.load_lora_state_dict(filename)
+    keys = list(lora_sd.keys())
+    try:
+        with open(cache_path, "wb") as f:
+            pickle.dump(keys, f)
+    except Exception:
+        pass
+    return keys

sd-webui-weight-helper-main/backup_before_patch/scripts/api.py

@@ -0,0 +1,145 @@
+
+import ast
+import json
+import os
+from typing import Any, Optional
+from modules import shared
+from scripts import analyzer, const, utils
+
+
+def get_metadata(key, force) -> dict[str, Optional[str]]:
+    ret: dict[str, Optional[str]] = {
+        "modelType": None,
+        "algorithm": None,
+        "baseModel": None,
+        "usingBlocks": None
+    }
+
+    lora_on_disk = utils.get_lora_on_disk(key)
+    if not lora_on_disk:
+        return ret
+
+    ret["modelType"] = analyzer.get_model_type(lora_on_disk)
+
+    metadata = lora_on_disk.metadata
+    ss_network_args = __get_network_args(lora_on_disk)
+    ret["algorithm"] = analyzer.get_algorithm(metadata, ss_network_args)
+
+    model_path, _ = os.path.splitext(lora_on_disk.filename)
+    civitai_info = __find_civitai_info(model_path)
+    ret["baseModel"] = civitai_info.get("baseModel", None)
+
+    if shared.opts.weight_helper_parse_lora_blocks:
+        try:
+            ret["usingBlocks"] = analyzer.get_blocks(lora_on_disk, force)
+        except Exception:
+            pass
+    return ret
+
+
+def get_preview_info(key):
+    ret: dict[str, Optional[bool | str | list[str]]] = {
+        "modelId": None,
+        "triggerWords": [],
+        "negativeTriggerWords": [],
+        "modelName": None,
+        "thumbUrl": None,
+        "hasMetadata": False,
+        "description": None,
+    }
+
+    lora_on_disk = utils.get_lora_on_disk(key)
+    if lora_on_disk:
+        model_path, _ = os.path.splitext(lora_on_disk.filename)
+        ret["modelName"] = lora_on_disk.name
+        ret["thumbUrl"] = __find_preview(model_path)
+        ret["hasMetadata"] = __find_metadata(lora_on_disk)
+        ret["description"] = __find_description(model_path)
+
+        trigger_words = []
+        civitai_info = __find_civitai_info(model_path)
+        ret["modelId"] = civitai_info.get("modelId", None)
+        trigger_words = civitai_info.get("trainedWords", [])
+        trigger_words = ",".join(trigger_words).split(",")
+        ret["triggerWords"] = [w.strip() for w in trigger_words if w]
+
+        lora_json = __get_lora_json(model_path)
+
+        activation_text = lora_json.get("activation text", "")
+        if activation_text:
+            activation_text = activation_text.split(",")
+            activation_text = [w.strip() for w in activation_text if w]
+            ret["triggerWords"] = activation_text
+
+        negative_text = lora_json.get("negative text", "")
+        if negative_text:
+            negative_text = negative_text.split(",")
+            negative_text = [w.strip() for w in negative_text if w]
+            ret["negativeTriggerWords"] = negative_text
+
+        json_description = lora_json.get("description", None)
+        if json_description:
+            ret["description"] = json_description
+
+    return ret
+
+
+def __get_network_args(lora) -> dict[str, Any]:
+    metadata = lora.metadata
+    ss_network_args = metadata.get("ss_network_args", {})
+    if type(ss_network_args) is str:
+        try:
+            ss_network_args = ast.literal_eval(ss_network_args)
+        except Exception:
+            ss_network_args = {}
+    return ss_network_args
+
+
+def __get_lora_json(path) -> dict[str, Any]:
+    if path:
+        lora_json = f"{path}.json"
+        if os.path.exists(lora_json):
+            try:
+                with open(lora_json, "r", encoding="utf-8", errors="replace") as f:
+                    return json.load(f)
+            except OSError:
+                pass
+    return {}
+
+
+def __find_preview(path) -> str:
+    if path:
+        potential_files = sum([[f"{path}.{ext}", f"{path}.preview.{ext}"] for ext in const.ALLOWED_PREVIEW_EXTENSIONS], [])
+        for file in potential_files:
+            if os.path.exists(file):
+                dpi_scale = 1.5
+                return utils.resize_to_base64(file, int(shared.opts.weight_helper_preview_height * dpi_scale))
+    return "./file=html/card-no-preview.png"
+
+
+def __find_metadata(lora) -> bool:
+    return len(lora.metadata) > 0 if lora else False
+
+
+def __find_description(path) -> Optional[str]:
+    if path:
+        for file in [f"{path}.txt", f"{path}.description.txt"]:
+            if os.path.exists(file):
+                try:
+                    with open(file, "r", encoding="utf-8", errors="replace") as f:
+                        return f.read()
+                except OSError:
+                    pass
+    return None
+
+
+def __find_civitai_info(path) -> dict[str, Any]:
+    if path:
+        civitai_info = f"{path}.civitai.info"
+        if os.path.exists(civitai_info):
+            try:
+                with open(civitai_info, "r", encoding="utf-8", errors="replace") as f:
+                    return json.load(f)
+            except OSError:
+                pass
+    return {}

sd-webui-weight-helper-main/backup_before_patch/scripts/const.py

@@ -0,0 +1,11 @@
+import enum
+
+
+API_PREFIX = "/whapi/v1"
+ALLOWED_PREVIEW_EXTENSIONS = ["png", "jpg", "jpeg", "webp", "gif"]
+
+
+class UI_TYPE(enum.Enum):
+    Unknown = 1
+    A1111 = 2
+    NEW_FORGE = 3

sd-webui-weight-helper-main/backup_before_patch/scripts/env.py

@@ -0,0 +1,29 @@
+
+import importlib
+from scripts import const
+
+network = None
+network_lora = None
+ui_type = None
+
+try:
+    from modules_forge import forge_version  # noqa: F401
+    ui_type = const.UI_TYPE.NEW_FORGE
+except ImportError:
+    ui_type = const.UI_TYPE.A1111
+
+try:
+    network = importlib.import_module("extensions-builtin.sd_forge_lora.network")
+    network_lora = importlib.import_module("extensions-builtin.sd_forge_lora.ui_extra_networks_lora").networks
+except Exception:
+    pass
+
+if network_lora is None:
+    try:
+        network = importlib.import_module("extensions-builtin.Lora.network")
+        network_lora = importlib.import_module("extensions-builtin.Lora.ui_extra_networks_lora").networks
+    except Exception:
+        pass
+
+if network_lora is None:
+    ui_type = const.UI_TYPE.Unknown

sd-webui-weight-helper-main/backup_before_patch/scripts/utils.py

@@ -0,0 +1,58 @@
+import base64
+import cv2
+import urllib
+
+import numpy as np
+from scripts import env
+
+
+def get_lora_on_disk(key):
+    if env.network_lora:
+        lora_on_disk = env.network_lora.available_network_aliases.get(key)
+        if not lora_on_disk:
+            # sdnext
+            lora_on_disk = next((v for v in env.network_lora.available_network_aliases.values() if v.alias == key), None)
+        return lora_on_disk
+    return None
+
+
+def imread_unicode(path):
+    try:
+        with open(path, 'rb') as f:
+            data = np.frombuffer(f.read(), np.uint8)
+        img = cv2.imdecode(data, cv2.IMREAD_UNCHANGED)
+        return img
+    except Exception as e:
+        print(f"imread_unicode error: {e}")
+        return None
+
+
+def resize_to_base64(input_path, target_height):
+    output_format = '.webp'
+    quality = 85
+
+    img = imread_unicode(input_path)
+
+    h, w = img.shape[:2]
+    if h < target_height:
+        quoted_filename = urllib.parse.quote(input_path.replace('\\', '/'))
+        return f"./sd_extra_networks/thumb?filename={quoted_filename}"
+
+    scale = target_height / h
+    target_size = (int(w * scale), target_height)
+
+    resized = cv2.resize(img, target_size, interpolation=cv2.INTER_AREA)
+
+    sharpen_kernel = np.array([
+        [0, -0.2, 0],
+        [-0.2, 1.8, -0.2],
+        [0, -0.2, 0]
+    ])
+    sharpened = cv2.filter2D(resized, -1, sharpen_kernel)
+    success, buffer = cv2.imencode(output_format, sharpened, [cv2.IMWRITE_WEBP_QUALITY, quality])
+    if not success:
+        quoted_filename = urllib.parse.quote(input_path.replace('\\', '/'))
+        return f"./sd_extra_networks/thumb?filename={quoted_filename}"
+
+    base64_str = base64.b64encode(buffer).decode('utf-8')
+    return f'data:image/webp;base64,{base64_str}'

sd-webui-weight-helper-main/backup_before_patch/scripts/weight_helper.py

@@ -0,0 +1,82 @@
+import gradio as gr
+from fastapi import FastAPI
+from modules import script_callbacks, shared
+from scripts import api, const
+
+
+def on_app_started(_: gr.Blocks, app: FastAPI):
+    @app.post(const.API_PREFIX + "/get_metadata")
+    async def _(key: str, force: bool = False):
+        return api.get_metadata(key, force)
+
+    @app.post(const.API_PREFIX + "/get_preview_info")
+    async def _(key: str):
+        return api.get_preview_info(key)
+
+
+def on_ui_settings():
+    """
+    Set up the UI settings for the weight helper by adding options to the shared configuration.
+    """
+    section = ('weight_helper', 'Weight Helper')
+
+    shared.options_templates.update(shared.options_section(section, {
+        'weight_helper_enabled': shared.OptionInfo(True, 'Enabled'),
+        'weight_helper_context_menu_scale': shared.OptionInfo(0.8, "Context Menu Scale", gr.Number),
+        'weight_helper_using_execCommand': shared.OptionInfo(True, "Use the deprecated execCommand function to replace text.").info(
+            "You can use 'undo' to revert the text to its previous state, "
+            "but it will no longer be updated in real-time."),
+
+        "weight_helper_slider_length": shared.OptionInfo(160, "Slider Length (px)", gr.Number),
+
+        "weight_helper_TEnc_min": shared.OptionInfo(0, "TEnc Min Value", gr.Number),
+        "weight_helper_TEnc_max": shared.OptionInfo(1, "TEnc Max Value", gr.Number),
+        "weight_helper_TEnc_step": shared.OptionInfo(0.05, "TEnc Step", gr.Number),
+
+        "weight_helper_UNet_min": shared.OptionInfo(0, "UNet Min Value", gr.Number),
+        "weight_helper_UNet_max": shared.OptionInfo(1, "UNet Max Value", gr.Number),
+        "weight_helper_UNet_step": shared.OptionInfo(0.05, "UNet Step", gr.Number),
+
+        "weight_helper_LBW_min": shared.OptionInfo(0, "LBW (LoRA Block Weight) Min Value", gr.Number),
+        "weight_helper_LBW_max": shared.OptionInfo(1, "LBW (LoRA Block Weight) Max Value", gr.Number),
+        "weight_helper_LBW_step": shared.OptionInfo(0.05, "LBW (LoRA Block Weight) Step", gr.Number),
+
+        'weight_helper_show_preview': shared.OptionInfo(True, 'Show Preview'),
+        "weight_helper_preview_height": shared.OptionInfo(400, "Preview Height (px)", gr.Number),
+        "weight_helper_preview_position": shared.OptionInfo("Top Right", "Preview Position", gr.Radio, {
+            "choices": ["Top Right", "Bottom Right", "Top Left", "Bottom Left"]
+        }),
+
+        'weight_helper_parse_lora_blocks': shared.OptionInfo(True, '[experimental] Parse LoRA file to get block information'),
+
+        "weight_helper_LBW_SD_lora_block_points": shared.OptionInfo(
+            "BASE, IN01-IN04, IN05-IN08, M00, OUT03-OUT06, OUT07-OUT11",
+            "Advanced option - LoRA-LierLa block points")
+        .info("default: BASE, IN01-IN04, IN05-IN08, M00, OUT03-OUT06, OUT07-OUT11"),
+        "weight_helper_LBW_SD_lycoris_block_points": shared.OptionInfo(
+            "BASE, IN00-IN05, IN06-IN11, M00, OUT00-OUT05, OUT06-OUT11",
+            "Advanced option - LyCORIS or LoRA-C3Lier block points")
+        .info("default: BASE, IN00-IN05, IN06-IN11, M00, OUT00-OUT05, OUT06-OUT11"),
+
+        "weight_helper_LBW_SDXL_lora_block_points": shared.OptionInfo(
+            "BASE, IN04-IN08, M00, OUT00-OUT05",
+            "Advanced option - SDXL LoRA-LierLa block points")
+        .info("default: BASE, IN04-IN08, M00, OUT00-OUT05"),
+        "weight_helper_LBW_SDXL_lycoris_block_points": shared.OptionInfo(
+            "BASE, IN00-IN03, IN04-IN08, M00, OUT00-OUT03, OUT04-OUT08",
+            "Advanced option - SDXL LyCORIS or LoRA-C3Lier block points")
+        .info("default: BASE, IN00-IN03, IN04-IN08, M00, OUT00-OUT03, OUT04-OUT08"),
+
+        "weight_helper_LBW_Flux_lora_block_points": shared.OptionInfo(
+            "CLIP, T5, IN, OUT, D00-D18, S00-S18, S19-S37",
+            "Advanced option - Flux LoRA-LierLa block points")
+        .info("default: CLIP, T5, IN, OUT, D00-D18, S00-S18, S19-S37"),
+        "weight_helper_LBW_Flux_lycoris_block_points": shared.OptionInfo(
+            "CLIP, T5, IN, OUT, D00-D18, S00-S18, S19-S37",
+            "Advanced option - Flux LyCORIS or LoRA-C3Lier block points")
+        .info("default: CLIP, T5, IN, OUT, D00-D18, S00-S18, S19-S37"),
+    }))
+
+
+script_callbacks.on_ui_settings(on_ui_settings)
+script_callbacks.on_app_started(on_app_started)

sd-webui-weight-helper-main/esbuild.config.mjs

@@ -0,0 +1,16 @@
+import { build } from 'esbuild';
+
+const baseOptions = {
+    entryPoints: ['src/index.ts'],
+    bundle: true,
+    outfile: 'javascript/weight_helper.js',
+    target: 'es2022',
+    format: 'iife',
+    platform: 'browser',
+};
+
+build({
+    ...baseOptions,
+    minify: process.argv.includes('--minify'),
+    sourcemap: process.argv.includes('--sourcemap'),
+}).catch(() => process.exit(1));

sd-webui-weight-helper-main/eslint.config.mjs

@@ -0,0 +1,19 @@
+import { defineConfig } from "eslint/config";
+import globals from "globals";
+import tseslint from "typescript-eslint";
+import prettier from 'eslint-config-prettier';
+
+export default defineConfig([
+    ...tseslint.configs.recommended,
+    {
+        files: ['**/*.ts'],
+        languageOptions: {
+            globals: globals.browser,
+        },
+        rules: {
+            semi: ['error', 'always'],
+            quotes: ['error', 'single'],
+        },
+    },
+    prettier,
+]);

sd-webui-weight-helper-main/javascript/weight_helper.js

@@ -0,0 +1,188 @@
+"use strict";(()=>{var S={lora:"lora",lycoris:"lycoris",Unknown:"Unknown"},y={SD:"SD",SDXL:"SDXL",Flux:"Flux",Unknown:"Unknown"},P={te:"te",unet:"unet",start:"start",stop:"stop",step:"step",lbw:"lbw",lbwe:"lbwe"},m={TENC:"TEnc",UNET:"UNet",START:"Start",STOP:"Stop",LBW:"LBW"};var De={STABLE_DIFFUSION:["BASE","IN00","IN01","IN02","IN03","IN04","IN05","IN06","IN07","IN08","IN09","IN10","IN11","M00","OUT00","OUT01","OUT02","OUT03","OUT04","OUT05","OUT06","OUT07","OUT08","OUT09","OUT10","OUT11"],BLACK_FOREST_LABS:["CLIP","T5","IN","D00","D01","D02","D03","D04","D05","D06","D07","D08","D09","D10","D11","D12","D13","D14","D15","D16","D17","D18","S00","S01","S02","S03","S04","S05","S06","S07","S08","S09","S10","S11","S12","S13","S14","S15","S16","S17","S18","S19","S20","S21","S22","S23","S24","S25","S26","S27","S28","S29","S30","S31","S32","S33","S34","S35","S36","S37","OUT"]},ne={[y.SD]:De.STABLE_DIFFUSION,[y.SDXL]:De.STABLE_DIFFUSION,[y.Flux]:De.BLACK_FOREST_LABS,[y.Unknown]:[]},rt={[y.SD]:{[S.lora]:[1,0,1,1,0,1,1,0,1,1,0,0,0,1,0,0,0,1,1,1,1,1,1,1,1,1],[S.lycoris]:[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1],[S.Unknown]:[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]},[y.SDXL]:{[S.lora]:[1,0,0,0,0,1,1,0,1,1,0,0,0,1,1,1,1,1,1,1,0,0,0,0,0,0],[S.lycoris]:[1,1,1,1,1,1,1,1,1,1,0,0,0,1,1,1,1,1,1,1,1,1,1,0,0,0],[S.Unknown]:[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]},[y.Flux]:{[S.lora]:[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1],[S.lycoris]:[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1],[S.Unknown]:[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]},[y.Unknown]:{[S.lora]:[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],[S.lycoris]:[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],[S.Unknown]:[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]}},nt={[y.SD]:{[S.lora]:"XYZ(17)",[S.lycoris]:"XYZ(26)",[S.Unknown]:"XYZ(26)"},[y.SDXL]:{[S.lora]:"XYZ(12)",[S.lycoris]:"XYZ(20)",[S.Unknown]:"XYZ(26)"},[y.Flux]:{[S.lora]:"XYZ(19)",[S.lycoris]:"XYZ(19)",[S.Unknown]:"XYZ(19)"},[y.Unknown]:{[S.lora]:"XYZ(17)",[S.lycoris]:"XYZ(26)",[S.Unknown]:"XYZ(26)"}},K={"":y.Unknown,SD:y.SD,SDXL:y.SDXL,Flux:y.Flux},J={"":S.Unknown,"LoRA(LierLa)":S.lora,"LyCORIS,etc":S.lycoris},be="./file=html/card-no-preview.png";var ye={[m.TENC]:{min:0,max:0,default:1,step:.5},[m.UNET]:{min:0,max:0,default:0,step:.5},[m.START]:{min:0,max:0,default:0,step:1},[m.STOP]:{min:0,max:0,default:0,step:1},[m.LBW]:{min:0,max:0,default:1,step:.5}},st={[y.SD]:{[S.lora]:[["BASE","IN01-IN04","IN05-IN08","M00","OUT03-OUT06","OUT07-OUT11"]],[S.lycoris]:[["BASE","IN00-IN05","IN06-IN11","M00","OUT00-OUT05","OUT06-OUT11"]],[S.Unknown]:[[]]},[y.SDXL]:{[S.lora]:[["BASE","IN04-IN08","M00","OUT00-OUT05"]],[S.lycoris]:[["BASE","IN00-IN03","IN04-IN08","M00","OUT00-OUT03","OUT04-OUT08"]],[S.Unknown]:[[]]},[y.Flux]:{[S.lora]:[["CLIP","T5","IN","OUT"],["D00-D18"],["S00-S18"],["S19-S37"]],[S.lycoris]:[["CLIP","T5","IN","OUT"],["D00-D18"],["S00-S18"],["S19-S37"]],[S.Unknown]:[[]]},[y.Unknown]:{[S.lora]:[[]],[S.lycoris]:[[]],[S.Unknown]:[[]]}};function at(){let e=/((BASE|MID|M00|CLIP|T5|(IN|OUT|D|S)[0-9]{2}(-(IN|OUT|D|S)[0-9]{2})?) *(, *|$))+/;for(let t of Object.values(K))if(!(t===y.Unknown||t===y.Flux)){for(let o of Object.values(J))if(o!==S.Unknown)try{let r=opts[`weight_helper_LBW_${t}_${o}_block_points`];if(r=r.replace("MID","M00"),e.exec(r)){let n=r.split(",").map(a=>a.trim().replace(/\d+/g,s=>s.length===1?`0${s}`:s));st[t][o]=[n]}}catch(r){console.warn(`${t}_${o} block definition format is invalid.`,r)}}for(let t of[m.TENC,m.UNET,m.LBW])ye[t].min=+opts[`weight_helper_${t}_min`],ye[t].max=+opts[`weight_helper_${t}_max`],ye[t].step=+opts[`weight_helper_${t}_step`]}function T(e){return ye[e]}function we(e){return e in ne?ne[e]:ne[y.Unknown]}function Q(e,t){let o=V(e,t);return ne[e||y.Unknown].filter((r,n)=>o[n]===1)}function V(e,t){return rt[e||y.Unknown][t||S.Unknown]}function N(e,t){return st[e||y.Unknown][t||S.Unknown]}function it(e,t){return nt[e||y.Unknown][t||S.Unknown]}var A={},E={};function O(e){return A[e]??[]}function ee(e,t){t={loraParams:t.loraParams,selectedLoraBlockType:t.selectedLoraBlockType,selectedModelType:t.selectedModelType},e in A?(A[e]=A[e].filter(o=>o.loraParams!==t.loraParams),A[e].push(t)):A[e]=[t]}function lt(e){if(e in A){let t=E[e];A[e]=A[e].filter(o=>t&&t.has(o.loraParams))}}function ct(e,t){e in E?E[e].add(t):E[e]=new Set([t])}function gt(e,t){e in E&&E[e].delete(t)}function U(e,t){return e in E?E[e].has(t):!1}function dt(){localStorage.setItem("wh://histories",JSON.stringify(A));let e={};for(let t in E)e[t]=Array.from(E[t]);localStorage.setItem("wh://locks",JSON.stringify(e))}function pt(){let e=localStorage.getItem("wh://histories");e&&(A=JSON.parse(e));let t=localStorage.getItem("wh://locks");if(t){let o=JSON.parse(t);for(let r in o)E[r]=new Set(o[r])}}function G(e,t,o,r){let n=new Set(Q(e,t)),a=Object.entries(r).filter(([l])=>n.has(l)).map(l=>l[1].value).join(","),s=Object.entries(o).find(([,l])=>l===a);return s?s[0]:""}function Te(e){let t,o=typeof window.TAC_CFG<"u"&&window.TAC_CFG,r,n=typeof window.pilotIsActive<"u"&&window.pilotIsActive;try{o&&(t=window.TAC_CFG.activeIn.global,window.TAC_CFG.activeIn.global=!1),n&&(r=window.pilotIsActive,window.pilotIsActive=!1),e()}finally{o&&(window.TAC_CFG.activeIn.global=t),n&&(window.pilotIsActive=r)}}var ht={},ut={};function ft(e){return ht[e]}function mt(e,t){ht[e]={metadataState:t.metadataState,selectedLoraBlockType:t.weightState.selectedLoraBlockType,selectedModelType:t.weightState.selectedModelType,usingBlocks:t.usingBlocks}}function St(e){return ut[e]}function bt(e,t){ut[e]=t}var yt,wt,Tt,Mt,kt,xt={},vt=0;function se(){return yt}function Lt(e){yt=e}function ae(){return wt}function Ct(e){wt=e}function He(){return Tt}function $t(e){Tt=e}function Re(){return Mt}function Me(e){Mt=e}function _t(){return kt}function Bt(e){kt=e}function ke(){return vt||0}function At(e){vt=+gradioApp().querySelector(`div:is([id^='tab_${e}'][class*='tabitem'])`).querySelector('div:is([id$="_steps"]) input[type=number]').value}function $(e,t){return xt[e||y.Unknown][t||S.Unknown]}function Et(e){let n=(gradioApp().querySelector(`div:is([id^='tab_${e}'][class*='tabitem'])`).querySelector("#lbw_ratiospreset textarea").value??"").split(`
+`).filter(s=>s.trim()!==""),a={};for(let s of Object.values(y)){a[s]={};for(let l of Object.values(S)){let i={},g={};a[s][l]=i;let u=V(s,l).filter(c=>c==1).length;for(let c of n){let h=c.split(":");h.length==2&&h[1].split(",").length==u&&(i[h[0]]=h[1],g[h[1]]=h[0])}}}xt=a}var Pt=e=>e.toLowerCase(),z=e=>e?"":"display: none",D=e=>e?"":"visibility: hidden",I=(e,t=void 0)=>{if(e.uiState.isWaiting)return!0;switch(t){case m.TENC:case m.UNET:case m.START:case m.STOP:case void 0:return!1}return e.usingBlocks!==void 0&&!e.usingBlocks.has(t)};function xe(e){let t=[];return e.split(",").map(o=>o.trim()).forEach(o=>{let r=o.match(/^([A-Z]+)(\d+)-\1(\d+)$/);if(!r){t.push(o);return}let[,n,a,s]=r,l=parseInt(a),i=parseInt(s),g=l;for(;t.push(`${n}${g.toString().padStart(2,"0")}`),g!=i;)g+=l<i?1:-1}),t}var wo=/<(lora|lyco):([^:]+):([^>]+)>/;function It(e){let t=e.target;if(!opts.weight_helper_enabled)return;let o=window.getSelection()?.toString();if(o)return;let r=t.value,n=t.selectionStart,a=r.lastIndexOf("<",n-1),s=r.indexOf(">",n);if(a<0||s<0||(o=r.substring(a,s+1),(o.match(/</g)||[]).length!=1||(o.match(/>/g)||[]).length!=1))return;n=a;let l=wo.exec(o);if(l){e.preventDefault();let i=l[1].toLowerCase(),g=l[2],u=l[3],c=t.closest("[id^='tab_'][class*='tabitem']").id.split("_")[1],h=n+l.index,p=h+l.input.trim().length;return{tabId:c,textarea:t,selectionStart:h,selectionEnd:p,namespace:i,loraName:g,loraParams:u,top:e.pageY+15,left:e.pageX}}}function X(e){let t=ae(),o=He(),r=Re();t.value=t.value.substring(0,o)+e+t.value.substring(r),Me(o+e.length)}function To(e){Te(()=>{let t=ae(),o=He(),r=Re();t.focus(),t.setSelectionRange(o,r),document.execCommand("insertText",!1,e)})}function _(e,t){let o=String(t.weights[m.TENC].value),r=0,n=0;for(let p of[m.UNET]){let w=t.weights[p],b=T(p).default,f=+w.value;(w.checkState||w.checkState===void 0&&f!=b)&&(n===r?o+=`:${f}`:o+=`:${Pt(p)}=${f}`,r++),n++}let a=T(m.START).default,s=t.weights[m.START].value,l=T(m.STOP).default,i=t.weights[m.STOP].value;i<0&&(i=T(m.STOP).default),s!=a&&i!=l?o+=`:step=${s}-${i}`:s!=a?o+=`:start=${s}`:i!=l&&(o+=`:stop=${i}`);let{selectedModelType:g,selectedLoraBlockType:u}=t,c=t.xyzMode,h=o;if(c)o+=":lbw=XYZ",h+=":lbw=XYZ";else if(g!==y.Unknown){let p=V(g,u).map((b,f)=>{if(b===1)return t.weights[we(g)[f]]}).filter(b=>b!==void 0),w=T(m.LBW).default;if(!p.every(b=>b.value===w)){let b=p.map(C=>C.value).join(",");o+=`:lbw=${b}`;let f=$(g,u),M=Object.entries(f).find(([,C])=>b===C);M?h+=`:lbw=${M[0]}`:h+=`:lbw=${b}`}}return t.lbwe&&(o+=`:lbwe=${t.lbwe}`,h+=`:lbwe=${t.lbwe}`),{loraParams:o,loraDefine:`<lora:${e}:${h}>`}}function le(e,t=!1){let o=_(e.loraName,e.weightState);opts.weight_helper_using_execCommand?e.srcLoraParams!==o.loraParams&&To(o.loraDefine):t&&X(_t()),mt(e.loraName,e),e.weightState.xyzMode||ee(e.loraName,{loraParams:o.loraParams,...e.weightState}),dt()}var d,ve=[],Le=[];function v(){ve.forEach(e=>e(d))}function Mo(e){Le.forEach(t=>t(e))}function Wt(e){return ve.push(e),()=>{let t=ve.indexOf(e);t>-1&&ve.splice(t,1)}}function Nt(e){return Le.push(e),()=>{let t=Le.indexOf(e);t>-1&&Le.splice(t,1)}}function L(){return d}function Ut(e){d=e,v()}function Dt(e){d=e,v()}function Ot(e){d=e,v()}function Ht(e,t){Mo({top:e,left:t})}function Ce(e){let t={...d,metadataState:{...d.metadataState,algorithm:e.algorithm||"Unknown",baseModel:e.baseModel||"Unknown",modelType:e.modelType||"Unknown"},weightState:{...d.weightState,selectedModelType:e.selectedModelType,weights:e.weights},blockGroups:e.blockGroups,lbwPresets:e.lbwPresets,uiState:{...d.uiState,isWaiting:!1}};e.usingBlocks===null?t.usingBlocks=void 0:e.usingBlocks&&(t.usingBlocks=new Set(e.usingBlocks)),d=t,v()}function ce(e){d={...d,uiState:{...d.uiState,isWaiting:e}},v()}function Ve(e){let t={...d,weightState:{...d.weightState,selectedLoraBlockType:e.selectedLoraBlockType,selectedModelType:e.selectedModelType,weights:e.weights},blockGroups:e.blockGroups,lbwPresets:e.lbwPresets};opts.weight_helper_using_execCommand||X(_(t.loraName,t.weightState).loraDefine),d=t,v()}function Rt(){let e={...d,weightState:{...d.weightState,xyzMode:!d.weightState.xyzMode}};opts.weight_helper_using_execCommand||X(_(e.loraName,e.weightState).loraDefine),d=e,v()}function Vt(e){let t={...d.weightState,weights:{...d.weightState.weights,...e.weights}},o=_(d.loraName,t),r=U(d.loraName,o.loraParams),n={...d,weightState:t,preset:e.preset,lock:r};opts.weight_helper_using_execCommand||X(o.loraDefine),d=n,v()}function Gt(){let e=_(d.loraName,d.weightState).loraParams,t=U(d.loraName,e);t?gt(d.loraName,e):ct(d.loraName,e),d={...d,lock:!t},v()}function Ge(e){let{locked:t,historyIndex:o,preset:r}=e,{selectedLoraBlockType:n,selectedModelType:a,weights:s,lbwe:l,xyzMode:i}=e.weightState,g={...d.weightState,weights:{...d.weightState.weights,...s},selectedLoraBlockType:n,selectedModelType:a,lbwe:l,xyzMode:i},u={...d,historyIndex:o,preset:r,weightState:g,lock:t};opts.weight_helper_using_execCommand||X(_(u.loraName,u.weightState).loraDefine),d=u,v()}function zt(e){d={...d,historyIndex:e},v()}function $e(e){d.weightState.xyzMode=!1;let t={...d.weightState.weights,[e.block]:{...d.weightState.weights[e.block],value:e.value,sliderMin:Math.min(e.value,d.weightState.weights[e.block].sliderMin),sliderMax:Math.max(e.value,d.weightState.weights[e.block].sliderMax),checkState:e.checkState}},o={...d.weightState,weights:t,xyzMode:!1},r=_(d.loraName,o),n=U(d.loraName,r.loraParams),a={...d,weightState:o,preset:G(d.weightState.selectedModelType,d.weightState.selectedLoraBlockType,d.lbwPresets,t),lock:n};opts.weight_helper_using_execCommand||X(r.loraDefine),d=a,v()}function Xt(){d={...d,uiState:{...d.uiState,isMoreButtonVisible:!1}},v()}function Ft(){lt(d.loraName),d.weightState.xyzMode||ee(d.loraName,{loraParams:_(d.loraName,d.weightState).loraParams,...d.weightState});let e=O(d.loraName).length-1;d={...d,historyIndex:e<0?0:e},v()}function Yt(e){d={...d,previewState:e},v()}function ze(e){d={...d,previewState:{...d.previewState,isDescriptionVisible:e}},v()}function jt(){d={...d,previewState:{...d.previewState,isTagInsertButtonVisible:!1}},v()}function _e(){return{loraName:"",metadataState:void 0,srcLoraParams:"",weightState:{selectedLoraBlockType:S.lora,selectedModelType:y.SD,weights:{},lbwe:"",xyzMode:!1},blockGroups:[],usingBlocks:void 0,preset:"",lbwPresets:{},uiState:{isVisible:!1,scale:1,isMoreButtonVisible:!0,isWaiting:!1},lock:!1,historyIndex:0,previewState:{modelId:"",triggerWords:[],negativeTriggerWords:[],modelName:"",thumbUrl:"",hasMetadata:!1,description:"",isDescriptionVisible:!1,isTagInsertButtonVisible:!1}}}function Zt(e,t,o){let r=y.Unknown,n=S.Unknown,a=O(t);if(a&&a.length>0){let p=a[a.length-1];r=p.selectedModelType??y.Unknown,n=p.selectedLoraBlockType??S.Unknown}let s,l,i=ft(t);i&&(s=i.metadataState,r=i.selectedModelType??y.Unknown,n=i.selectedLoraBlockType??S.Unknown,l=i.usingBlocks);let g=ko(e,o,r,n),u=$(g.selectedModelType,g.selectedLoraBlockType),c=_(t,g).loraParams,h=!1;for(let p of[m.UNET,m.START,m.STOP]){let w=g.weights[p],b=T(p).default,f=+w.value;if(!(w.checkState||w.checkState===void 0&&f!=b)){h=!0;break}}return{loraName:t,metadataState:s,srcLoraParams:c,weightState:g,blockGroups:N(g.selectedModelType,g.selectedLoraBlockType),usingBlocks:l,uiState:{isVisible:!0,scale:opts.weight_helper_context_menu_scale,isMoreButtonVisible:h,isWaiting:!s},previewState:xo(t),preset:G(g.selectedModelType,g.selectedLoraBlockType,u,g.weights),lbwPresets:u,lock:U(t,c),historyIndex:a.length-1}}function ko(e,t,o,r){let n=Be(t,o,r),{weights:a,lbwe:s,xyzMode:l}=n;return r===S.Unknown&&(r=e==="lora"?S.lora:S.lycoris),{selectedLoraBlockType:r,selectedModelType:o,weights:a,lbwe:s,xyzMode:l}}function Be(e,t,o){let r=e.split(":"),n={};for(let b=0;b<r.length;b++){let f,M;if(r[b].indexOf("=")>=0){let C=r[b].split("=");f=C[0].toLowerCase(),M=C[1]}else f=[P.te,P.unet][b],M=r[b];n[f]=M}let a=[];for(let b of Object.values(K))for(let f of Object.values(J))a.push({modelType:b,loraBlockType:f,presets:$(b,f),masks:V(b,f)});let s=T(m.TENC).default,l,i=T(m.START).default,g=T(m.STOP).default,u=[],c="",h=!1;Object.entries(n).forEach(([b,f])=>{switch(b){case P.lbw:if(f==="XYZ"){h=!0;break}let M=f.split(",");for(let W of a){if(M.length===1)if(f in W.presets)M=W.presets[f].split(",");else continue;if(M.length===W.masks.filter(q=>q==1).length){t=W.modelType,o=W.loraBlockType;let q=0;for(let Ue of W.masks)Ue?(u.push(+M[q]),q++):u.push(0);break}}break;case P.step:let C=f.split("-");i=Math.round(+C[0]),g=Math.round(+C[1]);break;case P.lbwe:c=f;break;case P.te:s=+f;break;case P.unet:l=+f;break;case P.start:i=Math.round(+f);break;case P.stop:g=Math.round(+f);break}});let p=!0;l===void 0&&(l=T(m.UNET).default,p=!1);let w={};if(t!==y.Unknown){let b=we(t);if(u.length===0)for(let f=0;f<b.length;f++)u.push(T(m.LBW).default);u.forEach((f,M)=>{w[b[M]]={initValue:f,value:f,sliderMin:Math.min(f,T(m.LBW).min),sliderMax:Math.max(f,T(m.LBW).max)}})}return{selectedLoraBlockType:o,selectedModelType:t,weights:{[m.TENC]:{initValue:s,value:s,sliderMin:Math.min(s,T(m.TENC).min),sliderMax:Math.max(s,T(m.TENC).max)},[m.UNET]:{initValue:l,value:l,sliderMin:Math.min(l,T(m.UNET).min),sliderMax:Math.max(l,T(m.UNET).max),initCheckState:p,checkState:p},[m.START]:{initValue:i,value:i,sliderMin:Math.min(i,T(m.START).min),sliderMax:Math.max(i,T(m.START).max)},[m.STOP]:{initValue:g,value:g,sliderMin:Math.min(g,T(m.STOP).min),sliderMax:Math.max(g,T(m.STOP).max)},...w},lbwe:c,xyzMode:h}}function xo(e){let t=St(e);return t||{modelId:"",triggerWords:[""],negativeTriggerWords:[""],modelName:"",thumbUrl:be,hasMetadata:!1,description:"",isDescriptionVisible:!1,isTagInsertButtonVisible:!1}}var Ae=null,Ee=null;function Pe(){return Ae&&Ae.abort(),Ae=new AbortController,Ae}function qt(){return Ee&&Ee.abort(),Ee=new AbortController,Ee}async function Ie(e,t,o){let r=await fetch(`/whapi/v1/get_metadata?key=${encodeURIComponent(e)}&force=${t}`,{method:"POST",signal:o});if(!r.ok)throw new Error(`Failed to set up API endpoints. Please reload the page.: ${r.status}`);return await r.json()}async function Kt(e,t){let o=await fetch(`/whapi/v1/get_preview_info?key=${encodeURIComponent(e)}`,{method:"POST",signal:t});if(!o.ok)throw new Error(`Failed to set up API endpoints. Please reload the page.: ${o.status}`);return{...await o.json(),isDescriptionVisible:!1,isTagInsertButtonVisible:!0}}function F(e,t){let o={...e},r=[];return t.forEach(n=>{n.forEach(a=>{xe(a).forEach(l=>{r.includes(l)||r.push(l)})})}),r.forEach(n=>{o[n]||(o[n]={initValue:T(m.LBW).default,value:T(m.LBW).default,sliderMin:T(m.LBW).min,sliderMax:T(m.LBW).max})}),o}var Ke=globalThis,We=Ke.trustedTypes,Jt=We?We.createPolicy("lit-html",{createHTML:e=>e}):void 0,no="$lit$",R=`lit$${Math.random().toFixed(9).slice(2)}$`,so="?"+R,vo=`<${so}>`,Z=document,de=()=>Z.createComment(""),pe=e=>e===null||typeof e!="object"&&typeof e!="function",Je=Array.isArray,Lo=e=>Je(e)||typeof e?.[Symbol.iterator]=="function",Xe=`[ 	
+\f\r]`,ge=/<(?:(!--|\/[^a-zA-Z])|(\/?[a-zA-Z][^>\s]*)|(\/?$))/g,Qt=/-->/g,eo=/>/g,Y=RegExp(`>|${Xe}(?:([^\\s"'>=/]+)(${Xe}*=${Xe}*(?:[^ 	
+\f\r"'\`<>=]|("|')|))|$)`,"g"),to=/'/g,oo=/"/g,ao=/^(?:script|style|textarea|title)$/i,Qe=e=>(t,...o)=>({_$litType$:e,strings:t,values:o}),k=Qe(1),rr=Qe(2),nr=Qe(3),he=Symbol.for("lit-noChange"),x=Symbol.for("lit-nothing"),ro=new WeakMap,j=Z.createTreeWalker(Z,129);function io(e,t){if(!Je(e)||!e.hasOwnProperty("raw"))throw Error("invalid template strings array");return Jt!==void 0?Jt.createHTML(t):t}var Co=(e,t)=>{let o=e.length-1,r=[],n,a=t===2?"<svg>":t===3?"<math>":"",s=ge;for(let l=0;l<o;l++){let i=e[l],g,u,c=-1,h=0;for(;h<i.length&&(s.lastIndex=h,u=s.exec(i),u!==null);)h=s.lastIndex,s===ge?u[1]==="!--"?s=Qt:u[1]!==void 0?s=eo:u[2]!==void 0?(ao.test(u[2])&&(n=RegExp("</"+u[2],"g")),s=Y):u[3]!==void 0&&(s=Y):s===Y?u[0]===">"?(s=n??ge,c=-1):u[1]===void 0?c=-2:(c=s.lastIndex-u[2].length,g=u[1],s=u[3]===void 0?Y:u[3]==='"'?oo:to):s===oo||s===to?s=Y:s===Qt||s===eo?s=ge:(s=Y,n=void 0);let p=s===Y&&e[l+1].startsWith("/>")?" ":"";a+=s===ge?i+vo:c>=0?(r.push(g),i.slice(0,c)+no+i.slice(c)+R+p):i+R+(c===-2?l:p)}return[io(e,a+(e[o]||"<?>")+(t===2?"</svg>":t===3?"</math>":"")),r]},ue=class e{constructor({strings:t,_$litType$:o},r){let n;this.parts=[];let a=0,s=0,l=t.length-1,i=this.parts,[g,u]=Co(t,o);if(this.el=e.createElement(g,r),j.currentNode=this.el.content,o===2||o===3){let c=this.el.content.firstChild;c.replaceWith(...c.childNodes)}for(;(n=j.nextNode())!==null&&i.length<l;){if(n.nodeType===1){if(n.hasAttributes())for(let c of n.getAttributeNames())if(c.endsWith(no)){let h=u[s++],p=n.getAttribute(c).split(R),w=/([.?@])?(.*)/.exec(h);i.push({type:1,index:a,name:w[2],strings:p,ctor:w[1]==="."?Ye:w[1]==="?"?je:w[1]==="@"?Ze:re}),n.removeAttribute(c)}else c.startsWith(R)&&(i.push({type:6,index:a}),n.removeAttribute(c));if(ao.test(n.tagName)){let c=n.textContent.split(R),h=c.length-1;if(h>0){n.textContent=We?We.emptyScript:"";for(let p=0;p<h;p++)n.append(c[p],de()),j.nextNode(),i.push({type:2,index:++a});n.append(c[h],de())}}}else if(n.nodeType===8)if(n.data===so)i.push({type:2,index:a});else{let c=-1;for(;(c=n.data.indexOf(R,c+1))!==-1;)i.push({type:7,index:a}),c+=R.length-1}a++}}static createElement(t,o){let r=Z.createElement("template");return r.innerHTML=t,r}};function oe(e,t,o=e,r){if(t===he)return t;let n=r!==void 0?o._$Co?.[r]:o._$Cl,a=pe(t)?void 0:t._$litDirective$;return n?.constructor!==a&&(n?._$AO?.(!1),a===void 0?n=void 0:(n=new a(e),n._$AT(e,o,r)),r!==void 0?(o._$Co??=[])[r]=n:o._$Cl=n),n!==void 0&&(t=oe(e,n._$AS(e,t.values),n,r)),t}var Fe=class{constructor(t,o){this._$AV=[],this._$AN=void 0,this._$AD=t,this._$AM=o}get parentNode(){return this._$AM.parentNode}get _$AU(){return this._$AM._$AU}u(t){let{el:{content:o},parts:r}=this._$AD,n=(t?.creationScope??Z).importNode(o,!0);j.currentNode=n;let a=j.nextNode(),s=0,l=0,i=r[0];for(;i!==void 0;){if(s===i.index){let g;i.type===2?g=new fe(a,a.nextSibling,this,t):i.type===1?g=new i.ctor(a,i.name,i.strings,this,t):i.type===6&&(g=new qe(a,this,t)),this._$AV.push(g),i=r[++l]}s!==i?.index&&(a=j.nextNode(),s++)}return j.currentNode=Z,n}p(t){let o=0;for(let r of this._$AV)r!==void 0&&(r.strings!==void 0?(r._$AI(t,r,o),o+=r.strings.length-2):r._$AI(t[o])),o++}},fe=class e{get _$AU(){return this._$AM?._$AU??this._$Cv}constructor(t,o,r,n){this.type=2,this._$AH=x,this._$AN=void 0,this._$AA=t,this._$AB=o,this._$AM=r,this.options=n,this._$Cv=n?.isConnected??!0}get parentNode(){let t=this._$AA.parentNode,o=this._$AM;return o!==void 0&&t?.nodeType===11&&(t=o.parentNode),t}get startNode(){return this._$AA}get endNode(){return this._$AB}_$AI(t,o=this){t=oe(this,t,o),pe(t)?t===x||t==null||t===""?(this._$AH!==x&&this._$AR(),this._$AH=x):t!==this._$AH&&t!==he&&this._(t):t._$litType$!==void 0?this.$(t):t.nodeType!==void 0?this.T(t):Lo(t)?this.k(t):this._(t)}O(t){return this._$AA.parentNode.insertBefore(t,this._$AB)}T(t){this._$AH!==t&&(this._$AR(),this._$AH=this.O(t))}_(t){this._$AH!==x&&pe(this._$AH)?this._$AA.nextSibling.data=t:this.T(Z.createTextNode(t)),this._$AH=t}$(t){let{values:o,_$litType$:r}=t,n=typeof r=="number"?this._$AC(t):(r.el===void 0&&(r.el=ue.createElement(io(r.h,r.h[0]),this.options)),r);if(this._$AH?._$AD===n)this._$AH.p(o);else{let a=new Fe(n,this),s=a.u(this.options);a.p(o),this.T(s),this._$AH=a}}_$AC(t){let o=ro.get(t.strings);return o===void 0&&ro.set(t.strings,o=new ue(t)),o}k(t){Je(this._$AH)||(this._$AH=[],this._$AR());let o=this._$AH,r,n=0;for(let a of t)n===o.length?o.push(r=new e(this.O(de()),this.O(de()),this,this.options)):r=o[n],r._$AI(a),n++;n<o.length&&(this._$AR(r&&r._$AB.nextSibling,n),o.length=n)}_$AR(t=this._$AA.nextSibling,o){for(this._$AP?.(!1,!0,o);t!==this._$AB;){let r=t.nextSibling;t.remove(),t=r}}setConnected(t){this._$AM===void 0&&(this._$Cv=t,this._$AP?.(t))}},re=class{get tagName(){return this.element.tagName}get _$AU(){return this._$AM._$AU}constructor(t,o,r,n,a){this.type=1,this._$AH=x,this._$AN=void 0,this.element=t,this.name=o,this._$AM=n,this.options=a,r.length>2||r[0]!==""||r[1]!==""?(this._$AH=Array(r.length-1).fill(new String),this.strings=r):this._$AH=x}_$AI(t,o=this,r,n){let a=this.strings,s=!1;if(a===void 0)t=oe(this,t,o,0),s=!pe(t)||t!==this._$AH&&t!==he,s&&(this._$AH=t);else{let l=t,i,g;for(t=a[0],i=0;i<a.length-1;i++)g=oe(this,l[r+i],o,i),g===he&&(g=this._$AH[i]),s||=!pe(g)||g!==this._$AH[i],g===x?t=x:t!==x&&(t+=(g??"")+a[i+1]),this._$AH[i]=g}s&&!n&&this.j(t)}j(t){t===x?this.element.removeAttribute(this.name):this.element.setAttribute(this.name,t??"")}},Ye=class extends re{constructor(){super(...arguments),this.type=3}j(t){this.element[this.name]=t===x?void 0:t}},je=class extends re{constructor(){super(...arguments),this.type=4}j(t){this.element.toggleAttribute(this.name,!!t&&t!==x)}},Ze=class extends re{constructor(t,o,r,n,a){super(t,o,r,n,a),this.type=5}_$AI(t,o=this){if((t=oe(this,t,o,0)??x)===he)return;let r=this._$AH,n=t===x&&r!==x||t.capture!==r.capture||t.once!==r.once||t.passive!==r.passive,a=t!==x&&(r===x||n);n&&this.element.removeEventListener(this.name,this,r),a&&this.element.addEventListener(this.name,this,t),this._$AH=t}handleEvent(t){typeof this._$AH=="function"?this._$AH.call(this.options?.host??this.element,t):this._$AH.handleEvent(t)}},qe=class{constructor(t,o,r){this.element=t,this.type=6,this._$AN=void 0,this._$AM=o,this.options=r}get _$AU(){return this._$AM._$AU}_$AI(t){oe(this,t)}};var $o=Ke.litHtmlPolyfillSupport;$o?.(ue,fe),(Ke.litHtmlVersions??=[]).push("3.3.1");var et=(e,t,o)=>{let r=o?.renderBefore??t,n=r._$litPart$;if(n===void 0){let a=o?.renderBefore??null;r._$litPart$=n=new fe(t.insertBefore(de(),a),a,void 0,o??{})}return n._$AI(e),n};function me(e,t){let o=L(),r=T(e)??T(m.LBW),n=u=>{let c=u.target;$e({block:e,value:o.weightState.weights[e].value,checkState:c.checked})},a=u=>{let h=+u.target.value/100,p=o.weightState.weights[e].checkState;p!==void 0&&(p=!0),$e({block:e,value:h,checkState:p})},s=u=>{let h=+u.target.value,p=o.weightState.weights[e].checkState;p!==void 0&&(p=!0),$e({block:e,value:h,checkState:p})},l=t[e],i=u=>{let c=u.target,h=+c.value/100,w=c.parentElement.querySelector("input.value");w&&(w.value=h.toString())},g=u=>{let c=u.target,p=c.parentElement.querySelector("input.slider");if(!p)return;let w=+c.value;if(isNaN(w))return;let b=+p.min,f=+p.max,M=Math.round(w*100);M<b&&(b=M),M>f&&(f=M),b!==+p.min&&(p.min=String(b)),f!==+p.max&&(p.max=String(f)),p.value=String(M)};return k`
+        <span>
+            <label>${e}</label>
+            ${o.weightState.weights[e].checkState!==void 0?k`<input type="checkbox" .checked=${o.weightState.weights[e].checkState} @input=${n} />`:""}
+        </span>
+        <div class="f f-c g-4">
+            <input
+                class="slider"
+                type="range"
+                .min=${Math.round(l.sliderMin*100)}
+                .max=${Math.round(l.sliderMax*100)}
+                .step=${Math.round(r.step*100)}
+                .value=${Math.round(l.value*100)}
+                ?disabled=${I(o,e)}
+                @input=${i}
+                @change=${a}
+            />
+            <input
+                class="value"
+                type="number"
+                .value=${+l.value}
+                .step=${r.step}
+                ?disabled=${I(o,e)}
+                @input=${g}
+                @change=${s}
+            />
+        </div>
+    `}var tt=!1,ot={x:0,y:0};function lo(e,t){let o=L(),r=O(o.loraName),n=c=>{tt=!0;let h=document.getElementById("weight-helper");if(!h)return;let p=h.getBoundingClientRect();ot={x:c.clientX-p.left,y:c.clientY-p.top},document.addEventListener("mousemove",a),document.addEventListener("mouseup",s)},a=c=>{if(!tt)return;let h=c.clientX-ot.x+window.scrollX,p=c.clientY-ot.y+window.scrollY;Ht(p,h)},s=()=>{tt=!1,document.removeEventListener("mousemove",a),document.removeEventListener("mouseup",s)},l=()=>{Ft()},i=()=>{Gt()},g=()=>{let c=Math.max(0,o.historyIndex-1),h=r[c],p=Be(h.loraParams,h.selectedModelType,h.selectedLoraBlockType),w=U(o.loraName,h.loraParams),b=G(h.selectedModelType,h.selectedLoraBlockType,o.lbwPresets,p.weights);Ge({historyIndex:c,weightState:p,locked:w,preset:b})},u=()=>{let c=Math.min(r.length-1,o.historyIndex+1),h=r[c],p=Be(h.loraParams,h.selectedModelType,h.selectedLoraBlockType),w=U(o.loraName,h.loraParams),b=G(h.selectedModelType,h.selectedLoraBlockType,o.lbwPresets,p.weights);Ge({historyIndex:c,weightState:p,locked:w,preset:b})};return k`
+        <header @mousedown=${n}>
+            <span>
+                <span class="lock ${t?"like":"unlike"}" style="${D(!o.weightState.xyzMode)}" @click=${i}>
+                    <svg version="1.1" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24">
+                        <path
+                            d="M21,10H19V7A7,7,0,0,0,5,7v3H3a1,1,0,0,0-1,1v9a4,4,0,0,0,4,4H18a4,4,0,0,0,4-4V11A1,1,0,0,0,21,10Zm-9,9.5A2.5,2.5,0,1,1,14.5,17,2.5,2.5,0,0,1,12,19.5ZM15,10H9V7a3,3,0,0,1,6,0v3Z"
+                        ></path>
+                    </svg>
+                </span>
+                <label class="name">${e}</label>
+            </span>
+            <div class="history" style="${z(r.length>1)}">
+                <a class="icon" @click=${l}>clear</a>
+                <div class="page">
+                    <a class="icon" @click=${g} style="${D(o.historyIndex>0)}">&lt;</a>
+                    <label class="page-label">${o.historyIndex+1}/${r.length}</label>
+                    <a class="icon" @click=${u} style="${D(o.historyIndex<r.length-1)}">&gt;</a>
+                </div>
+            </div>
+        </header>
+    `}function co(){let e=L(),{blockGroups:t,weightState:o}=e,{selectedModelType:r,selectedLoraBlockType:n}=o,a=new Set(Q(r,n));return k`
+        <div class="lbw-column">
+            ${t.map(s=>k`
+                    <div class="f col g-2" style="display: flex">
+                        ${s.map(l=>k`
+                                <div class="border p f g-2 col">
+                                    ${xe(l).map(i=>a.has(i)&&o.weights[i]?k` <div class="f g-2">${me(i,o.weights)}</div> `:"")}
+                                </div>
+                            `)}
+                    </div>
+                `)}
+        </div>
+    `}function go(){let e=L(),t=a=>{let l=a.target.value,i=N(e.weightState.selectedModelType,l),g=$(e.weightState.selectedModelType,l),u=F(e.weightState.weights,i);Ve({selectedLoraBlockType:l,selectedModelType:e.weightState.selectedModelType,blockGroups:i,lbwPresets:g,weights:u})},o=a=>{let l=a.target.value,i=N(l,e.weightState.selectedLoraBlockType),g=$(l,e.weightState.selectedLoraBlockType),u=F(e.weightState.weights,i);Ve({selectedModelType:l,selectedLoraBlockType:e.weightState.selectedLoraBlockType,blockGroups:i,lbwPresets:g,weights:u})},r=()=>{Rt()},n=a=>{let s=a.target,l=s.value;if(l===""&&(l=e.preset,l===""))return;let i=$(e.weightState.selectedModelType,e.weightState.selectedLoraBlockType)[l].split(",").map(c=>+c),g={};Q(e.weightState.selectedModelType,e.weightState.selectedLoraBlockType).forEach((c,h)=>{g[c]={initValue:e.weightState.weights[c].initValue,value:i[h],sliderMin:Math.min(i[h],e.weightState.weights[c].sliderMin),sliderMax:Math.max(i[h],e.weightState.weights[c].sliderMax)}}),Vt({weights:g,preset:s.value})};return k`
+        <div class="f g-2 f-end">
+            <select .value="${e.weightState.selectedModelType}" style="flex-grow: 1" ?disabled="${I(e)}" @change="${o}">
+                ${Object.entries(K).map(([a,s])=>k` <option value="${s}">${a}</option> `)}
+            </select>
+            <select
+                .value="${e.weightState.selectedLoraBlockType}"
+                style="flex-grow: 1"
+                ?disabled="${I(e)}"
+                @change="${t}"
+            >
+                ${Object.entries(J).map(([a,s])=>k` <option value="${s}">${a}</option> `)}
+            </select>
+        </div>
+        <div class="f g-2 f-end">
+            <select .value="${e.preset}" style="flex-grow: 1" ?disabled="${I(e)}" @change="${n}">
+                <option value=""></option>
+                ${Object.keys(e.lbwPresets).map(a=>k` <option value="${a}">${a}</option> `)}
+            </select>
+            <div class="p f g-2 f-end">
+                <div class="f g-2 f-end">
+                    <input
+                        id="wh:xyz"
+                        value="XYZ"
+                        type="checkbox"
+                        .checked="${e.weightState.xyzMode}"
+                        ?disabled="${I(e)}"
+                        @change="${r}"
+                    />
+                    <label class="radio-label" for="wh:xyz">
+                        ${it(e.weightState.selectedModelType,e.weightState.selectedLoraBlockType)}
+                    </label>
+                </div>
+            </div>
+        </div>
+    `}function po(){let e=L(),t=()=>{if(e.uiState.isWaiting)return;let o=Pe();ce(!0),Ie(e.loraName,!0,o.signal).then(r=>{let n=e.weightState.selectedModelType;(!n||n===y.Unknown)&&(n=r.modelType??y.Unknown);let a=N(n,e.weightState.selectedLoraBlockType),s=F(e.weightState.weights,a);Ce({algorithm:r.algorithm||"Unknown",modelType:r.modelType||"Unknown",baseModel:r.baseModel||"Unknown",selectedModelType:n,usingBlocks:r.usingBlocks,blockGroups:a,lbwPresets:$(n,e.weightState.selectedLoraBlockType),weights:s})}).catch(r=>{r.name!=="AbortError"&&(console.error("Error fetching metadata:",r),ce(!1))})};return k`
+        <span class="metadata">
+            <span>
+                <span>Model</span>
+                <span>${e.metadataState?.modelType||"Unknown"}</span>
+            </span>
+            <span>
+                <span>Base</span>
+                <span>${e.metadataState?.baseModel||"Unknown"}</span>
+            </span>
+            <a class="icon mini svg" @click=${t}>
+                <svg version="1.1" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 492 492">
+                    <g>
+                        <path
+                            d="M484.08,296.216c-5.1-5.128-11.848-7.936-19.032-7.936H330.516c-14.828,0-26.86,12.036-26.86,26.868v22.796c0,7.168,2.784,14.064,7.884,19.16c5.092,5.088,11.82,8.052,18.976,8.052H366.1c-31.544,30.752-74.928,50.08-120.388,50.08c-71.832,0-136.028-45.596-159.744-113.344c-5.392-15.404-19.972-25.784-36.28-25.784c-4.316,0-8.592,0.708-12.7,2.144c-9.692,3.396-17.48,10.352-21.932,19.596c-4.456,9.248-5.04,19.684-1.648,29.368c34.496,98.54,127.692,164.74,232.144,164.74c64.132,0,123.448-23.948,169.572-67.656v25.22c0,14.836,12.384,27.108,27.224,27.108h22.792c14.84,0,26.86-12.272,26.86-27.108V315.24C492,308.056,489.2,301.304,484.08,296.216z"
+                        ></path>
+                    </g>
+                    <g>
+                        <path
+                            d="M478.628,164.78C444.132,66.244,350.916,0.044,246.464,0.044c-64.136,0-123.464,23.952-169.588,67.66v-25.22c0-14.832-12.344-27.112-27.184-27.112H26.896C12.06,15.372,0,27.652,0,42.484V176.76c0,7.18,2.824,13.868,7.944,18.964c5.096,5.128,11.86,7.932,19.044,7.932l-0.08,0.06h134.604c14.84,0,26.832-12.028,26.832-26.86v-22.8c0-14.836-11.992-27.216-26.832-27.216h-35.576c31.544-30.752,74.932-50.076,120.392-50.076c71.832,0,136.024,45.596,159.74,113.348c5.392,15.404,19.968,25.78,36.28,25.78c4.32,0,8.588-0.704,12.7-2.144c9.696-3.396,17.48-10.348,21.932-19.596C481.432,184.9,482.02,174.472,478.628,164.78z"
+                        ></path>
+                    </g>
+                </svg>
+            </a>
+        </span>
+    `}function ho(e){let t=i=>{extraNetworksRequestMetadata(i,"lora",e.modelName)},o=i=>{extraNetworksEditUserMetadata(i,se(),"lora",e.modelName)},r=()=>{window.open(`https://civitai.com/models/${e.modelId}`,"_blank")},n=()=>{ze(!0)},a=()=>{ze(!1)},s=()=>{let i=se(),g=document.querySelector(`#${i}_prompt textarea`),u=document.querySelector(`#${i}_neg_prompt textarea`),c=ae();if(c===u||c!==g&&(g=c,u=void 0),!opts.weight_helper_using_execCommand)(()=>{let p=[{triggerWords:e.triggerWords,textarea:g},{triggerWords:e.negativeTriggerWords,textarea:u}];for(let{triggerWords:w,textarea:b}of p)if(w.length>0&&b){let f=w.join(", ");b.value&&(f=", "+f),b.value+=f}})();else{let h=()=>{let p=[{triggerWords:e.triggerWords,textarea:g},{triggerWords:e.negativeTriggerWords,textarea:u}];for(let{triggerWords:w,textarea:b}of p)if(w.length>0&&b){let f=w.join(", ");b.value&&(f=", "+f),b.focus();let M=b.value.length;b.setSelectionRange(M,M),document.execCommand("insertText",!1,f)}};Te(()=>{h()})}jt()},l=["preview-pane","card"];switch(opts.weight_helper_preview_position){case"Bottom Right":l.push("preview-bottom-right");break;case"Top Left":l.push("preview-top-left");break;case"Bottom Left":l.push("preview-bottom-left");break;default:l.push("preview-top-right");break}return k`
+        <div class="${l.join(" ")}" data-name="${e.modelName}">
+            <img class="preview" src="${e.thumbUrl}" style="height: ${opts.weight_helper_preview_height}px" />
+            <div class="action-row button-top" style="${D(!e.isDescriptionVisible)}">
+                ${e.modelName&&e.hasMetadata&&typeof extraNetworksRequestMetadata=="function"?k` <div class="metadata-btn card-btn" title="Show internal metadata" @click="${t}"></div> `:""}
+                ${e.modelName&&typeof extraNetworksEditUserMetadata=="function"?k` <div class="edit-btn card-btn" title="Edit metadata" @click="${o}"></div> `:""}
+                ${e.modelId?k` <div class="civitai-btn card-btn" title="Open civitai" @click="${r}"></div> `:""}
+                ${e.triggerWords.length>0||e.negativeTriggerWords.length>0?k`
+                          <div
+                              class="add-trigger-btn card-btn"
+                              title="Add trigger words"
+                              @click="${s}"
+                              style="${z(e.isTagInsertButtonVisible)}"
+                          ></div>
+                      `:""}
+            </div>
+
+            ${e.description?k`
+                      <div class="action-row button-bottom" style="${D(!e.isDescriptionVisible)}">
+                          <div class="card-btn note-btn" @click="${n}"></div>
+                      </div>
+                      <textarea class="description" style="${D(e.isDescriptionVisible)}">
+                    ${e.description}
+                </textarea
+                      >
+                      <div
+                          class="card-btn description-close-btn"
+                          style="${D(e.isDescriptionVisible)}"
+                          @click="${a}"
+                      ></div>
+                  `:""}
+        </div>
+    `}var uo=!1;function fo(){let e=L(),{uiState:t,weightState:o,lock:r}=e;uo||(_o(),uo=!0);let n=`transform: scale(${t.scale}); ${z(t.isVisible)}`;return k`
+        <div id="weight-helper" class="${e.uiState.isWaiting?"waiting":""}" style="${n}">
+            ${lo(e.loraName,r)} ${po()}
+            ${o.weights[m.TENC]&&k`<section class="border p">${me(m.TENC,o.weights)}</section>`}
+            ${[m.UNET,m.START,m.STOP].filter(a=>!!o.weights[a]).map(a=>k`<section
+                            class="border p"
+                            style="${z(!t.isMoreButtonVisible||o.weights[a].initCheckState||o.weights[a].initValue!==T(a).default)}"
+                        >
+                            ${me(a,o.weights)}
+                        </section>`)}
+
+            <button style="${t.isMoreButtonVisible?"":"display: none"}" ?disabled="${I(e)}" @click="${()=>Xt()}">
+                show more options
+            </button>
+
+            <section class="border p">
+                <label>LBW</label>
+                <div class="f col g-4 w-fill">${go()} ${co()}</div>
+            </section>
+
+            ${e.previewState.modelName?ho(e.previewState):""}
+        </div>
+    `}function _o(){let e=()=>{let s=L();s.uiState.isVisible&&(Se(),le(s))},t=s=>{let l=L();if(l.uiState.isVisible){s.preventDefault(),Se(),le(l);return}let i=It(s);i&&(s.preventDefault(),Bo(i))},o=s=>{let l=L();if(l.uiState.isVisible){let i=document.getElementById("weight-helper");if(!i)return;let g=se();if(s){let u=s.target;if(i.contains(u)||u.id===`${g}_token_button`||u.id===`${g}_lora_edit_user_metadata_button`||u.className==="global-popup-close")return;if(u.id.indexOf("_interrupt")>0){Se();return}}Se(),le(l)}},r=s=>{if(s.key==="Escape"){let l=L();Ao(l)}};gradioApp().querySelectorAll("button:is([id*='_generate'])").forEach(s=>{s.addEventListener("click",e,!0)}),gradioApp().querySelectorAll("*:is([id*='_toprow'] [id*='_prompt'], .prompt) textarea").forEach(s=>{s.addEventListener("contextmenu",t)}),document.addEventListener("click",o),document.addEventListener("keyup",r)}var mo=!1;function So(e){et(fo(),e),mo||(Wt(()=>{et(fo(),e)}),Nt(t=>{let o=e.querySelector("#weight-helper");o&&(o.style.top=t.top+"px",o.style.left=t.left+"px")}),mo=!0)}function Bo(e){let{tabId:t,textarea:o,selectionStart:r,selectionEnd:n,namespace:a,loraName:s,loraParams:l,top:i,left:g}=e;Lt(t),Ct(o),$t(r),Me(n),Bt(o.value.substring(r,n)),At(t),Et(t),T(m.START).max=ke(),T(m.STOP).max=ke(),T(m.STOP).default=ke();let u=document.getElementById("weight-helper").getBoundingClientRect(),c=window.innerHeight-u.bottom,h=i;c<0&&(u.top<0?h=window.scrollY:h+=c);let p=Zt(a,s,l);Dt(p);let w=document.getElementById("weight-helper");if(w&&(w.style.top=h+"px",w.style.left=g+"px"),!p.metadataState){let b=Pe();Ie(s,!1,b.signal).then(f=>{let M=p.weightState.selectedModelType;(!M||M===y.Unknown)&&(M=f.modelType??y.Unknown);let C=p.weightState.selectedLoraBlockType,W=N(M,C),q=F(p.weightState.weights,W),Ue={...p.weightState,selectedModelType:M,selectedLoraBlockType:C},bo=_(p.loraName,Ue).loraParams;p.weightState.xyzMode||(ee(s,{loraParams:bo,selectedLoraBlockType:C,selectedModelType:M}),zt(O(s).length-1)),Ce({algorithm:f.algorithm||"Unknown",modelType:f.modelType||"Unknown",baseModel:f.baseModel||"Unknown",selectedModelType:M,usingBlocks:f.usingBlocks,blockGroups:W,lbwPresets:$(M,p.weightState.selectedLoraBlockType),weights:q})}).catch(f=>{f.name!=="AbortError"&&(console.error("Error fetching metadata:",f),ce(!1))})}if(!p.previewState.modelId){let b=qt();Kt(s,b.signal).then(async f=>{f.thumbUrl!==be&&(f.thumbUrl=f.thumbUrl),bt(s,f),Yt(f)}).catch(f=>{f.name!=="AbortError"&&console.error("Error fetching metadata:",f)})}}function Se(){let e=_e();Ot(e)}function Ao(e){opts.weight_helper_using_execCommand||le(e,!0),Se()}var Ne,Eo=new Promise(e=>{Ne=e});pt();onUiLoaded(()=>{if(!gradioApp().querySelector("#lbw_ratiospreset"))return;let o=[...getComputedStyle(document.documentElement).getPropertyValue("--body-text-color").trim().slice(1).match(/.{1,2}/g).map(r=>parseInt(r,16)),.3];document.documentElement.style.setProperty("--weight-helper-shadow",`rgba(${o.join(",")})`),document.documentElement.style.setProperty("--weight-helper-slider_size",String(opts.weight_helper_slider_length)),Eo.then(()=>{let r=document.createElement("div");r.id="weight-helper-container",gradioApp().appendChild(r),at(),Ut(_e()),So(r)})});onOptionsChanged(()=>{Ne&&(Ne(!0),Ne=null)});})();
+/*! Bundled license information:
+
+lit-html/lit-html.js:
+  (**
+   * @license
+   * Copyright 2017 Google LLC
+   * SPDX-License-Identifier: BSD-3-Clause
+   *)
+*/
+//# sourceMappingURL=weight_helper.js.map

sd-webui-weight-helper-main/package.json

@@ -0,0 +1,32 @@
+{
+  "name": "sd-webui-weight-helper",
+  "version": "1.3.3",
+  "description": "",
+  "scripts": {
+    "type-check": "tsc --noEmit",
+    "lint": "eslint src/**/*.ts",
+    "lint-fix": "eslint src/**/*.ts --fix",
+    "format": "prettier --write 'src/**/*.{ts,js,json,css,md}'",
+    "build": "node esbuild.config.mjs",
+    "build-min-map": "node esbuild.config.mjs --minify --sourcemap"
+  },
+  "keywords": [],
+  "author": "",
+  "devDependencies": {
+    "@eslint/js": "^9.34.0",
+    "@trivago/prettier-plugin-sort-imports": "^5.2.2",
+    "@typescript-eslint/eslint-plugin": "^8.41.0",
+    "@typescript-eslint/parser": "^8.41.0",
+    "esbuild": "^0.25.9",
+    "eslint": "^9.34.0",
+    "eslint-config-prettier": "^10.1.8",
+    "eslint-plugin-prettier": "^5.5.4",
+    "globals": "^16.3.0",
+    "prettier": "^3.6.2",
+    "typescript": "^5.9.2",
+    "typescript-eslint": "^8.41.0"
+  },
+  "dependencies": {
+    "lit-html": "^3.3.1"
+  }
+}

sd-webui-weight-helper-main/scripts/weight_helper.py

@@ -0,0 +1,91 @@
+import os
+import sys
+
+import gradio as gr
+from fastapi import FastAPI
+from modules import script_callbacks, shared
+
+_THIS_DIR = os.path.dirname(os.path.abspath(__file__))
+if _THIS_DIR not in sys.path:
+    sys.path.insert(0, _THIS_DIR)
+
+import wh_api as api
+import wh_const as const
+
+
+def on_app_started(_: gr.Blocks, app: FastAPI):
+    @app.post(const.API_PREFIX + "/get_metadata")
+    async def _(key: str, force: bool = False):
+        return api.get_metadata(key, force)
+
+    @app.post(const.API_PREFIX + "/get_preview_info")
+    async def _(key: str):
+        return api.get_preview_info(key)
+
+
+def on_ui_settings():
+    """
+    Set up the UI settings for the weight helper by adding options to the shared configuration.
+    """
+    section = ('weight_helper', 'Weight Helper')
+
+    shared.options_templates.update(shared.options_section(section, {
+        'weight_helper_enabled': shared.OptionInfo(True, 'Enabled'),
+        'weight_helper_context_menu_scale': shared.OptionInfo(0.8, "Context Menu Scale", gr.Number),
+        'weight_helper_using_execCommand': shared.OptionInfo(True, "Use the deprecated execCommand function to replace text.").info(
+            "You can use 'undo' to revert the text to its previous state, "
+            "but it will no longer be updated in real-time."),
+
+        "weight_helper_slider_length": shared.OptionInfo(160, "Slider Length (px)", gr.Number),
+
+        "weight_helper_TEnc_min": shared.OptionInfo(0, "TEnc Min Value", gr.Number),
+        "weight_helper_TEnc_max": shared.OptionInfo(1, "TEnc Max Value", gr.Number),
+        "weight_helper_TEnc_step": shared.OptionInfo(0.05, "TEnc Step", gr.Number),
+
+        "weight_helper_UNet_min": shared.OptionInfo(0, "UNet Min Value", gr.Number),
+        "weight_helper_UNet_max": shared.OptionInfo(1, "UNet Max Value", gr.Number),
+        "weight_helper_UNet_step": shared.OptionInfo(0.05, "UNet Step", gr.Number),
+
+        "weight_helper_LBW_min": shared.OptionInfo(0, "LBW (LoRA Block Weight) Min Value", gr.Number),
+        "weight_helper_LBW_max": shared.OptionInfo(1, "LBW (LoRA Block Weight) Max Value", gr.Number),
+        "weight_helper_LBW_step": shared.OptionInfo(0.05, "LBW (LoRA Block Weight) Step", gr.Number),
+
+        'weight_helper_show_preview': shared.OptionInfo(True, 'Show Preview'),
+        "weight_helper_preview_height": shared.OptionInfo(400, "Preview Height (px)", gr.Number),
+        "weight_helper_preview_position": shared.OptionInfo("Top Right", "Preview Position", gr.Radio, {
+            "choices": ["Top Right", "Bottom Right", "Top Left", "Bottom Left"]
+        }),
+
+        'weight_helper_parse_lora_blocks': shared.OptionInfo(True, '[experimental] Parse LoRA file to get block information'),
+
+        "weight_helper_LBW_SD_lora_block_points": shared.OptionInfo(
+            "BASE, IN01-IN04, IN05-IN08, M00, OUT03-OUT06, OUT07-OUT11",
+            "Advanced option - LoRA-LierLa block points")
+        .info("default: BASE, IN01-IN04, IN05-IN08, M00, OUT03-OUT06, OUT07-OUT11"),
+        "weight_helper_LBW_SD_lycoris_block_points": shared.OptionInfo(
+            "BASE, IN00-IN05, IN06-IN11, M00, OUT00-OUT05, OUT06-OUT11",
+            "Advanced option - LyCORIS or LoRA-C3Lier block points")
+        .info("default: BASE, IN00-IN05, IN06-IN11, M00, OUT00-OUT05, OUT06-OUT11"),
+
+        "weight_helper_LBW_SDXL_lora_block_points": shared.OptionInfo(
+            "BASE, IN04-IN08, M00, OUT00-OUT05",
+            "Advanced option - SDXL LoRA-LierLa block points")
+        .info("default: BASE, IN04-IN08, M00, OUT00-OUT05"),
+        "weight_helper_LBW_SDXL_lycoris_block_points": shared.OptionInfo(
+            "BASE, IN00-IN03, IN04-IN08, M00, OUT00-OUT03, OUT04-OUT08",
+            "Advanced option - SDXL LyCORIS or LoRA-C3Lier block points")
+        .info("default: BASE, IN00-IN03, IN04-IN08, M00, OUT00-OUT03, OUT04-OUT08"),
+
+        "weight_helper_LBW_Flux_lora_block_points": shared.OptionInfo(
+            "CLIP, T5, IN, OUT, D00-D18, S00-S18, S19-S37",
+            "Advanced option - Flux LoRA-LierLa block points")
+        .info("default: CLIP, T5, IN, OUT, D00-D18, S00-S18, S19-S37"),
+        "weight_helper_LBW_Flux_lycoris_block_points": shared.OptionInfo(
+            "CLIP, T5, IN, OUT, D00-D18, S00-S18, S19-S37",
+            "Advanced option - Flux LyCORIS or LoRA-C3Lier block points")
+        .info("default: CLIP, T5, IN, OUT, D00-D18, S00-S18, S19-S37"),
+    }))
+
+
+script_callbacks.on_ui_settings(on_ui_settings)
+script_callbacks.on_app_started(on_app_started)

sd-webui-weight-helper-main/scripts/wh_analyzer.py

@@ -0,0 +1,196 @@
+import os
+import pickle
+import re
+import sys
+from typing import Optional
+
+_THIS_DIR = os.path.dirname(os.path.abspath(__file__))
+if _THIS_DIR not in sys.path:
+    sys.path.insert(0, _THIS_DIR)
+
+import wh_const as const
+import wh_env as env
+
+
+def get_model_type(lora) -> Optional[str]:
+    if env.ui_type == const.UI_TYPE.A1111:
+        sd_version: Optional[str] = lora.sd_version.name
+        if sd_version and env.network:
+            if sd_version == env.network.SdVersion.SD1.name or sd_version == env.network.SdVersion.SD2.name:
+                return "SD"
+            elif sd_version == env.network.SdVersion.SDXL.name:
+                return "SDXL"
+            return sd_version
+    elif env.ui_type == const.UI_TYPE.NEW_FORGE:
+        base_model_version: Optional[str] = lora.metadata.get("ss_base_model_version")
+        if base_model_version:
+            base_model_version = base_model_version.lower()
+            if "sd_" in base_model_version:
+                return "SD"
+            if "sd3" in base_model_version:
+                return "SD"
+            elif "sdxl" in base_model_version:
+                return "SDXL"
+            elif "flux" in base_model_version:
+                return "Flux"
+            return None
+    return None
+
+
+def get_algorithm(metadata, ss_network_args) -> Optional[str]:
+    ss_network_module = metadata.get("ss_network_module")
+    if not ss_network_module or ss_network_module == "Unknown":
+        return None
+
+    conv_dim = float(ss_network_args.get("conv_dim", "-1"))
+    conv_alpha = float(ss_network_args.get("conv_alpha", "-1"))
+    algo = ss_network_args.get("algo", "").lower()
+    unit = ss_network_args.get("unit", "").lower()
+    dora_wd = ss_network_args.get("dora_wd", False)
+
+    if ss_network_module.find("locon.locon_kohya") >= 0:
+        return "LoCon"
+
+    elif ss_network_module.find("lycoris.kohya") >= 0:
+        algoName = ""
+        if dora_wd:
+            algoName = "(DoRA)"
+        elif algo == "lora":
+            algoName += "(LoCon)"
+        elif algo == "locon":
+            if unit:
+                algoName += "(DyLoRA)"
+            else:
+                algoName += "(LoCon)"
+        elif algo == "loha":
+            algoName += "(LoHa)"
+        elif algo == "lokr":
+            algoName += "(Lokr)"
+        elif algo == "ia3":
+            algoName += "(IA3)"
+        elif algo == "full":
+            algoName += "(Full)"
+        elif algo == "glora":
+            algoName += "(GLoRA)"
+
+        return f"LyCORIS{algoName}"
+
+    elif ss_network_module.find("networks.dylora") >= 0:
+        if algo == "dylora" and unit:
+            if conv_dim > 0 or conv_alpha > 0:
+                return "DyLoRA(C3Lier)"
+        return "DyLoRA(LierLa)"
+
+    elif conv_dim > 0 or conv_alpha > 0:
+        return "LoRA(C3Lier)"
+
+    else:
+        return "LoRA(LierLa)"
+
+
+def get_blocks(lora_on_disk, force):
+    baseKeys = __get_lora_state_keys(lora_on_disk.filename, force)
+
+    def formatForDownMidUp(key):
+        blocks = ["BASE"]
+        for key in keys:
+            k, v = key.split("_", 1)
+            if k == "down":
+                k = "IN"
+                v1, v2 = v.split("_")
+                val = int(v1) * 3 + int(v2) + 1
+            elif k == "mid":
+                blocks.append("M00")
+                continue
+            elif k == "up":
+                k = "OUT"
+                v1, v2 = v.split("_")
+                val = int(v1) * 3 + int(v2)
+            blocks.append(f'{k}{int(val):02}')
+        return sorted(blocks)
+
+    def formatForInputMiddleOutput(key):
+        blocks = ["BASE"]
+        for key in keys:
+            k, v = key.split("_")
+            if k == "input":
+                k = "IN"
+            elif k == "middle":
+                blocks.append("M00")
+                continue
+            elif k == "output":
+                k = "OUT"
+            blocks.append(f'{k}{int(v):02}')
+        return sorted(blocks)
+
+    def formatForFlux(key):
+        blocks = ["CLIP", "T5", "IN", "OUT"]
+        for key in keys:
+            k, v = key.split("_")
+            v = int(v)
+            if k == "double":
+                k = "D"
+            if k == "single":
+                k = "S"
+            blocks.append(f'{k}{v:02}')
+        return sorted(blocks)
+
+    keys = [k for k in baseKeys if k.find("lora_unet") == 0]
+    if len(keys) > 0:
+        keys_tmp = [k for k in keys if re.match("^lora_unet_(down|mid|up)_", k)]
+        if len(keys_tmp) > 0:
+            keys = set(["_".join([k.split("_")[2], k.split("_")[4] + "_" + k.split("_")[6]]) for k in keys_tmp if "transformer_blocks" in k])
+            return formatForDownMidUp(keys)
+
+        keys_tmp = [k for k in keys if re.match("^lora_unet_(input|middle|output)_", k)]
+        if len(keys_tmp) > 0:
+            keys = set(["_".join([k.split("_")[2], k.split("_")[4]]) for k in keys_tmp if "transformer_blocks" in k])
+            return formatForInputMiddleOutput(keys)
+
+        keys_tmp = [k for k in keys if re.match("^lora_unet_(single|double)_", k)]
+        if len(keys_tmp) > 0:
+            keys = set(["_".join([k.split("_")[2], k.split("_")[4]]) for k in keys_tmp])
+            return formatForFlux(keys)
+
+        keys_tmp = [k for k in keys if re.match("^lora_unet_joint_blocks_", k)]
+        if len(keys_tmp) > 0:
+            # TODO SD3? 38 blocks
+            return None
+
+    keys = [k for k in baseKeys if "transformer_blocks" in k and "attentions" in k]
+    if len(keys) > 0:
+        # TODO unknown
+        keys = set([".".join(k.split(".")[2:6]) for k in keys])
+        in_blocks = set([f"{k.split('.')[1]}_{k.split('.')[3]}" for k in keys if "down_block" in k])
+        mid_block = set([k.split(".")[2] for k in keys if "mid_block" in k])
+        out_blocks = set([f"{k.split('.')[1]}_{k.split('.')[3]}" for k in keys if "up_block" in k])
+        keys = list(in_blocks) + list(mid_block) + list(out_blocks)
+        return None
+
+    keys = [k for k in baseKeys if "transformer_blocks" in k and k.find("transformer") == 0]
+    if len(keys) > 0:
+        keys = set([".".join(k.split(".")[1:3]) for k in keys])
+        keys = [k.replace("single_transformer_blocks.", "single_") for k in keys]
+        keys = [k.replace("transformer_blocks.", "double_") for k in keys]
+        return formatForFlux(keys)
+
+    return None
+
+
+def __get_lora_state_keys(filename, force):
+    cache_path = f"{os.path.splitext(filename)[0]}.whcache"
+    if not force and os.path.exists(cache_path):
+        try:
+            with open(cache_path, "rb") as f:
+                return pickle.load(f)
+        except Exception:
+            pass
+
+    lora_sd = env.network_lora.load_lora_state_dict(filename)
+    keys = list(lora_sd.keys())
+    try:
+        with open(cache_path, "wb") as f:
+            pickle.dump(keys, f)
+    except Exception:
+        pass
+    return keys

sd-webui-weight-helper-main/scripts/wh_api.py

@@ -0,0 +1,154 @@
+
+import ast
+import json
+import os
+import sys
+from typing import Any, Optional
+
+from modules import shared
+
+_THIS_DIR = os.path.dirname(os.path.abspath(__file__))
+if _THIS_DIR not in sys.path:
+    sys.path.insert(0, _THIS_DIR)
+
+import wh_analyzer as analyzer
+import wh_const as const
+import wh_utils as utils
+
+
+def get_metadata(key, force) -> dict[str, Optional[str]]:
+    ret: dict[str, Optional[str]] = {
+        "modelType": None,
+        "algorithm": None,
+        "baseModel": None,
+        "usingBlocks": None
+    }
+
+    lora_on_disk = utils.get_lora_on_disk(key)
+    if not lora_on_disk:
+        return ret
+
+    ret["modelType"] = analyzer.get_model_type(lora_on_disk)
+
+    metadata = lora_on_disk.metadata
+    ss_network_args = __get_network_args(lora_on_disk)
+    ret["algorithm"] = analyzer.get_algorithm(metadata, ss_network_args)
+
+    model_path, _ = os.path.splitext(lora_on_disk.filename)
+    civitai_info = __find_civitai_info(model_path)
+    ret["baseModel"] = civitai_info.get("baseModel", None)
+
+    if shared.opts.weight_helper_parse_lora_blocks:
+        try:
+            ret["usingBlocks"] = analyzer.get_blocks(lora_on_disk, force)
+        except Exception:
+            pass
+    return ret
+
+
+def get_preview_info(key):
+    ret: dict[str, Optional[bool | str | list[str]]] = {
+        "modelId": None,
+        "triggerWords": [],
+        "negativeTriggerWords": [],
+        "modelName": None,
+        "thumbUrl": None,
+        "hasMetadata": False,
+        "description": None,
+    }
+
+    lora_on_disk = utils.get_lora_on_disk(key)
+    if lora_on_disk:
+        model_path, _ = os.path.splitext(lora_on_disk.filename)
+        ret["modelName"] = lora_on_disk.name
+        ret["thumbUrl"] = __find_preview(model_path)
+        ret["hasMetadata"] = __find_metadata(lora_on_disk)
+        ret["description"] = __find_description(model_path)
+
+        trigger_words = []
+        civitai_info = __find_civitai_info(model_path)
+        ret["modelId"] = civitai_info.get("modelId", None)
+        trigger_words = civitai_info.get("trainedWords", [])
+        trigger_words = ",".join(trigger_words).split(",")
+        ret["triggerWords"] = [w.strip() for w in trigger_words if w]
+
+        lora_json = __get_lora_json(model_path)
+
+        activation_text = lora_json.get("activation text", "")
+        if activation_text:
+            activation_text = activation_text.split(",")
+            activation_text = [w.strip() for w in activation_text if w]
+            ret["triggerWords"] = activation_text
+
+        negative_text = lora_json.get("negative text", "")
+        if negative_text:
+            negative_text = negative_text.split(",")
+            negative_text = [w.strip() for w in negative_text if w]
+            ret["negativeTriggerWords"] = negative_text
+
+        json_description = lora_json.get("description", None)
+        if json_description:
+            ret["description"] = json_description
+
+    return ret
+
+
+def __get_network_args(lora) -> dict[str, Any]:
+    metadata = lora.metadata
+    ss_network_args = metadata.get("ss_network_args", {})
+    if type(ss_network_args) is str:
+        try:
+            ss_network_args = ast.literal_eval(ss_network_args)
+        except Exception:
+            ss_network_args = {}
+    return ss_network_args
+
+
+def __get_lora_json(path) -> dict[str, Any]:
+    if path:
+        lora_json = f"{path}.json"
+        if os.path.exists(lora_json):
+            try:
+                with open(lora_json, "r", encoding="utf-8", errors="replace") as f:
+                    return json.load(f)
+            except OSError:
+                pass
+    return {}
+
+
+def __find_preview(path) -> str:
+    if path:
+        potential_files = sum([[f"{path}.{ext}", f"{path}.preview.{ext}"] for ext in const.ALLOWED_PREVIEW_EXTENSIONS], [])
+        for file in potential_files:
+            if os.path.exists(file):
+                dpi_scale = 1.5
+                return utils.resize_to_base64(file, int(shared.opts.weight_helper_preview_height * dpi_scale))
+    return "./file=html/card-no-preview.png"
+
+
+def __find_metadata(lora) -> bool:
+    return len(lora.metadata) > 0 if lora else False
+
+
+def __find_description(path) -> Optional[str]:
+    if path:
+        for file in [f"{path}.txt", f"{path}.description.txt"]:
+            if os.path.exists(file):
+                try:
+                    with open(file, "r", encoding="utf-8", errors="replace") as f:
+                        return f.read()
+                except OSError:
+                    pass
+    return None
+
+
+def __find_civitai_info(path) -> dict[str, Any]:
+    if path:
+        civitai_info = f"{path}.civitai.info"
+        if os.path.exists(civitai_info):
+            try:
+                with open(civitai_info, "r", encoding="utf-8", errors="replace") as f:
+                    return json.load(f)
+            except OSError:
+                pass
+    return {}

sd-webui-weight-helper-main/scripts/wh_const.py

@@ -0,0 +1,11 @@
+import enum
+
+
+API_PREFIX = "/whapi/v1"
+ALLOWED_PREVIEW_EXTENSIONS = ["png", "jpg", "jpeg", "webp", "gif"]
+
+
+class UI_TYPE(enum.Enum):
+    Unknown = 1
+    A1111 = 2
+    NEW_FORGE = 3

sd-webui-weight-helper-main/scripts/wh_env.py

@@ -0,0 +1,36 @@
+
+import importlib
+import os
+import sys
+
+_THIS_DIR = os.path.dirname(os.path.abspath(__file__))
+if _THIS_DIR not in sys.path:
+    sys.path.insert(0, _THIS_DIR)
+
+import wh_const as const
+
+network = None
+network_lora = None
+ui_type = None
+
+try:
+    from modules_forge import forge_version  # noqa: F401
+    ui_type = const.UI_TYPE.NEW_FORGE
+except ImportError:
+    ui_type = const.UI_TYPE.A1111
+
+try:
+    network = importlib.import_module("extensions-builtin.sd_forge_lora.network")
+    network_lora = importlib.import_module("extensions-builtin.sd_forge_lora.ui_extra_networks_lora").networks
+except Exception:
+    pass
+
+if network_lora is None:
+    try:
+        network = importlib.import_module("extensions-builtin.Lora.network")
+        network_lora = importlib.import_module("extensions-builtin.Lora.ui_extra_networks_lora").networks
+    except Exception:
+        pass
+
+if network_lora is None:
+    ui_type = const.UI_TYPE.Unknown

sd-webui-weight-helper-main/scripts/wh_utils.py

@@ -0,0 +1,65 @@
+import base64
+import os
+import sys
+import urllib
+
+import cv2
+import numpy as np
+
+_THIS_DIR = os.path.dirname(os.path.abspath(__file__))
+if _THIS_DIR not in sys.path:
+    sys.path.insert(0, _THIS_DIR)
+
+import wh_env as env
+
+
+def get_lora_on_disk(key):
+    if env.network_lora:
+        lora_on_disk = env.network_lora.available_network_aliases.get(key)
+        if not lora_on_disk:
+            # sdnext
+            lora_on_disk = next((v for v in env.network_lora.available_network_aliases.values() if v.alias == key), None)
+        return lora_on_disk
+    return None
+
+
+def imread_unicode(path):
+    try:
+        with open(path, 'rb') as f:
+            data = np.frombuffer(f.read(), np.uint8)
+        img = cv2.imdecode(data, cv2.IMREAD_UNCHANGED)
+        return img
+    except Exception as e:
+        print(f"imread_unicode error: {e}")
+        return None
+
+
+def resize_to_base64(input_path, target_height):
+    output_format = '.webp'
+    quality = 85
+
+    img = imread_unicode(input_path)
+
+    h, w = img.shape[:2]
+    if h < target_height:
+        quoted_filename = urllib.parse.quote(input_path.replace('\\', '/'))
+        return f"./sd_extra_networks/thumb?filename={quoted_filename}"
+
+    scale = target_height / h
+    target_size = (int(w * scale), target_height)
+
+    resized = cv2.resize(img, target_size, interpolation=cv2.INTER_AREA)
+
+    sharpen_kernel = np.array([
+        [0, -0.2, 0],
+        [-0.2, 1.8, -0.2],
+        [0, -0.2, 0]
+    ])
+    sharpened = cv2.filter2D(resized, -1, sharpen_kernel)
+    success, buffer = cv2.imencode(output_format, sharpened, [cv2.IMWRITE_WEBP_QUALITY, quality])
+    if not success:
+        quoted_filename = urllib.parse.quote(input_path.replace('\\', '/'))
+        return f"./sd_extra_networks/thumb?filename={quoted_filename}"
+
+    base64_str = base64.b64encode(buffer).decode('utf-8')
+    return f'data:image/webp;base64,{base64_str}'

sd-webui-weight-helper-main/src/components/sections/common-section.ts

@@ -0,0 +1,139 @@
+import * as configManager from '@/shared/manager/config-manager';
+import * as globalState from '@/shared/state/global-weight-helper-state';
+import { WeightControllerTypes } from '@/shared/types/lora-types';
+import { disabled } from '@/shared/utils/helper-utils';
+
+import { TemplateResult, html } from 'lit-html';
+
+/**
+ * WeightController component provides UI controls for adjusting weight values.
+ * Displays a label, optional checkbox, slider, and number input for a weight parameter.
+ * @param label - The identifier for the weight block.
+ * @param values - The record of all weight values.
+ * @returns The WeightController template.
+ */
+export function createWeightController(label: string, values: Record<string, globalState.WeightControlState>): TemplateResult {
+    const state = globalState.getGlobalState();
+    const config = configManager.getWeightControllerConfig(label) ?? configManager.getWeightControllerConfig(WeightControllerTypes.LBW);
+
+    /**
+     * Handles the change event for the weight checkbox.
+     * Updates the check state for the weight block.
+     * @param e - The input change event.
+     */
+    const handleChangeWeightCheck = (e: Event) => {
+        const target = e.target as HTMLInputElement;
+        globalState.setWeight({
+            block: label,
+            value: state.weightState.weights[label].value,
+            checkState: target.checked,
+        });
+    };
+
+    /**
+     * Handles the change event for the weight slider.
+     * Updates the value and check state for the weight block.
+     * @param e - The input change event.
+     */
+    const handleChangeWeightSlider = (e: Event) => {
+        const target = e.target as HTMLInputElement;
+        const value = +target.value / 100;
+        let checkState = state.weightState.weights[label].checkState;
+        if (checkState !== undefined) {
+            checkState = true;
+        }
+        globalState.setWeight({
+            block: label,
+            value: value,
+            checkState: checkState,
+        });
+    };
+
+    /**
+     * Handles the change event for the weight number input.
+     * Updates the value and check state for the weight block.
+     * @param e - The input change event.
+     */
+    const handleChangeWeightNumber = (e: Event) => {
+        const target = e.target as HTMLInputElement;
+        const value = +target.value;
+        let checkState = state.weightState.weights[label].checkState;
+        if (checkState !== undefined) {
+            checkState = true;
+        }
+        globalState.setWeight({
+            block: label,
+            value: value,
+            checkState: checkState,
+        });
+    };
+
+    const weightController = values[label];
+
+    /**
+     * Live-sync handler for slider @input (does not commit state yet).
+     * Mirrors the slider value (scaled back to original 0-? range) into the numeric input
+     * so the user sees real-time feedback while dragging before the @change event fires.
+     */
+    const immediateSliderInput = (e: Event) => {
+        const target = e.target as HTMLInputElement;
+        const raw = +target.value / 100;
+        const wrapper = target.parentElement as HTMLElement;
+        const numberInput = wrapper.querySelector('input.value') as HTMLInputElement | null;
+        if (numberInput) numberInput.value = raw.toString();
+    };
+    /**
+     * Live-sync handler for number @input.
+     * Expands the slider range first when user enters a value outside current min/max
+     * (matching the logic used in setWeight) and then updates the slider thumb position.
+     * State commit is deferred to the @change handler.
+     */
+    const immediateNumberInput = (e: Event) => {
+        const target = e.target as HTMLInputElement;
+        const wrapper = target.parentElement as HTMLElement;
+        const slider = wrapper.querySelector('input.slider') as HTMLInputElement | null;
+        if (!slider) return;
+        const raw = +target.value;
+        if (isNaN(raw)) return; // Unconfirmed / empty input etc.
+        let sliderMin = +slider.min; // Scaled (0-100x) current min
+        let sliderMax = +slider.max;
+        const scaled = Math.round(raw * 100);
+        // Expand range first if the input value is outside (same policy as setWeight)
+        if (scaled < sliderMin) sliderMin = scaled;
+        if (scaled > sliderMax) sliderMax = scaled;
+        if (sliderMin !== +slider.min) slider.min = String(sliderMin);
+        if (sliderMax !== +slider.max) slider.max = String(sliderMax);
+        // Finally assign the value (no clamp needed after expansion)
+        slider.value = String(scaled);
+    };
+    return html`
+        <span>
+            <label>${label}</label>
+            ${state.weightState.weights[label].checkState !== undefined
+                ? html`<input type="checkbox" .checked=${state.weightState.weights[label].checkState} @input=${handleChangeWeightCheck} />`
+                : ''}
+        </span>
+        <div class="f f-c g-4">
+            <input
+                class="slider"
+                type="range"
+                .min=${Math.round(weightController.sliderMin * 100)}
+                .max=${Math.round(weightController.sliderMax * 100)}
+                .step=${Math.round(config.step * 100)}
+                .value=${Math.round(weightController.value * 100)}
+                ?disabled=${disabled(state, label)}
+                @input=${immediateSliderInput}
+                @change=${handleChangeWeightSlider}
+            />
+            <input
+                class="value"
+                type="number"
+                .value=${+weightController.value}
+                .step=${config.step}
+                ?disabled=${disabled(state, label)}
+                @input=${immediateNumberInput}
+                @change=${handleChangeWeightNumber}
+            />
+        </div>
+    `;
+}

sd-webui-weight-helper-main/src/components/sections/header-section.ts

@@ -0,0 +1,129 @@
+import * as historyManager from '@/shared/manager/history-manager';
+import * as globalState from '@/shared/state/global-weight-helper-state';
+import { getPreset } from '@/shared/utils/common-utils';
+import { getDisplayStyle, getVisibilityStyle } from '@/shared/utils/helper-utils';
+import { createLoraParamsState } from '@/shared/utils/state-utils';
+
+import { TemplateResult, html } from 'lit-html';
+
+// ドラッグ状態を管理するグローバル変数
+let isDragging = false;
+let offset = { x: 0, y: 0 };
+
+/**
+ * HeaderSection component displays the title bar of the Weight Helper.
+ * Provides LoRA name display, lock toggle, and history navigation controls.
+ * @param loraName - The name of the LoRA model.
+ * @param lock - The current lock status.
+ * @returns The HeaderSection template.
+ */
+export function createHeaderSection(loraName: string, lock: boolean): TemplateResult {
+    const state = globalState.getGlobalState();
+    const histories = historyManager.getHistories(state.loraName);
+
+    /**
+     * Handles the mouse down event for dragging the header.
+     * @param e - The mouse event.
+     */
+    const handleMouseDown = (e: MouseEvent) => {
+        isDragging = true;
+        const el = document.getElementById('weight-helper');
+        if (!el) return;
+        const rect = el.getBoundingClientRect();
+        offset = {
+            x: e.clientX - rect.left,
+            y: e.clientY - rect.top,
+        };
+        document.addEventListener('mousemove', handleMouseMove);
+        document.addEventListener('mouseup', handleMouseUp);
+    };
+
+    /**
+     * Handles the mouse move event for dragging the header.
+     * Updates the position of the header based on mouse movement.
+     * @param e - The mouse event.
+     */
+    const handleMouseMove = (e: MouseEvent) => {
+        if (!isDragging) return;
+
+        const x = e.clientX - offset.x + window.scrollX;
+        const y = e.clientY - offset.y + window.scrollY;
+
+        globalState.setPosition(y, x);
+    };
+
+    /**
+     * Handles the mouse up event to stop dragging the header.
+     * Removes the event listeners for mouse move and mouse up.
+     */
+    const handleMouseUp = () => {
+        isDragging = false;
+        document.removeEventListener('mousemove', handleMouseMove);
+        document.removeEventListener('mouseup', handleMouseUp);
+    };
+
+    /**
+     * Handles the clear history button click event.
+     * Clears the history of the Weight Helper.
+     */
+    const handleClearHistory = () => {
+        globalState.clearHistory();
+    };
+
+    /**
+     * Handles the lock toggle button click event.
+     * Toggles the lock status for the current LoRA and loraParams.
+     */
+    const handleToggleLockStatus = () => {
+        globalState.toggleLockStatus();
+    };
+
+    /**
+     * Handles the previous history button click event.
+     * Navigates to the previous history state.
+     */
+    const handlePrevHistory = () => {
+        const newIndex = Math.max(0, state.historyIndex - 1);
+        const history = histories[newIndex];
+        const loraParams = createLoraParamsState(history.loraParams, history.selectedModelType, history.selectedLoraBlockType);
+        const locked = historyManager.isLocked(state.loraName, history.loraParams);
+        const preset = getPreset(history.selectedModelType, history.selectedLoraBlockType, state.lbwPresets, loraParams.weights);
+        globalState.setHistory({ historyIndex: newIndex, weightState: loraParams, locked: locked, preset: preset });
+    };
+
+    /**
+     * Handles the next history button click event.
+     * Navigates to the next history state.
+     */
+    const handleNextHistory = () => {
+        const newIndex = Math.min(histories.length - 1, state.historyIndex + 1);
+        const history = histories[newIndex];
+        const loraParams = createLoraParamsState(history.loraParams, history.selectedModelType, history.selectedLoraBlockType);
+        const locked = historyManager.isLocked(state.loraName, history.loraParams);
+        const preset = getPreset(history.selectedModelType, history.selectedLoraBlockType, state.lbwPresets, loraParams.weights);
+        globalState.setHistory({ historyIndex: newIndex, weightState: loraParams, locked: locked, preset: preset });
+    };
+
+    return html`
+        <header @mousedown=${handleMouseDown}>
+            <span>
+                <span class="lock ${lock ? 'like' : 'unlike'}" style="${getVisibilityStyle(!state.weightState.xyzMode)}" @click=${handleToggleLockStatus}>
+                    <svg version="1.1" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24">
+                        <path
+                            d="M21,10H19V7A7,7,0,0,0,5,7v3H3a1,1,0,0,0-1,1v9a4,4,0,0,0,4,4H18a4,4,0,0,0,4-4V11A1,1,0,0,0,21,10Zm-9,9.5A2.5,2.5,0,1,1,14.5,17,2.5,2.5,0,0,1,12,19.5ZM15,10H9V7a3,3,0,0,1,6,0v3Z"
+                        ></path>
+                    </svg>
+                </span>
+                <label class="name">${loraName}</label>
+            </span>
+            <div class="history" style="${getDisplayStyle(histories.length > 1)}">
+                <a class="icon" @click=${handleClearHistory}>clear</a>
+                <div class="page">
+                    <a class="icon" @click=${handlePrevHistory} style="${getVisibilityStyle(state.historyIndex > 0)}">&lt;</a>
+                    <label class="page-label">${state.historyIndex + 1}/${histories.length}</label>
+                    <a class="icon" @click=${handleNextHistory} style="${getVisibilityStyle(state.historyIndex < histories.length - 1)}">&gt;</a>
+                </div>
+            </div>
+        </header>
+    `;
+}

sd-webui-weight-helper-main/src/components/sections/lbw-section.ts

@@ -0,0 +1,42 @@
+import * as configManager from '@/shared/manager/config-manager';
+import * as globalState from '@/shared/state/global-weight-helper-state';
+import { expandRange } from '@/shared/utils/helper-utils';
+
+import { createWeightController } from '@/components/sections/common-section';
+
+import { TemplateResult, html } from 'lit-html';
+
+/**
+ * LbwSection creates weight controllers for LoRA block weights.
+ * Shows groups of blocks based on the current model and block type configuration.
+ * @returns The LbwSection template.
+ */
+export function createLbwSection(): TemplateResult {
+    const state = globalState.getGlobalState();
+    const { blockGroups, weightState } = state;
+    const { selectedModelType, selectedLoraBlockType } = weightState;
+
+    const maskedBlockSet = new Set(configManager.getMaskedLbwBlocks(selectedModelType, selectedLoraBlockType));
+
+    return html`
+        <div class="lbw-column">
+            ${blockGroups.map(
+                (cols) => html`
+                    <div class="f col g-2" style="display: flex">
+                        ${cols.map(
+                            (group) => html`
+                                <div class="border p f g-2 col">
+                                    ${expandRange(group).map((blockLabel) =>
+                                        maskedBlockSet.has(blockLabel) && weightState.weights[blockLabel]
+                                            ? html` <div class="f g-2">${createWeightController(blockLabel, weightState.weights)}</div> `
+                                            : '',
+                                    )}
+                                </div>
+                            `,
+                        )}
+                    </div>
+                `,
+            )}
+        </div>
+    `;
+}

sd-webui-weight-helper-main/src/components/sections/lora-settings-section.ts

@@ -0,0 +1,143 @@
+import { SELECTABLE_LORA_BLOCK_TYPES, SELECTABLE_MODEL_TYPES } from '@/shared/constants/common-const';
+import * as configManager from '@/shared/manager/config-manager';
+import * as currentState from '@/shared/state/current-state';
+import * as globalState from '@/shared/state/global-weight-helper-state';
+import { LoraBlockTypes, ModelTypes } from '@/shared/types/lora-types';
+import { disabled } from '@/shared/utils/helper-utils';
+import { mergeWeightsWithGroups } from '@/shared/utils/weight-utils';
+
+import { TemplateResult, html } from 'lit-html';
+
+/**
+ * LoraSettingsSection creates controls for model settings.
+ * Allows selection of model type, LoRA block type, presets, and XYZ mode.
+ * @returns The LoraSettingsSection template.
+ */
+export function createLoraSettingsSection(): TemplateResult {
+    const state = globalState.getGlobalState();
+
+    /**
+     * Handles the change event for the LoRA block type select box.
+     * Updates the block groups, presets, and weights for the selected block type.
+     * @param e - The select change event.
+     */
+    const handleChangeLoraBlockType = (e: Event) => {
+        const target = e.target as HTMLSelectElement;
+        const selectedLoraBlockType = target.value as LoraBlockTypes;
+
+        const blockGroups = configManager.getLbwBlockGroups(state.weightState.selectedModelType, selectedLoraBlockType);
+        const lbwPresets = currentState.getLbwPresets(state.weightState.selectedModelType, selectedLoraBlockType);
+        const weights = mergeWeightsWithGroups(state.weightState.weights, blockGroups);
+
+        globalState.setModelInfo({
+            selectedLoraBlockType: selectedLoraBlockType,
+            selectedModelType: state.weightState.selectedModelType,
+            blockGroups: blockGroups,
+            lbwPresets: lbwPresets,
+            weights: weights,
+        });
+    };
+
+    /**
+     * Handles the change event for the model type select box.
+     * Updates the block groups, presets, and weights for the selected model type.
+     * @param e - The select change event.
+     */
+    const handleChangeModelType = (e: Event) => {
+        const target = e.target as HTMLSelectElement;
+        const selectedModelType = target.value as ModelTypes;
+
+        const blockGroups = configManager.getLbwBlockGroups(selectedModelType, state.weightState.selectedLoraBlockType);
+        const lbwPresets = currentState.getLbwPresets(selectedModelType, state.weightState.selectedLoraBlockType);
+        const weights = mergeWeightsWithGroups(state.weightState.weights, blockGroups);
+
+        globalState.setModelInfo({
+            selectedModelType: selectedModelType,
+            selectedLoraBlockType: state.weightState.selectedLoraBlockType,
+            blockGroups: blockGroups,
+            lbwPresets: lbwPresets,
+            weights: weights,
+        });
+    };
+
+    /**
+     * Handles the toggle event for the XYZ mode checkbox.
+     * Dispatches an action to toggle the XYZ mode state.
+     */
+    const handleToggleXyzMode = () => {
+        globalState.toggleXyzMode();
+    };
+
+    /**
+     * Handles the change event for the preset select box.
+     * Applies the selected preset weights to the state.
+     * @param e - The select change event.
+     */
+    const handleApplyPresetWeights = (e: Event) => {
+        const target = e.target as HTMLSelectElement;
+        let preset = target.value as string;
+        if (preset === '') {
+            preset = state.preset;
+            if (preset === '') {
+                return;
+            }
+        }
+        const presetValues = currentState
+            .getLbwPresets(state.weightState.selectedModelType, state.weightState.selectedLoraBlockType)
+            [preset].split(',')
+            .map((v) => +v);
+
+        const weights: Record<string, globalState.WeightControlState> = {};
+        const maskedLbwBlocks = configManager.getMaskedLbwBlocks(state.weightState.selectedModelType, state.weightState.selectedLoraBlockType);
+        maskedLbwBlocks.forEach((block, i) => {
+            weights[block] = {
+                initValue: state.weightState.weights[block].initValue,
+                value: presetValues[i],
+                sliderMin: Math.min(presetValues[i], state.weightState.weights[block].sliderMin),
+                sliderMax: Math.max(presetValues[i], state.weightState.weights[block].sliderMax),
+            };
+        });
+
+        globalState.setPreset({
+            weights: weights,
+            preset: target.value,
+        });
+    };
+
+    return html`
+        <div class="f g-2 f-end">
+            <select .value="${state.weightState.selectedModelType}" style="flex-grow: 1" ?disabled="${disabled(state)}" @change="${handleChangeModelType}">
+                ${Object.entries(SELECTABLE_MODEL_TYPES).map(([name, modelType]) => html` <option value="${modelType}">${name}</option> `)}
+            </select>
+            <select
+                .value="${state.weightState.selectedLoraBlockType}"
+                style="flex-grow: 1"
+                ?disabled="${disabled(state)}"
+                @change="${handleChangeLoraBlockType}"
+            >
+                ${Object.entries(SELECTABLE_LORA_BLOCK_TYPES).map(([name, loraBlockType]) => html` <option value="${loraBlockType}">${name}</option> `)}
+            </select>
+        </div>
+        <div class="f g-2 f-end">
+            <select .value="${state.preset}" style="flex-grow: 1" ?disabled="${disabled(state)}" @change="${handleApplyPresetWeights}">
+                <option value=""></option>
+                ${Object.keys(state.lbwPresets).map((presetName) => html` <option value="${presetName}">${presetName}</option> `)}
+            </select>
+            <div class="p f g-2 f-end">
+                <div class="f g-2 f-end">
+                    <input
+                        id="wh:xyz"
+                        value="XYZ"
+                        type="checkbox"
+                        .checked="${state.weightState.xyzMode}"
+                        ?disabled="${disabled(state)}"
+                        @change="${handleToggleXyzMode}"
+                    />
+                    <label class="radio-label" for="wh:xyz">
+                        ${configManager.getXyzLabel(state.weightState.selectedModelType, state.weightState.selectedLoraBlockType)}
+                    </label>
+                </div>
+            </div>
+        </div>
+    `;
+}

sd-webui-weight-helper-main/src/components/sections/metadata-section.ts

@@ -0,0 +1,84 @@
+import { resetMetadataAbortController } from '@/shared/manager/api-manager';
+import * as configManager from '@/shared/manager/config-manager';
+import * as currentState from '@/shared/state/current-state';
+import * as globalState from '@/shared/state/global-weight-helper-state';
+import { ModelTypes } from '@/shared/types/lora-types';
+import { fetchMetadata } from '@/shared/utils/api-utils';
+import { mergeWeightsWithGroups } from '@/shared/utils/weight-utils';
+
+import { TemplateResult, html } from 'lit-html';
+
+/**
+ * MetadataSection component displays model and base information.
+ * Provides a button to fetch and analyze detailed metadata for the current LoRA model.
+ * @returns The MetadataSection template.
+ */
+export function createMetadataSection(): TemplateResult {
+    const state = globalState.getGlobalState();
+
+    /**
+     * Handles the click event for the analyze full metadata button.
+     * Initiates fetching and loading of full metadata for the current LoRA model.
+     */
+    const handleAnalyzeFull = () => {
+        if (state.uiState.isWaiting) {
+            return;
+        }
+        const metadataAbortController = resetMetadataAbortController();
+
+        globalState.setWaiting(true);
+
+        fetchMetadata(state.loraName, true, metadataAbortController.signal)
+            .then((result) => {
+                let selectedModelType = state.weightState.selectedModelType;
+                if (!selectedModelType || selectedModelType === ModelTypes.Unknown) {
+                    selectedModelType = result.modelType ?? ModelTypes.Unknown;
+                }
+                const blockGroups = configManager.getLbwBlockGroups(selectedModelType, state.weightState.selectedLoraBlockType);
+                const weights = mergeWeightsWithGroups(state.weightState.weights, blockGroups);
+                globalState.loadMetadata({
+                    algorithm: result.algorithm || 'Unknown',
+                    modelType: result.modelType || 'Unknown',
+                    baseModel: result.baseModel || 'Unknown',
+                    selectedModelType: selectedModelType,
+                    usingBlocks: result.usingBlocks,
+                    blockGroups: blockGroups,
+                    lbwPresets: currentState.getLbwPresets(selectedModelType, state.weightState.selectedLoraBlockType),
+                    weights: weights,
+                });
+            })
+            .catch((error) => {
+                if (error.name !== 'AbortError') {
+                    console.error('Error fetching metadata:', error);
+                    globalState.setWaiting(false);
+                }
+            });
+    };
+
+    return html`
+        <span class="metadata">
+            <span>
+                <span>Model</span>
+                <span>${state.metadataState?.modelType || 'Unknown'}</span>
+            </span>
+            <span>
+                <span>Base</span>
+                <span>${state.metadataState?.baseModel || 'Unknown'}</span>
+            </span>
+            <a class="icon mini svg" @click=${handleAnalyzeFull}>
+                <svg version="1.1" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 492 492">
+                    <g>
+                        <path
+                            d="M484.08,296.216c-5.1-5.128-11.848-7.936-19.032-7.936H330.516c-14.828,0-26.86,12.036-26.86,26.868v22.796c0,7.168,2.784,14.064,7.884,19.16c5.092,5.088,11.82,8.052,18.976,8.052H366.1c-31.544,30.752-74.928,50.08-120.388,50.08c-71.832,0-136.028-45.596-159.744-113.344c-5.392-15.404-19.972-25.784-36.28-25.784c-4.316,0-8.592,0.708-12.7,2.144c-9.692,3.396-17.48,10.352-21.932,19.596c-4.456,9.248-5.04,19.684-1.648,29.368c34.496,98.54,127.692,164.74,232.144,164.74c64.132,0,123.448-23.948,169.572-67.656v25.22c0,14.836,12.384,27.108,27.224,27.108h22.792c14.84,0,26.86-12.272,26.86-27.108V315.24C492,308.056,489.2,301.304,484.08,296.216z"
+                        ></path>
+                    </g>
+                    <g>
+                        <path
+                            d="M478.628,164.78C444.132,66.244,350.916,0.044,246.464,0.044c-64.136,0-123.464,23.952-169.588,67.66v-25.22c0-14.832-12.344-27.112-27.184-27.112H26.896C12.06,15.372,0,27.652,0,42.484V176.76c0,7.18,2.824,13.868,7.944,18.964c5.096,5.128,11.86,7.932,19.044,7.932l-0.08,0.06h134.604c14.84,0,26.832-12.028,26.832-26.86v-22.8c0-14.836-11.992-27.216-26.832-27.216h-35.576c31.544-30.752,74.932-50.076,120.392-50.076c71.832,0,136.024,45.596,159.74,113.348c5.392,15.404,19.968,25.78,36.28,25.78c4.32,0,8.588-0.704,12.7-2.144c9.696-3.396,17.48-10.348,21.932-19.596C481.432,184.9,482.02,174.472,478.628,164.78z"
+                        ></path>
+                    </g>
+                </svg>
+            </a>
+        </span>
+    `;
+}

sd-webui-weight-helper-main/src/components/sections/preview-section.ts

@@ -0,0 +1,167 @@
+import { getEditor, getTabId } from '@/shared/state/current-state';
+import * as globalState from '@/shared/state/global-weight-helper-state';
+import { withoutPromptAssist } from '@/shared/utils/common-utils';
+import { getDisplayStyle, getVisibilityStyle } from '@/shared/utils/helper-utils';
+
+import { TemplateResult, html } from 'lit-html';
+
+declare function extraNetworksRequestMetadata(e: Event, type: string, modelName: string): void;
+declare function extraNetworksEditUserMetadata(e: Event, tabId: string, type: string, modelName: string): void;
+
+/**
+ * PreviewSection creates information display about the selected model.
+ * Shows preview image, metadata actions, trigger word insertion buttons, and model description.
+ * @param preview - The preview state containing model information.
+ * @returns The PreviewSection template.
+ */
+export function createPreviewSection(preview: globalState.PreviewState): TemplateResult {
+    /**
+     * Handles the click event to show internal metadata for the model.
+     * @param e - The click event.
+     */
+    const handleShowMetadata = (e: Event) => {
+        extraNetworksRequestMetadata(e, 'lora', preview.modelName);
+    };
+
+    /**
+     * Handles the click event to edit user metadata for the model.
+     * @param e - The click event.
+     */
+    const handleEditMetadata = (e: Event) => {
+        extraNetworksEditUserMetadata(e, getTabId(), 'lora', preview.modelName);
+    };
+
+    /**
+     * Handles the click event to open the Civitai model page in a new tab.
+     */
+    const handleOpenCivitai = () => {
+        window.open(`https://civitai.com/models/${preview.modelId}`, '_blank');
+    };
+
+    /**
+     * Handles the click event to show the preview description.
+     */
+    const handleOpenPreviewDescription = () => {
+        globalState.setPreviewDescriptionVisible(true);
+    };
+
+    /**
+     * Handles the click event to close the preview description.
+     */
+    const handleClosePreviewDescription = () => {
+        globalState.setPreviewDescriptionVisible(false);
+    };
+
+    /**
+     * Handles the click event to insert trigger words into the prompt or negative prompt textarea.
+     */
+    const handleInsertTriggerWords = () => {
+        const tabId = getTabId();
+        let promptTextarea = document.querySelector(`#${tabId}_prompt textarea`) as HTMLTextAreaElement;
+        let negativeTextarea = document.querySelector(`#${tabId}_neg_prompt textarea`) as HTMLTextAreaElement | undefined;
+
+        const textarea = getEditor();
+        if (textarea === negativeTextarea) {
+        } else if (textarea !== promptTextarea) {
+            promptTextarea = textarea;
+            negativeTextarea = undefined;
+        }
+        if (!opts.weight_helper_using_execCommand) {
+            const insert = () => {
+                const zipped = [
+                    { triggerWords: preview.triggerWords, textarea: promptTextarea },
+                    { triggerWords: preview.negativeTriggerWords, textarea: negativeTextarea },
+                ];
+                for (const { triggerWords, textarea } of zipped) {
+                    if (triggerWords.length > 0 && textarea) {
+                        let words = triggerWords.join(', ');
+                        if (textarea.value) words = ', ' + words;
+                        textarea.value += words;
+                    }
+                }
+            };
+            insert();
+        } else {
+            const insert = () => {
+                const zipped = [
+                    { triggerWords: preview.triggerWords, textarea: promptTextarea },
+                    { triggerWords: preview.negativeTriggerWords, textarea: negativeTextarea },
+                ];
+                for (const { triggerWords, textarea } of zipped) {
+                    if (triggerWords.length > 0 && textarea) {
+                        let words = triggerWords.join(', ');
+                        if (textarea.value) words = ', ' + words;
+                        textarea.focus();
+                        const eolIndex = textarea.value.length;
+                        textarea.setSelectionRange(eolIndex, eolIndex);
+                        document.execCommand('insertText', false, words);
+                    }
+                }
+            };
+            withoutPromptAssist(() => {
+                insert();
+            });
+        }
+        globalState.hideTagInsertButton();
+    };
+
+    const classNames = ['preview-pane', 'card'];
+
+    // position based on settings
+    switch (opts.weight_helper_preview_position) {
+        case 'Bottom Right':
+            classNames.push('preview-bottom-right');
+            break;
+        case 'Top Left':
+            classNames.push('preview-top-left');
+            break;
+        case 'Bottom Left':
+            classNames.push('preview-bottom-left');
+            break;
+        default:
+            classNames.push('preview-top-right');
+            break;
+    }
+
+    return html`
+        <div class="${classNames.join(' ')}" data-name="${preview.modelName}">
+            <img class="preview" src="${preview.thumbUrl}" style="height: ${opts.weight_helper_preview_height}px" />
+            <div class="action-row button-top" style="${getVisibilityStyle(!preview.isDescriptionVisible)}">
+                ${preview.modelName && preview.hasMetadata && typeof extraNetworksRequestMetadata === 'function'
+                    ? html` <div class="metadata-btn card-btn" title="Show internal metadata" @click="${handleShowMetadata}"></div> `
+                    : ''}
+                ${preview.modelName && typeof extraNetworksEditUserMetadata === 'function'
+                    ? html` <div class="edit-btn card-btn" title="Edit metadata" @click="${handleEditMetadata}"></div> `
+                    : ''}
+                ${preview.modelId ? html` <div class="civitai-btn card-btn" title="Open civitai" @click="${handleOpenCivitai}"></div> ` : ''}
+                ${preview.triggerWords.length > 0 || preview.negativeTriggerWords.length > 0
+                    ? html`
+                          <div
+                              class="add-trigger-btn card-btn"
+                              title="Add trigger words"
+                              @click="${handleInsertTriggerWords}"
+                              style="${getDisplayStyle(preview.isTagInsertButtonVisible)}"
+                          ></div>
+                      `
+                    : ''}
+            </div>
+
+            ${preview.description
+                ? html`
+                      <div class="action-row button-bottom" style="${getVisibilityStyle(!preview.isDescriptionVisible)}">
+                          <div class="card-btn note-btn" @click="${handleOpenPreviewDescription}"></div>
+                      </div>
+                      <textarea class="description" style="${getVisibilityStyle(preview.isDescriptionVisible)}">
+                    ${preview.description}
+                </textarea
+                      >
+                      <div
+                          class="card-btn description-close-btn"
+                          style="${getVisibilityStyle(preview.isDescriptionVisible)}"
+                          @click="${handleClosePreviewDescription}"
+                      ></div>
+                  `
+                : ''}
+        </div>
+    `;
+}

sd-webui-weight-helper-main/src/components/ui-template.ts

@@ -0,0 +1,309 @@
+import { CARD_NO_IMAGE_PATH } from '@/shared/constants/common-const';
+import { resetMetadataAbortController, resetPreviewAbortController } from '@/shared/manager/api-manager';
+import * as cacheManager from '@/shared/manager/cache-manager';
+import * as configManager from '@/shared/manager/config-manager';
+import * as historyManager from '@/shared/manager/history-manager';
+import * as currentState from '@/shared/state/current-state';
+import * as globalState from '@/shared/state/global-weight-helper-state';
+import { ModelTypes, WeightControllerTypes } from '@/shared/types/lora-types';
+import { fetchMetadata, fetchPreviewData } from '@/shared/utils/api-utils';
+import { getLoraDefineFromEditor, getOutputStrings, syncEditorWithState } from '@/shared/utils/editor-utils';
+import { disabled, getDisplayStyle } from '@/shared/utils/helper-utils';
+import { createWeightHelperInitState, createWeightHelperState } from '@/shared/utils/state-utils';
+import { mergeWeightsWithGroups } from '@/shared/utils/weight-utils';
+
+import { createWeightController } from '@/components/sections/common-section';
+import { createHeaderSection } from '@/components/sections/header-section';
+import { createLbwSection } from '@/components/sections/lbw-section';
+import { createLoraSettingsSection } from '@/components/sections/lora-settings-section';
+import { createMetadataSection } from '@/components/sections/metadata-section';
+import { createPreviewSection } from '@/components/sections/preview-section';
+
+import { TemplateResult, html, render } from 'lit-html';
+
+declare function gradioApp(): HTMLElement;
+
+// Track if global (one-time) event listeners were attached
+let isEventListenersInitialized = false;
+
+/**
+ * Build main Weight Helper UI template.
+ * Note: top/left positioning styles are applied imperatively (drag) and excluded from state to avoid re-renders.
+ * @returns The main Weight Helper UI template
+ */
+export function createUITemplate(): TemplateResult {
+    const state = globalState.getGlobalState();
+    const { uiState, weightState, lock } = state;
+
+    // Attach global listeners only once
+    if (!isEventListenersInitialized) {
+        setupEventListeners();
+        isEventListenersInitialized = true;
+    }
+
+    // top/left not included here (managed outside state for performance)
+    const containerStyle = `transform: scale(${uiState.scale}); ${getDisplayStyle(uiState.isVisible)}`;
+
+    return html`
+        <div id="weight-helper" class="${state.uiState.isWaiting ? 'waiting' : ''}" style="${containerStyle}">
+            ${createHeaderSection(state.loraName, lock)} ${createMetadataSection()}
+            ${weightState.weights[WeightControllerTypes.TENC] &&
+            html`<section class="border p">${createWeightController(WeightControllerTypes.TENC, weightState.weights)}</section>`}
+            ${[WeightControllerTypes.UNET, WeightControllerTypes.START, WeightControllerTypes.STOP]
+                .filter((wt) => !!weightState.weights[wt])
+                .map(
+                    (weightType) =>
+                        html`<section
+                            class="border p"
+                            style="${getDisplayStyle(
+                                !uiState.isMoreButtonVisible ||
+                                    weightState.weights[weightType].initCheckState ||
+                                    weightState.weights[weightType].initValue !== configManager.getWeightControllerConfig(weightType).default,
+                            )}"
+                        >
+                            ${createWeightController(weightType, weightState.weights)}
+                        </section>`,
+                )}
+
+            <button style="${uiState.isMoreButtonVisible ? '' : 'display: none'}" ?disabled="${disabled(state)}" @click="${() => globalState.hideMoreButton()}">
+                show more options
+            </button>
+
+            <section class="border p">
+                <label>LBW</label>
+                <div class="f col g-4 w-fill">${createLoraSettingsSection()} ${createLbwSection()}</div>
+            </section>
+
+            ${state.previewState.modelName ? createPreviewSection(state.previewState) : ''}
+        </div>
+    `;
+}
+
+/**
+ * Register global event listeners (context menu open/close lifecycle, outside click, ESC key, generate button etc.).
+ */
+function setupEventListeners(): void {
+    const handleClickGenButton = () => {
+        const state = globalState.getGlobalState();
+        if (state.uiState.isVisible) {
+            closeWeightHelperContext();
+            syncEditorWithState(state);
+        }
+    };
+
+    const handleRightClickTextarea = (e: MouseEvent) => {
+        const state = globalState.getGlobalState();
+        if (state.uiState.isVisible) {
+            e.preventDefault();
+            closeWeightHelperContext();
+            syncEditorWithState(state);
+            return;
+        }
+        const loraDefine = getLoraDefineFromEditor(e);
+        if (loraDefine) {
+            e.preventDefault();
+            openWeightHelperContext(loraDefine);
+        }
+    };
+
+    const handleClickAnyware = (e: MouseEvent) => {
+        const state = globalState.getGlobalState();
+        if (state.uiState.isVisible) {
+            const contextMenuElement = document.getElementById('weight-helper');
+            if (!contextMenuElement) return;
+            const tabId = currentState.getTabId();
+            if (e) {
+                const target = e.target as HTMLBodyElement;
+                if (contextMenuElement.contains(target)) return;
+                if (target.id === `${tabId}_token_button`) return;
+                if (target.id === `${tabId}_lora_edit_user_metadata_button`) return;
+                if (target.className === 'global-popup-close') return;
+                if (target.id.indexOf('_interrupt') > 0) {
+                    closeWeightHelperContext();
+                    return;
+                }
+            }
+            closeWeightHelperContext();
+            syncEditorWithState(state);
+        }
+    };
+
+    const handleKeyupAnyware = (e: KeyboardEvent) => {
+        if ((e as KeyboardEvent).key === 'Escape') {
+            const state = globalState.getGlobalState();
+            cancelWeightHelperContext(state);
+        }
+    };
+
+    // Attach listeners to generate buttons & prompt textareas
+    const genButtons = gradioApp().querySelectorAll("button:is([id*='_generate'])");
+    genButtons.forEach((genBtn) => {
+        genBtn.addEventListener('click', handleClickGenButton, true);
+    });
+
+    const textareas = gradioApp().querySelectorAll("*:is([id*='_toprow'] [id*='_prompt'], .prompt) textarea") as NodeListOf<HTMLTextAreaElement>;
+    textareas.forEach((textarea) => {
+        textarea.addEventListener('contextmenu', handleRightClickTextarea);
+    });
+
+    document.addEventListener('click', handleClickAnyware);
+    document.addEventListener('keyup', handleKeyupAnyware);
+}
+
+// One-time flag for state watching → subscribe only once
+let isStateWatchingInitialized = false;
+
+/**
+ * Render (and set up subscriptions) of the UI into container.
+ * Full state changes trigger template re-render; position changes only mutate style.
+ * @param container Target DOM element
+ */
+export function renderUI(container: HTMLElement): void {
+    // 初期レンダリング
+    render(createUITemplate(), container);
+
+    // 状態変化を監視して自動再レンダリング（一度だけ設定）
+    if (!isStateWatchingInitialized) {
+        // Subscribe to full state updates
+        globalState.subscribeToStateChanges(() => {
+            render(createUITemplate(), container);
+        });
+
+        // Subscribe to position-only updates (avoid re-render)
+        globalState.subscribeToPositionChanges((pos) => {
+            const el = container.querySelector('#weight-helper') as HTMLDivElement;
+            if (!el) return;
+            el.style.top = pos.top + 'px';
+            el.style.left = pos.left + 'px';
+        });
+
+        isStateWatchingInitialized = true;
+    }
+}
+
+/**
+ * Open Weight Helper context at given coordinates and load metadata / preview lazily.
+ */
+function openWeightHelperContext(loraDefine: {
+    tabId: string;
+    textarea: HTMLTextAreaElement;
+    selectionStart: number;
+    selectionEnd: number;
+    namespace: string;
+    loraName: string;
+    loraParams: string;
+    top: number;
+    left: number;
+}) {
+    const { tabId, textarea, selectionStart, selectionEnd, namespace, loraName, loraParams, top, left } = loraDefine;
+
+    currentState.setTabId(tabId);
+    currentState.setEditor(textarea);
+    currentState.setSelectionStart(selectionStart);
+    currentState.setSelectionEnd(selectionEnd);
+    currentState.setLoraDefineString(textarea.value.substring(selectionStart, selectionEnd));
+
+    currentState.loadStep(tabId);
+    currentState.loadLbwPresets(tabId);
+
+    configManager.getWeightControllerConfig(WeightControllerTypes.START).max = currentState.getStep();
+    configManager.getWeightControllerConfig(WeightControllerTypes.STOP).max = currentState.getStep();
+    configManager.getWeightControllerConfig(WeightControllerTypes.STOP).default = currentState.getStep();
+
+    const rect = document.getElementById('weight-helper')!.getBoundingClientRect();
+    const diffBottom = window.innerHeight - rect.bottom;
+    let arrangedTop = top;
+    if (diffBottom < 0) {
+        if (rect.top < 0) {
+            arrangedTop = window.scrollY;
+        } else {
+            arrangedTop += diffBottom;
+        }
+    }
+    const props = createWeightHelperState(namespace, loraName, loraParams);
+
+    globalState.openContext(props);
+    // Apply position directly (not part of state)
+    const container = document.getElementById('weight-helper');
+    if (container) {
+        container.style.top = arrangedTop + 'px';
+        container.style.left = left + 'px';
+    }
+
+    if (!props.metadataState) {
+        const metadataAbortController = resetMetadataAbortController();
+        fetchMetadata(loraName, false, metadataAbortController.signal)
+            .then((result) => {
+                let selectedModelType = props.weightState.selectedModelType;
+                if (!selectedModelType || selectedModelType === ModelTypes.Unknown) {
+                    selectedModelType = result.modelType ?? ModelTypes.Unknown;
+                }
+                const selectedLoraBlockType = props.weightState.selectedLoraBlockType;
+                const blockGroups = configManager.getLbwBlockGroups(selectedModelType, selectedLoraBlockType);
+                const weights = mergeWeightsWithGroups(props.weightState.weights, blockGroups);
+
+                const weightState = {
+                    ...props.weightState,
+                    selectedModelType: selectedModelType,
+                    selectedLoraBlockType: selectedLoraBlockType,
+                };
+                const loraParams = getOutputStrings(props.loraName, weightState).loraParams;
+                if (!props.weightState.xyzMode) {
+                    historyManager.addHistory(loraName, {
+                        loraParams: loraParams,
+                        selectedLoraBlockType: selectedLoraBlockType,
+                        selectedModelType: selectedModelType,
+                    });
+                    globalState.setHistoryIndex(historyManager.getHistories(loraName).length - 1);
+                }
+
+                globalState.loadMetadata({
+                    algorithm: result.algorithm || 'Unknown',
+                    modelType: result.modelType || 'Unknown',
+                    baseModel: result.baseModel || 'Unknown',
+                    selectedModelType: selectedModelType,
+                    usingBlocks: result.usingBlocks,
+                    blockGroups: blockGroups,
+                    lbwPresets: currentState.getLbwPresets(selectedModelType, props.weightState.selectedLoraBlockType),
+                    weights: weights,
+                });
+            })
+            .catch((error) => {
+                if (error.name !== 'AbortError') {
+                    console.error('Error fetching metadata:', error);
+                    globalState.setWaiting(false);
+                }
+            });
+    }
+
+    if (!props.previewState.modelId) {
+        const previewAbortController = resetPreviewAbortController();
+        fetchPreviewData(loraName, previewAbortController.signal)
+            .then(async (result) => {
+                if (result.thumbUrl !== CARD_NO_IMAGE_PATH) {
+                    result.thumbUrl = result.thumbUrl;
+                }
+                cacheManager.setPreviewCache(loraName, result);
+                globalState.loadPreviewData(result);
+            })
+            .catch((error) => {
+                if (error.name !== 'AbortError') {
+                    console.error('Error fetching metadata:', error);
+                }
+            });
+    }
+}
+
+/** Close context menu and reset to initial hidden state. */
+function closeWeightHelperContext(): void {
+    const weightHelperState = createWeightHelperInitState();
+    globalState.closeContext(weightHelperState);
+}
+
+/** Cancel editing: revert editor text (if enabled) then close. */
+function cancelWeightHelperContext(state: globalState.BasicState): void {
+    if (!opts.weight_helper_using_execCommand) {
+        syncEditorWithState(state, true);
+    }
+    closeWeightHelperContext();
+}

sd-webui-weight-helper-main/src/index.ts

@@ -0,0 +1,61 @@
+import * as configManager from '@/shared/manager/config-manager';
+import * as historyManager from '@/shared/manager/history-manager';
+import * as globalState from '@/shared/state/global-weight-helper-state';
+import { createWeightHelperInitState } from '@/shared/utils/state-utils';
+
+import { renderUI } from '@/components/ui-template';
+
+declare function gradioApp(): HTMLElement;
+declare function onUiLoaded(callback: VoidFunction): void;
+declare function onOptionsChanged(callback: VoidFunction): void;
+
+/**
+ * Promise resolver for the initialization process.
+ * Used to signal when options have been loaded and initialization can proceed.
+ */
+export let resolveInitialized: ((value: boolean) => void) | null;
+
+/**
+ * Promise that resolves when the Weight Helper is ready to initialize.
+ * Used to defer initialization until options are loaded.
+ */
+const initializedPromise = new Promise<boolean>((resolve) => {
+    resolveInitialized = resolve;
+});
+
+/**
+ * Entry point for the Weight Helper React application.
+ * Handles initialization, theme adaptation, and state setup when the UI is loaded or options are changed.
+ */
+historyManager.loadLocalStorage();
+
+onUiLoaded(() => {
+    if (!gradioApp().querySelector('#lbw_ratiospreset')) {
+        return;
+    }
+    const textColor = getComputedStyle(document.documentElement).getPropertyValue('--body-text-color').trim();
+    const textColorRgb = textColor
+        .slice(1)
+        .match(/.{1,2}/g)!
+        .map((hex) => parseInt(hex, 16));
+    const textColorRgba = [...textColorRgb, 0.3];
+    document.documentElement.style.setProperty('--weight-helper-shadow', `rgba(${textColorRgba.join(',')})`);
+    document.documentElement.style.setProperty('--weight-helper-slider_size', String(opts.weight_helper_slider_length));
+
+    initializedPromise.then(() => {
+        const weightHelperContainer = document.createElement('div');
+        weightHelperContainer.id = 'weight-helper-container';
+        gradioApp().appendChild(weightHelperContainer);
+
+        configManager.initialize();
+        globalState.initializeGlobalState(createWeightHelperInitState());
+        renderUI(weightHelperContainer);
+    });
+});
+
+onOptionsChanged(() => {
+    if (resolveInitialized) {
+        resolveInitialized(true);
+        resolveInitialized = null;
+    }
+});

sd-webui-weight-helper-main/src/shared/constants/common-const.ts

@@ -0,0 +1,172 @@
+import { LoraBlockTypes, ModelTypes } from '@/shared/types/lora-types';
+
+const BASE_MODEL_TYPE = {
+    STABLE_DIFFUSION: [
+        'BASE',
+        'IN00',
+        'IN01',
+        'IN02',
+        'IN03',
+        'IN04',
+        'IN05',
+        'IN06',
+        'IN07',
+        'IN08',
+        'IN09',
+        'IN10',
+        'IN11',
+        'M00',
+        'OUT00',
+        'OUT01',
+        'OUT02',
+        'OUT03',
+        'OUT04',
+        'OUT05',
+        'OUT06',
+        'OUT07',
+        'OUT08',
+        'OUT09',
+        'OUT10',
+        'OUT11',
+    ],
+    BLACK_FOREST_LABS: [
+        'CLIP',
+        'T5',
+        'IN',
+        'D00',
+        'D01',
+        'D02',
+        'D03',
+        'D04',
+        'D05',
+        'D06',
+        'D07',
+        'D08',
+        'D09',
+        'D10',
+        'D11',
+        'D12',
+        'D13',
+        'D14',
+        'D15',
+        'D16',
+        'D17',
+        'D18',
+        'S00',
+        'S01',
+        'S02',
+        'S03',
+        'S04',
+        'S05',
+        'S06',
+        'S07',
+        'S08',
+        'S09',
+        'S10',
+        'S11',
+        'S12',
+        'S13',
+        'S14',
+        'S15',
+        'S16',
+        'S17',
+        'S18',
+        'S19',
+        'S20',
+        'S21',
+        'S22',
+        'S23',
+        'S24',
+        'S25',
+        'S26',
+        'S27',
+        'S28',
+        'S29',
+        'S30',
+        'S31',
+        'S32',
+        'S33',
+        'S34',
+        'S35',
+        'S36',
+        'S37',
+        'OUT',
+    ],
+};
+
+export const LBW_BLOCKS = {
+    [ModelTypes.SD]: BASE_MODEL_TYPE.STABLE_DIFFUSION,
+    [ModelTypes.SDXL]: BASE_MODEL_TYPE.STABLE_DIFFUSION,
+    [ModelTypes.Flux]: BASE_MODEL_TYPE.BLACK_FOREST_LABS,
+    [ModelTypes.Unknown]: [],
+};
+
+export const lbwMaskDefine = {
+    [ModelTypes.SD]: {
+        [LoraBlockTypes.lora]: [1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1],
+        [LoraBlockTypes.lycoris]: [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
+        [LoraBlockTypes.Unknown]: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+    },
+    [ModelTypes.SDXL]: {
+        [LoraBlockTypes.lora]: [1, 0, 0, 0, 0, 1, 1, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0],
+        [LoraBlockTypes.lycoris]: [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0],
+        [LoraBlockTypes.Unknown]: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+    },
+    [ModelTypes.Flux]: {
+        [LoraBlockTypes.lora]: [
+            1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
+            1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
+        ],
+        [LoraBlockTypes.lycoris]: [
+            1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
+            1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
+        ],
+        [LoraBlockTypes.Unknown]: [
+            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
+            0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
+        ],
+    },
+    [ModelTypes.Unknown]: {
+        [LoraBlockTypes.lora]: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+        [LoraBlockTypes.lycoris]: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+        [LoraBlockTypes.Unknown]: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+    },
+};
+
+export const XYZ_PRESETS = {
+    [ModelTypes.SD]: {
+        [LoraBlockTypes.lora]: 'XYZ(17)',
+        [LoraBlockTypes.lycoris]: 'XYZ(26)',
+        [LoraBlockTypes.Unknown]: 'XYZ(26)',
+    },
+    [ModelTypes.SDXL]: {
+        [LoraBlockTypes.lora]: 'XYZ(12)',
+        [LoraBlockTypes.lycoris]: 'XYZ(20)',
+        [LoraBlockTypes.Unknown]: 'XYZ(26)',
+    },
+    [ModelTypes.Flux]: {
+        [LoraBlockTypes.lora]: 'XYZ(19)',
+        [LoraBlockTypes.lycoris]: 'XYZ(19)',
+        [LoraBlockTypes.Unknown]: 'XYZ(19)',
+    },
+    [ModelTypes.Unknown]: {
+        [LoraBlockTypes.lora]: 'XYZ(17)',
+        [LoraBlockTypes.lycoris]: 'XYZ(26)',
+        [LoraBlockTypes.Unknown]: 'XYZ(26)',
+    },
+};
+
+export const SELECTABLE_MODEL_TYPES = {
+    '': ModelTypes.Unknown,
+    SD: ModelTypes.SD,
+    SDXL: ModelTypes.SDXL,
+    Flux: ModelTypes.Flux,
+};
+
+export const SELECTABLE_LORA_BLOCK_TYPES = {
+    '': LoraBlockTypes.Unknown,
+    'LoRA(LierLa)': LoraBlockTypes.lora,
+    'LyCORIS,etc': LoraBlockTypes.lycoris,
+};
+
+export const CARD_NO_IMAGE_PATH = './file=html/card-no-preview.png';

sd-webui-weight-helper-main/src/shared/hooks/use-global-state.ts

@@ -0,0 +1,11 @@
+// このファイルはlit-html移行により不要になりました
+// React/Preactのhooksは使用しないため、ダミーファイルとして残しています
+import { BasicState, getGlobalState } from '@/shared/state/global-weight-helper-state';
+
+/**
+ * 互換性のためのダミー関数
+ * lit-htmlベースではこの関数は使用されません
+ */
+export function useGlobalState(): BasicState {
+    return getGlobalState();
+}

sd-webui-weight-helper-main/src/shared/manager/api-manager.ts

@@ -0,0 +1,50 @@
+let metadataAbortController: AbortController | null = null;
+let previewAbortController: AbortController | null = null;
+
+/**
+ * Returns the global AbortController for metadata fetch requests.
+ * Creates a new controller if one doesn't exist.
+ * @returns The AbortController instance for metadata requests.
+ */
+export function getMetadataAbortController(): AbortController {
+    if (!metadataAbortController) {
+        metadataAbortController = new AbortController();
+    }
+    return metadataAbortController;
+}
+
+/**
+ * Aborts any ongoing metadata fetch request and creates a new AbortController.
+ * @returns The new AbortController instance for metadata requests.
+ */
+export function resetMetadataAbortController(): AbortController {
+    if (metadataAbortController) {
+        metadataAbortController.abort();
+    }
+    metadataAbortController = new AbortController();
+    return metadataAbortController;
+}
+
+/**
+ * Returns the global AbortController for preview fetch requests.
+ * Creates a new controller if one doesn't exist.
+ * @returns The AbortController instance for preview requests.
+ */
+export function getPreviewAbortController(): AbortController {
+    if (!previewAbortController) {
+        previewAbortController = new AbortController();
+    }
+    return previewAbortController;
+}
+
+/**
+ * Aborts any ongoing preview fetch request and creates a new AbortController.
+ * @returns The new AbortController instance for preview requests.
+ */
+export function resetPreviewAbortController(): AbortController {
+    if (previewAbortController) {
+        previewAbortController.abort();
+    }
+    previewAbortController = new AbortController();
+    return previewAbortController;
+}

sd-webui-weight-helper-main/src/shared/manager/cache-manager.ts

@@ -0,0 +1,62 @@
+import * as globalState from '@/shared/state/global-weight-helper-state';
+import { LoraBlockTypes, ModelTypes } from '@/shared/types/lora-types';
+
+const _metadataCacheStore: Record<
+    string,
+    {
+        metadataState: globalState.MetadataState | undefined;
+        selectedLoraBlockType: LoraBlockTypes;
+        selectedModelType: ModelTypes;
+        usingBlocks: Set<string> | undefined;
+    }
+> = {};
+
+const _previewCacheStore: Record<string, globalState.PreviewState> = {};
+
+/**
+ * Retrieves the metadata cache for the given LoRA name.
+ * @param loraName - The name of the LoRA model.
+ * @returns The cached metadata state, selected block type, model type, and used blocks, or undefined if not cached.
+ */
+export function getMetadataCache(loraName: string):
+    | {
+          metadataState: globalState.MetadataState | undefined;
+          selectedLoraBlockType: LoraBlockTypes;
+          selectedModelType: ModelTypes;
+          usingBlocks: Set<string> | undefined;
+      }
+    | undefined {
+    return _metadataCacheStore[loraName];
+}
+
+/**
+ * Sets the metadata cache for the given LoRA name.
+ * @param loraName - The name of the LoRA model.
+ * @param cache - The BasicState containing metadata and related info to cache.
+ */
+export function setMetadataCache(loraName: string, cache: globalState.BasicState) {
+    _metadataCacheStore[loraName] = {
+        metadataState: cache.metadataState,
+        selectedLoraBlockType: cache.weightState.selectedLoraBlockType,
+        selectedModelType: cache.weightState.selectedModelType,
+        usingBlocks: cache.usingBlocks,
+    };
+}
+
+/**
+ * Retrieves the preview cache for the given LoRA name.
+ * @param loraName - The name of the LoRA model.
+ * @returns The cached PreviewState, or undefined if not cached.
+ */
+export function getPreviewCache(loraName: string): globalState.PreviewState | undefined {
+    return _previewCacheStore[loraName];
+}
+
+/**
+ * Sets the preview cache for the given LoRA name.
+ * @param loraName - The name of the LoRA model.
+ * @param cache - The PreviewState to cache.
+ */
+export function setPreviewCache(loraName: string, cache: globalState.PreviewState) {
+    _previewCacheStore[loraName] = cache;
+}

sd-webui-weight-helper-main/src/shared/manager/config-manager.ts

@@ -0,0 +1,167 @@
+import {
+    LBW_BLOCKS,
+    lbwMaskDefine as LBW_MASK_DEFINE,
+    SELECTABLE_LORA_BLOCK_TYPES,
+    SELECTABLE_MODEL_TYPES,
+    XYZ_PRESETS,
+} from '@/shared/constants/common-const';
+import { LoraBlockTypes, ModelTypes, WeightControllerTypes } from '@/shared/types/lora-types';
+
+interface WeightControllerInterface {
+    min: number;
+    max: number;
+    default: number;
+    step: number;
+}
+
+const WeightControllerConfig: Record<string, WeightControllerInterface> = {
+    [WeightControllerTypes.TENC]: {
+        min: 0,
+        max: 0,
+        default: 1,
+        step: 0.5,
+    },
+    [WeightControllerTypes.UNET]: {
+        min: 0,
+        max: 0,
+        default: 0,
+        step: 0.5,
+    },
+    [WeightControllerTypes.START]: {
+        min: 0,
+        max: 0,
+        default: 0,
+        step: 1,
+    },
+    [WeightControllerTypes.STOP]: {
+        min: 0,
+        max: 0,
+        default: 0,
+        step: 1,
+    },
+    [WeightControllerTypes.LBW]: {
+        min: 0,
+        max: 0,
+        default: 1,
+        step: 0.5,
+    },
+};
+
+const lbwGroupConfig = {
+    [ModelTypes.SD]: {
+        [LoraBlockTypes.lora]: [['BASE', 'IN01-IN04', 'IN05-IN08', 'M00', 'OUT03-OUT06', 'OUT07-OUT11']],
+        [LoraBlockTypes.lycoris]: [['BASE', 'IN00-IN05', 'IN06-IN11', 'M00', 'OUT00-OUT05', 'OUT06-OUT11']],
+        [LoraBlockTypes.Unknown]: [[]] as string[][],
+    },
+    [ModelTypes.SDXL]: {
+        [LoraBlockTypes.lora]: [['BASE', 'IN04-IN08', 'M00', 'OUT00-OUT05']],
+        [LoraBlockTypes.lycoris]: [['BASE', 'IN00-IN03', 'IN04-IN08', 'M00', 'OUT00-OUT03', 'OUT04-OUT08']],
+        [LoraBlockTypes.Unknown]: [[]] as string[][],
+    },
+    [ModelTypes.Flux]: {
+        [LoraBlockTypes.lora]: [['CLIP', 'T5', 'IN', 'OUT'], ['D00-D18'], ['S00-S18'], ['S19-S37']],
+        [LoraBlockTypes.lycoris]: [['CLIP', 'T5', 'IN', 'OUT'], ['D00-D18'], ['S00-S18'], ['S19-S37']],
+        [LoraBlockTypes.Unknown]: [[]] as string[][],
+    },
+    [ModelTypes.Unknown]: {
+        [LoraBlockTypes.lora]: [[]] as string[][],
+        [LoraBlockTypes.lycoris]: [[]] as string[][],
+        [LoraBlockTypes.Unknown]: [[]] as string[][],
+    },
+};
+
+/**
+ * Initializes the configuration for block groups and weight controllers.
+ * Updates configuration values with settings from opts.
+ */
+export function initialize() {
+    const optBlockPattern = /((BASE|MID|M00|CLIP|T5|(IN|OUT|D|S)[0-9]{2}(-(IN|OUT|D|S)[0-9]{2})?) *(, *|$))+/;
+    for (const modelType of Object.values(SELECTABLE_MODEL_TYPES)) {
+        if (modelType === ModelTypes.Unknown || modelType === ModelTypes.Flux) continue;
+        for (const loraBlockType of Object.values(SELECTABLE_LORA_BLOCK_TYPES)) {
+            if (loraBlockType === LoraBlockTypes.Unknown) continue;
+            try {
+                let optBlockPoints = opts[`weight_helper_LBW_${modelType}_${loraBlockType}_block_points`] as string;
+                optBlockPoints = optBlockPoints.replace('MID', 'M00');
+                if (optBlockPattern.exec(optBlockPoints)) {
+                    const blockPoints = optBlockPoints.split(',').map((v) => {
+                        return v.trim().replace(/\d+/g, (match) => (match.length === 1 ? `0${match}` : match));
+                    });
+                    lbwGroupConfig[modelType][loraBlockType] = [blockPoints];
+                }
+            } catch (e) {
+                console.warn(`${modelType}_${loraBlockType} block definition format is invalid.`, e);
+            }
+        }
+    }
+
+    for (const k of [WeightControllerTypes.TENC, WeightControllerTypes.UNET, WeightControllerTypes.LBW]) {
+        WeightControllerConfig[k].min = +opts[`weight_helper_${k}_min`];
+        WeightControllerConfig[k].max = +opts[`weight_helper_${k}_max`];
+        WeightControllerConfig[k].step = +opts[`weight_helper_${k}_step`];
+    }
+}
+
+/**
+ * Retrieves the configuration for a specific weight controller type.
+ * @param type - The weight controller type.
+ * @returns The configuration object for the specified controller type.
+ */
+export function getWeightControllerConfig(type: string): WeightControllerInterface {
+    return WeightControllerConfig[type];
+}
+
+/**
+ * Returns the list of LBW blocks for the selected model type.
+ * @param selectedModelType - The selected model type.
+ * @returns An array of block labels for the given model type.
+ */
+export function getLbwBlocks(selectedModelType: ModelTypes): string[] {
+    if (selectedModelType in LBW_BLOCKS) {
+        return LBW_BLOCKS[selectedModelType];
+    }
+    return LBW_BLOCKS[ModelTypes.Unknown];
+}
+
+/**
+ * Returns an array of enabled LBW block labels for the selected model and LoRA block type.
+ * Only blocks with a mask value of 1 are included.
+ * @param selectedModelType - The selected model type.
+ * @param selectedLoraBlockType - The selected LoRA block type.
+ * @returns An array of enabled block labels.
+ */
+export function getMaskedLbwBlocks(selectedModelType: ModelTypes, selectedLoraBlockType: LoraBlockTypes): string[] {
+    const masks = getLbwMasks(selectedModelType, selectedLoraBlockType);
+    return LBW_BLOCKS[selectedModelType || ModelTypes.Unknown].filter((_, i) => masks[i] === 1);
+}
+
+/**
+ * Returns the mask array for LBW blocks for the selected model and block type.
+ * Each value in the array (1 or 0) indicates whether the corresponding block is enabled.
+ * @param selectedModelType - The selected model type.
+ * @param selectedLoraBlockType - The selected LoRA block type.
+ * @returns An array of binary values (1=enabled, 0=disabled).
+ */
+export function getLbwMasks(selectedModelType: ModelTypes, selectedLoraBlockType: LoraBlockTypes): number[] {
+    return LBW_MASK_DEFINE[selectedModelType || ModelTypes.Unknown][selectedLoraBlockType || LoraBlockTypes.Unknown];
+}
+
+/**
+ * Returns the block groups configuration for the selected model and block type.
+ * @param selectedModelType - The selected model type.
+ * @param selectedLoraBlockType - The selected LoRA block type.
+ * @returns An array of block group identifiers.
+ */
+export function getLbwBlockGroups(selectedModelType: ModelTypes, selectedLoraBlockType: LoraBlockTypes): string[][] {
+    return lbwGroupConfig[selectedModelType || ModelTypes.Unknown][selectedLoraBlockType || LoraBlockTypes.Unknown];
+}
+
+/**
+ * Returns the XYZ label for the selected model and block type.
+ * @param selectedModelType - The selected model type.
+ * @param selectedLoraBlockType - The selected LoRA block type.
+ * @returns The XYZ identifier string.
+ */
+export function getXyzLabel(selectedModelType: ModelTypes, selectedLoraBlockType: LoraBlockTypes): string {
+    return XYZ_PRESETS[selectedModelType || ModelTypes.Unknown][selectedLoraBlockType || LoraBlockTypes.Unknown];
+}