Avoid reusing loras from prompted confinements

Author: mcmonkey4eva

src/BuiltinExtensions/ComfyUIBackend/WorkflowGenerator.cs

@@ -238,7 +238,7 @@ public void DownloadModel(string name, string filePath, string url)
             if (confinements is not null && confinements.Count > i)
             {
                 int confinementId = int.Parse(confinements[i]);
-                if (confinementId != confinement)
+                if (confinementId >= 0 && confinementId != confinement)
                 {
                     continue;
                 }

src/Text2Image/T2IParamInput.cs

@@ -378,20 +378,17 @@ static T2IParamInput()
             weights.Add(strength.ToString());
             context.Input.Set(T2IParamTypes.Loras, loraList);
             context.Input.Set(T2IParamTypes.LoraWeights, weights);
-            if (context.SectionID > 0)
+            if (confinements is null)
             {
-                if (confinements is null)
+                confinements = [];
+                for (int i = 0; i < loraList.Count - 1; i++)
                 {
-                    confinements = [];
-                    for (int i = 0; i < loraList.Count - 1; i++)
-                    {
-                        confinements.Add("0");
-                    }
+                    confinements.Add("-1");
                 }
-                Logs.Verbose($"LoRA {lora} confined to section {context.SectionID}.");
-                confinements.Add($"{context.SectionID}");
-                context.Input.Set(T2IParamTypes.LoraSectionConfinement, confinements);
             }
+            Logs.Verbose($"LoRA {lora} confined to section {context.SectionID}.");
+            confinements.Add($"{context.SectionID}");
+            context.Input.Set(T2IParamTypes.LoraSectionConfinement, confinements);
             return "";
         };
         PromptTagPostProcessors["segment"] = (data, context) =>

src/wwwroot/js/genpage/main.js

@@ -82,6 +82,24 @@ function copy_current_image_params() {
     if ('original_negativeprompt' in metadata) {
         metadata.negativeprompt = metadata.original_negativeprompt;
     }
+    if ('lorasectionconfinement' in metadata && 'loras' in metadata && 'loraweights' in metadata) {
+        let confinements = metadata.lorasectionconfinement;
+        let loras = metadata.loras;
+        let weights = metadata.loraweights;
+        if (confinements.length == loras.length && loras.length == weights.length) {
+            let newLoras = [];
+            let newWeights = [];
+            for (let i = 0; i < confinements.length; i++) {
+                if (confinements[i] == -1) {
+                    newLoras.push(loras[i]);
+                    newWeights.push(weights[i]);
+                }
+            }
+            metadata.loras = newLoras;
+            metadata.loraweights = newWeights;
+            delete metadata.lorasectionconfinement;
+        }
+    }
     let exclude = getUserSetting('reuseparamexcludelist').split(',').map(s => cleanParamName(s));
     resetParamsToDefault(exclude);
     for (let param of gen_param_types) {