feat: node editor

squashed rebase on main after backendd refactor

invokeai/app/invocations/prompt.py

@@ -2,56 +2,52 @@ from os.path import exists
 from typing import Literal, Optional
 
 import numpy as np
-from pydantic import Field, validator
+from pydantic import validator
 
-from .baseinvocation import BaseInvocation, BaseInvocationOutput, InvocationConfig, InvocationContext
+from .baseinvocation import (
+    BaseInvocation,
+    BaseInvocationOutput,
+    InputField,
+    InvocationContext,
+    OutputField,
+    UIComponent,
+    UITypeHint,
+    title,
+    tags,
+)
 from dynamicprompts.generators import RandomPromptGenerator, CombinatorialPromptGenerator
 
 
 class PromptOutput(BaseInvocationOutput):
     """Base class for invocations that output a prompt"""
 
-    # fmt: off
     type: Literal["prompt"] = "prompt"
 
-    prompt: str = Field(default=None, description="The output prompt")
-    # fmt: on
-
-    class Config:
-        schema_extra = {
-            "required": [
-                "type",
-                "prompt",
-            ]
-        }
+    prompt: str = OutputField(description="The output prompt")
 
 
 class PromptCollectionOutput(BaseInvocationOutput):
     """Base class for invocations that output a collection of prompts"""
 
-    # fmt: off
     type: Literal["prompt_collection_output"] = "prompt_collection_output"
 
-    prompt_collection: list[str] = Field(description="The output prompt collection")
-    count: int = Field(description="The size of the prompt collection")
-    # fmt: on
-
-    class Config:
-        schema_extra = {"required": ["type", "prompt_collection", "count"]}
+    prompt_collection: list[str] = OutputField(
+        description="The output prompt collection", ui_type_hint=UITypeHint.StringCollection
+    )
+    count: int = OutputField(description="The size of the prompt collection")
 
 
+@title("Dynamic Prompt")
+@tags("prompt", "collection")
 class DynamicPromptInvocation(BaseInvocation):
     """Parses a prompt using adieyal/dynamicprompts' random or combinatorial generator"""
 
     type: Literal["dynamic_prompt"] = "dynamic_prompt"
-    prompt: str = Field(description="The prompt to parse with dynamicprompts")
-    max_prompts: int = Field(default=1, description="The number of prompts to generate")
-    combinatorial: bool = Field(default=False, description="Whether to use the combinatorial generator")
 
-    class Config(InvocationConfig):
-        schema_extra = {
-            "ui": {"title": "Dynamic Prompt", "tags": ["prompt", "dynamic"]},
-        }
+    # Inputs
+    prompt: str = InputField(description="The prompt to parse with dynamicprompts", ui_component=UIComponent.Textarea)
+    max_prompts: int = InputField(default=1, description="The number of prompts to generate")
+    combinatorial: bool = InputField(default=False, description="Whether to use the combinatorial generator")
 
     def invoke(self, context: InvocationContext) -> PromptCollectionOutput:
         if self.combinatorial:
@@ -64,24 +60,23 @@ class DynamicPromptInvocation(BaseInvocation):
         return PromptCollectionOutput(prompt_collection=prompts, count=len(prompts))
 
 
+@title("Prompts from File")
+@tags("prompt", "file")
 class PromptsFromFileInvocation(BaseInvocation):
     """Loads prompts from a text file"""
 
-    # fmt: off
-    type: Literal['prompt_from_file'] = 'prompt_from_file'
+    type: Literal["prompt_from_file"] = "prompt_from_file"
 
     # Inputs
-    file_path: str = Field(description="Path to prompt text file")
-    pre_prompt: Optional[str] = Field(description="String to prepend to each prompt")
-    post_prompt: Optional[str] = Field(description="String to append to each prompt")
-    start_line: int = Field(default=1, ge=1, description="Line in the file to start start from")
-    max_prompts: int = Field(default=1, ge=0, description="Max lines to read from file (0=all)")
-    # fmt: on
-
-    class Config(InvocationConfig):
-        schema_extra = {
-            "ui": {"title": "Prompts From File", "tags": ["prompt", "file"]},
-        }
+    file_path: str = InputField(description="Path to prompt text file", ui_type_hint=UITypeHint.FilePath)
+    pre_prompt: Optional[str] = InputField(
+        description="String to prepend to each prompt", ui_component=UIComponent.Textarea
+    )
+    post_prompt: Optional[str] = InputField(
+        description="String to append to each prompt", ui_component=UIComponent.Textarea
+    )
+    start_line: int = InputField(default=1, ge=1, description="Line in the file to start start from")
+    max_prompts: int = InputField(default=1, ge=0, description="Max lines to read from file (0=all)")
 
     @validator("file_path")
     def file_path_exists(cls, v):