From da15e5936e60a96b91b87778e391c52b6961db7c Mon Sep 17 00:00:00 2001
From: Patrick Morgan <patrick@patrick-morgan.net>
Date: Sun, 15 Dec 2024 22:00:13 -0800
Subject: [PATCH] stake

---
 nodes/nodes.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/nodes/nodes.py b/nodes/nodes.py
index f4042b7..d9cf79a 100644
--- a/nodes/nodes.py
+++ b/nodes/nodes.py
@@ -1,6 +1,6 @@
 import torch
 import numpy as np
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, ImageFilter
 
 class MaskContourProcessor:
     """A class that adds decorative flame-like contour effects to binary masks.
@@ -410,7 +410,11 @@ class MaskContourProcessor:
 
         combined_mask = np.clip(mask_np + output_mask, 0, 1)
 
-        return (torch.from_numpy(combined_mask),)
+        # Apply Gaussian blur
+        blurred_image = Image.fromarray(combined_mask).filter(ImageFilter.GaussianBlur(10))
+        blurred_mask = np.array(blurred_image)
+
+        return (torch.from_numpy(blurred_mask),)
 
 # Node class mappings
 NODE_CLASS_MAPPINGS = {