fix process image

utils/helpers.py

@@ -15,20 +15,29 @@ def calculate_mask_area(mask: Image.Image, background=False) -> int:
     return non_zero_pixels
 
 def process_image(input_image: Image.Image) -> Image.Image:
-    if input_image.mode != 'RGBA':
-        input_image = input_image.convert('RGBA')
+
 
     data = np.array(input_image)
     # Split the image into its component channels
-    r, g, b, a = data.T
 
     # Create a mask where all pixels that are black (0,0,0) will have 0 alpha
-    black_areas = (r == 0) & (g == 0) & (b == 0)
+    black_areas = data == 0
+    
+    rgba_image = Image.new('RGBA', input_image.size)
+    rgba_data = np.array(rgba_image)
+
+    # Copy the grayscale data to all RGB channels
+    rgba_data[..., 0] = data
+    rgba_data[..., 1] = data
+    rgba_data[..., 2] = data
+
+    # Set alpha channel to 255 (fully opaque) for all pixels
+    rgba_data[..., 3] = 255
 
     # Apply the mask to the alpha channel
-    data[..., 3][black_areas] = 0
+    rgba_data[..., 3][black_areas] = 0
 
-    return Image.fromarray(data)
+    return Image.fromarray(rgba_data)
 
 def combine_images(original_image: Image.Image, masks: list) -> Image.Image:
     combined = original_image.copy()