minor improvement on lora segment confinement handling

Author: mcmonkey4eva

src/Text2Image/T2IParamInput.cs

@@ -356,38 +356,43 @@ static T2IParamInput()
             }
             context.Loras ??= [.. Program.T2IModelSets["LoRA"].ListModelNamesFor(context.Input.SourceSession)];
             string matched = T2IParamTypes.GetBestModelInList(lora, context.Loras);
-            if (matched is not null)
+            if (matched is null)
             {
-                List<string> loraList = context.Input.Get(T2IParamTypes.Loras);
-                List<string> weights = context.Input.Get(T2IParamTypes.LoraWeights);
-                List<string> confinements = context.Input.Get(T2IParamTypes.LoraSectionConfinement);
-                if (loraList is null)
-                {
-                    loraList = [];
-                    weights = [];
-                }
-                loraList.Add(matched);
-                weights.Add(strength.ToString());
-                context.Input.Set(T2IParamTypes.Loras, loraList);
-                context.Input.Set(T2IParamTypes.LoraWeights, weights);
-                if (context.SectionID > 0)
+                Logs.Warning($"Lora '{lora}' does not exist and will be ignored.");
+                return null;
+            }
+            List<string> loraList = context.Input.Get(T2IParamTypes.Loras);
+            List<string> weights = context.Input.Get(T2IParamTypes.LoraWeights);
+            List<string> confinements = context.Input.Get(T2IParamTypes.LoraSectionConfinement);
+            if (loraList is null)
+            {
+                loraList = [];
+                weights = [];
+            }
+            if (confinements is not null && confinements.Count > loraList.Count)
+            {
+                context.Input.Remove(T2IParamTypes.LoraSectionConfinement);
+                confinements = null;
+            }
+            loraList.Add(matched);
+            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("0");
                     }
-                    Logs.Verbose($"LoRA {lora} confined to section {context.SectionID}.");
-                    confinements.Add($"{context.SectionID}");
-                    context.Input.Set(T2IParamTypes.LoraSectionConfinement, confinements);
                 }
-                return "";
+                Logs.Verbose($"LoRA {lora} confined to section {context.SectionID}.");
+                confinements.Add($"{context.SectionID}");
+                context.Input.Set(T2IParamTypes.LoraSectionConfinement, confinements);
             }
-            Logs.Warning($"Lora '{lora}' does not exist and will be ignored.");
-            return null;
+            return "";
         };
         PromptTagPostProcessors["segment"] = (data, context) =>
         {