From 5c0ef3e58909a2715b1d6206552b77466689647f Mon Sep 17 00:00:00 2001
From: toto <tototianhao@gmail.com>
Date: Wed, 22 Nov 2023 17:15:09 +0800
Subject: [PATCH] update node name

---
 __init__.py      | 4 ++--
 comfyui/nodes.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/__init__.py b/__init__.py
index 4778be4..70f93d5 100644
--- a/__init__.py
+++ b/__init__.py
@@ -17,7 +17,7 @@ NODE_CLASS_MAPPINGS = {
     "FC_Segment": FCSegment,
     "FC_ReplaceImage": FCReplaceImage,
     "FC_CropBottom": FCCropBottom,
-    "FC_FCCropFace": FCCropFace,
+    "FC_CropFace": FCCropFace,
 }
 NODE_DISPLAY_NAME_MAPPINGS = {
     "FC_FaceFusion": "FC FaceFusion",
@@ -26,7 +26,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "FC_CropMask": "FC CropMask",
     "FC_ReplaceImage": "FC ReplaceImage",
     "FC_CropBottom": "FC CropBottom",
-    "FC_FCCropFace": "FC FCCropFace",
+    "FC_CropFace": "FC CropFace",
 
 }
 __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
diff --git a/comfyui/nodes.py b/comfyui/nodes.py
index c4c6bdf..9d0c507 100644
--- a/comfyui/nodes.py
+++ b/comfyui/nodes.py
@@ -292,7 +292,7 @@ class FCCropFace:
             }
         }
 
-    RETURN_TYPES = ("IMAGE", "BOX")
+    RETURN_TYPES = ("IMAGE", "BOX", "KEY_POINT")
     FUNCTION = "face_crop"
     CATEGORY = "facechain/crop"
 
@@ -324,4 +324,4 @@ class FCCropFace:
         bbox[3] = np.clip(np.array(bbox[3], np.int32) + face_h * (crop_ratio - 1) / 2, 0, h - 1)
         bbox = np.array(bbox, np.int32)
         result_image = source_image[:, bbox[3]:bbox[1], bbox[0]:bbox[2], :]
-        return result_image, bbox
+        return result_image, bbox , points_array