add lora user metadata editor dialog inspired by MrKuenning's mockup from #7458

1 files changed, 14 insertions, 3 deletions

diff --git a/extensions-builtin/Lora/ui_extra_networks_lora.py b/extensions-builtin/Lora/ui_extra_networks_lora.py
index 29b16c1c..95296275 100644
--- a/extensions-builtin/Lora/ui_extra_networks_lora.py
+++ b/extensions-builtin/Lora/ui_extra_networks_lora.py
@@ -3,6 +3,7 @@ import os
 import lora

 

 from modules import shared, ui_extra_networks

+from ui_edit_user_metadata import LoraUserMetadataEditor

 

 

 class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):

@@ -18,19 +19,29 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
 

             alias = lora_on_disk.get_alias()

 

-            yield {

+            item = {

                 "name": name,

                 "filename": lora_on_disk.filename,

                 "preview": self.find_preview(path),

                 "description": self.find_description(path),

                 "search_term": self.search_terms_from_path(lora_on_disk.filename),

-                "prompt": json.dumps(f"<lora:{alias}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),

                 "local_preview": f"{path}.{shared.opts.samples_format}",

                 "metadata": json.dumps(lora_on_disk.metadata, indent=4) if lora_on_disk.metadata else None,

                 "sort_keys": {'default': index, **self.get_sort_keys(lora_on_disk.filename)},

-

             }

 

+            self.read_user_metadata(item)

+            activation_text = item["user_metadata"].get("activation text")

+            preferred_weight = item["user_metadata"].get("preferred weight", 0.0)

+            item["prompt"] = json.dumps(f"<lora:{alias}:") + " + " + (str(preferred_weight) if preferred_weight else "opts.extra_networks_default_multiplier") + " + " + json.dumps(">")

+

+            if activation_text:

+                item["prompt"] += " + " + json.dumps(" " + activation_text)

+

+            yield item

+

     def allowed_directories_for_previews(self):

         return [shared.cmd_opts.lora_dir]

 

+    def create_user_metadata_editor(self, ui, tabname):

+        return LoraUserMetadataEditor(ui, tabname, self)