From aeafbe2113024d8d03150e489e7d096e7bbc3470 Mon Sep 17 00:00:00 2001
From: Albert Trott <trott.alik@gmail.com>
Date: Tue, 30 Jul 2024 09:07:03 +0200
Subject: [PATCH] feat: rename opacity

---
 __init__.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/__init__.py b/__init__.py
index cd70fe7..e292eaa 100644
--- a/__init__.py
+++ b/__init__.py
@@ -12,10 +12,11 @@ class ImageBlender:
             "required": {
                 "base_image": ("IMAGE",),
                 "blend_image": ("IMAGE",),
-                "opacity": ("FLOAT", {
+                "strength": ("FLOAT", {
                     "default": 1,
                     "min": 0.0,
-                    "max": 1.0
+                    "max": 1.0,
+                    "step": 0.01
                 }),
                 "blend_mode": (
                     [mode.value for mode in BlendModes],
@@ -31,7 +32,7 @@ class ImageBlender:
     FUNCTION = "blend"
     CATEGORY = "ImageBlender"
 
-    def blend(self, base_image: torch.Tensor, blend_image: torch.Tensor, opacity: float, blend_mode: str, mask: torch.Tensor = None) -> tuple:
+    def blend(self, base_image: torch.Tensor, blend_image: torch.Tensor, strength: float, blend_mode: str, mask: torch.Tensor = None) -> tuple:
         blend_function = self.blend_functions.get(BlendModes(blend_mode), lambda x, y: x)
         result = blend_function(base_image, blend_image)
 
@@ -46,7 +47,7 @@ class ImageBlender:
                 result = result * mask + base_image * (1 - mask)
 
         # Apply opacity
-        result = base_image * opacity + result * (1 - opacity)
+        result = result * strength + base_image * (1 - strength)
         # Normalize the result
         result = torch.clamp(result, 0, 1)
         return (result,)