Parse lora weights from civitai resources

extensions-builtin/Lora/network.py

@@ -81,7 +81,7 @@ def read_hash(self):
 
     def get_alias(self):
         import networks
-        if shared.opts.lora_preferred_name == "Filename" or self.alias.lower() in networks.forbidden_network_aliases:
+        if (hasattr(shared.opts, "lora_preferred_name") and shared.opts.lora_preferred_name == "Filename") or self.alias.lower() in networks.forbidden_network_aliases:
             return self.name
         else:
             return self.alias

extensions-builtin/Lora/networks.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import json
 import logging
 import os
 import re
@@ -571,6 +572,7 @@ def list_available_networks():
     available_network_aliases.clear()
     forbidden_network_aliases.clear()
     available_network_hash_lookup.clear()
+    network_version_id_to_alias.clear()
     forbidden_network_aliases.update({"none": 1, "Addams": 1})
 
     os.makedirs(shared.cmd_opts.lora_dir, exist_ok=True)
@@ -596,6 +598,17 @@ def list_available_networks():
         available_network_aliases[name] = entry
         available_network_aliases[entry.alias] = entry
 
+        # Try to get modelVersionId from model json
+        json_path = os.path.join(os.path.dirname(filename), name + ".json")
+        if os.path.exists(json_path):
+            try:
+                with open(json_path, "r", encoding="utf-8") as f:
+                    model_json = json.load(f)
+                if "modelVersionId" in model_json:
+                    model_ver_id = str(model_json["modelVersionId"])
+                    network_version_id_to_alias[model_ver_id] = entry.get_alias()
+            except (OSError, json.JSONDecodeError):
+                errors.report(f"Failed to load network json: {json_path}", exc_info=False)
 
 re_network_name = re.compile(r"(.*)\s*\([0-9a-fA-F]+\)")
 
@@ -642,5 +655,6 @@ def infotext_pasted(infotext, params):
 networks_in_memory = {}
 available_network_hash_lookup = {}
 forbidden_network_aliases = {}
+network_version_id_to_alias = {}
 
 list_available_networks()

modules/infotext_utils.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 import base64
+import importlib
 import io
 import json
 import os
@@ -230,6 +231,45 @@ def restore_old_hires_fix_params(res):
     res['Hires resize-2'] = height
 
 
+def parse_civitai_resources(x: str):
+    """parses the Civitai resources string to automatically set the model weights in the prompt:
+```
+Civitai resources: [{"type":"checkpoint","modelVersionId":290640},{"type":"ImageJobNetworkParams { Strength = 0.4, TriggerWord = , Type = lora }","weight":0.4,"modelVersionId":135867},
+{"type":"embed","modelVersionId":250708}]
+```
+    returns a string to append to the prompt
+    """
+    x = "".join(x.strip().split("\n"))
+    if "Civitai resources:" not in x:
+        return ""
+    x = x[x.find("Civitai resources:") + 19:]
+    if "[" not in x or "]" not in x:
+        return ""
+
+    x = x[x.find("["):x.find("]") + 1]
+    try:
+        resources = json.loads(x)
+    except json.JSONDecodeError:
+        return ""
+
+    prompt_resources = []
+    for resource in resources:
+        if "weight" not in resource or "modelVersionId" not in resource:
+            continue
+
+        weight = resource["weight"]
+        model_version_id = str(resource["modelVersionId"])
+        networks = importlib.import_module("extensions-builtin.Lora.networks")
+        if model_version_id in networks.network_version_id_to_alias:
+            model_name = networks.network_version_id_to_alias[model_version_id]
+            prompt_resource = f"<lora:{model_name}:{weight}>"
+            prompt_resources.append(prompt_resource)
+
+    if prompt_resources:
+        return " " + " ".join(prompt_resources)
+    return ""
+
+
 def parse_generation_parameters(x: str, skip_fields: list[str] | None = None):
     """parses generation parameters string, the one you see in text field under the picture in UI:
 ```
@@ -296,6 +336,9 @@ def parse_generation_parameters(x: str, skip_fields: list[str] | None = None):
             if (shared.opts.infotext_styles == "Apply if any" and found_styles) or shared.opts.infotext_styles == "Apply":
                 res['Styles array'] = found_styles
 
+    # Parse civitai resources
+    prompt += parse_civitai_resources(x)
+
     res["Prompt"] = prompt
     res["Negative prompt"] = negative_prompt