update

Author: RodrigoSKohl

__init__.py

@@ -11,4 +11,6 @@
 }
 
 
-print("\033[34mComfyUI Custom Nodes: \033[92mLoaded Image Stitching Node\033[0m")
+print("\033[34mComfyUI Custom Nodes: \033[92mLoaded Image Stitching Node\033[0m")
+
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']

nodes/nodes.py

@@ -25,7 +25,7 @@ def INPUT_TYPES(cls):
     def stitch_images(self, images, device):
         # Verifica se recebeu pelo menos duas imagens
         if len(images) < 2:
-            raise ValueError("São necessárias pelo menos duas imagens para o stitching.")
+            raise ValueError("At least two images are required for stitching.")
 
         # Verifica se o dispositivo especificado está disponível
         device = torch.device(device if torch.cuda.is_available() and "cuda" in device else "cpu")
@@ -42,7 +42,7 @@ def stitch_images(self, images, device):
 
         # Verifica se o stitching foi bem-sucedido
         if status != cv2.Stitcher_OK:
-            raise RuntimeError(f"Erro ao fazer o stitching: {status}")
+            raise RuntimeError(f"Error when stitching: {status}")
 
         # Converte a imagem resultante para um tensor que o ComfyUI pode usar
         pano_pil = Image.fromarray(cv2.cvtColor(pano, cv2.COLOR_BGR2RGB))