From f3b293b5ccc09966513e0b8d9f92033ae5a664f6 Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Wed, 23 Aug 2023 23:44:56 +1200
Subject: [PATCH] feat: Add Blank Image Node

---
 invokeai/app/invocations/image.py | 35 ++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/invokeai/app/invocations/image.py b/invokeai/app/invocations/image.py
index f4a1648196..36157e195a 100644
--- a/invokeai/app/invocations/image.py
+++ b/invokeai/app/invocations/image.py
@@ -8,7 +8,7 @@ import numpy
 from PIL import Image, ImageChops, ImageFilter, ImageOps
 
 from invokeai.app.invocations.metadata import CoreMetadata
-from invokeai.app.invocations.primitives import ImageField, ImageOutput
+from invokeai.app.invocations.primitives import ColorField, ImageField, ImageOutput
 from invokeai.backend.image_util.invisible_watermark import InvisibleWatermark
 from invokeai.backend.image_util.safety_checker import SafetyChecker
 
@@ -41,6 +41,39 @@ class ShowImageInvocation(BaseInvocation):
         )
 
 
+@title("Blank Image")
+@tags("image")
+class BlankImageInvocation(BaseInvocation):
+    """Creates a blank image and forwards it to the pipeline"""
+
+    # Metadata
+    type: Literal["blank_image"] = "blank_image"
+
+    # Inputs
+    width: int = InputField(default=512, description="The width of the image")
+    height: int = InputField(default=512, description="The height of the image")
+    mode: Literal["RGB", "RGBA"] = InputField(default="RGB", description="The mode of the image")
+    color: ColorField = InputField(default=ColorField(r=0, g=0, b=0, a=255), description="The color of the image")
+
+    def invoke(self, context: InvocationContext) -> ImageOutput:
+        image = Image.new(mode=self.mode, size=(self.width, self.height), color=self.color.tuple())
+
+        image_dto = context.services.images.create(
+            image=image,
+            image_origin=ResourceOrigin.INTERNAL,
+            image_category=ImageCategory.GENERAL,
+            node_id=self.id,
+            session_id=context.graph_execution_state_id,
+            is_intermediate=self.is_intermediate,
+        )
+
+        return ImageOutput(
+            image=ImageField(image_name=image_dto.image_name),
+            width=image_dto.width,
+            height=image_dto.height,
+        )
+
+
 @title("Crop Image")
 @tags("image", "crop")
 class ImageCropInvocation(BaseInvocation):