Ass DarkPopLoraFromStack

__init__.py

@@ -18,6 +18,7 @@
     "DarkCheckpointSwitcher": checkpoints.DarkCheckpointSwitcher,
     "DarkAnyToString": convert.DarkAnyToString,
     "DarkLoraStackFromString": loras.DarkLoraStackFromString,
+    "DarkPopLoraFromStack": loras.DarkPopLoraFromStack,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
@@ -31,6 +32,7 @@
     "DarkCheckpointSwitcher": "Dark Checkpoint Switcher",
     "DarkAnyToString": "Dark Any to String",
     "DarkLoraStackFromString": "Dark LoRA Stack from String",
+    "DarkPopLoraFromStack": "Dark LoRA Pop LoRA string from LORA_STACK",
 }
 
 __all__ = [NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS]

modules/loras.py

@@ -10,6 +10,48 @@
 logger = logging.getLogger(__name__)
 
 
+class DarkPopLoraFromStack(object):
+    """
+    Accepts a LoRA stack and extracts the first LoRA it finds, removing it from the stack
+    """
+
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "LORA_STACK": (
+                    "LORA_STACK",
+                    {
+                        "forceInput": True,
+                    },
+                ),
+            },
+        }
+
+    RETURN_TYPES = (
+        "LORA_STACK",
+        "STRING",
+    )
+    RETURN_NAMES = ("LORA_STACK", "EXTRACTED_LORA")
+    FUNCTION = "extract_lora_from_stack"
+
+    CATEGORY = "DarkPrompt"
+
+    def extract_lora_from_stack(self, LORA_STACK):
+        print(LORA_STACK)
+        popped_lora = ""
+        if LORA_STACK:
+            popped_lora = LORA_STACK.pop(0)
+            popped_lora = popped_lora[0]
+
+        print(popped_lora)
+
+        return (LORA_STACK, popped_lora)
+
+
 class DarkLoraStackFromString(object):
     """
     Takes in a string (prompt), scans it for LoRA tags in the format <lora:somelora:x:y> and creates a LoRA stack from the string