From a984f5db38970ad8fbe32dad0ae8cf5b48cbd966 Mon Sep 17 00:00:00 2001
From: chflame <chflame@163.com>
Date: Sun, 12 May 2024 21:02:13 +0800
Subject: [PATCH] update badge info

---
 py/imagefunc.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/py/imagefunc.py b/py/imagefunc.py
index c2df7b9..18eb987 100644
--- a/py/imagefunc.py
+++ b/py/imagefunc.py
@@ -1,6 +1,13 @@
-'''Image process functions for ComfyUI nodes
+"""Image process functions for ComfyUI nodes
 by chflame https://github.com/chflame163
-'''
+"""
+"""
+@author: chflame
+@title: LayerStyle
+@nickname: LayerStyle
+@description: A set of nodes for ComfyUI that can composite layer and mask to achieve Photoshop like functionality.
+"""
+
 import os
 import sys
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
@@ -775,6 +782,9 @@ def histogram_equalization(image:Image, mask:Image=None, gamma_strength=0.5) ->
                     max_brightness = pixel
                 average_brightness += pixel
                 total_pixel += 1
+    if total_pixel == 0:
+        log(f"histogram_equalization: mask is not available, return orinianl image.")
+        return image
     average_brightness = int(average_brightness / total_pixel)
 
     for y in range(image.height):