Update app.py

app.py

@@ -26,12 +26,17 @@ except Exception as e:
 
 def apply_gaussian_blur_background(image, mask, radius):
     """Applies Gaussian blur to the background of the image."""
-    blurred_background = image.filter(ImageFilter.GaussianBlur(radius=radius))
     img_array = np.array(image)
+    
+    # Apply Gaussian blur to the entire image
+    blurred_background = image.filter(ImageFilter.GaussianBlur(radius=radius))
     blurred_array = np.array(blurred_background)
-    foreground_mask = mask > 0
-    foreground_mask_3d = np.stack([foreground_mask] * 3, axis=-1)
+
+    # Ensure the mask is in the same size and format as the image
+    foreground_mask = np.array(mask) > 0
+    foreground_mask_3d = np.stack([foreground_mask] * 3, axis=-1)  # Ensure it's 3 channels (RGB)
     final_image_array = np.where(foreground_mask_3d, img_array, blurred_array)
+
     return Image.fromarray(final_image_array.astype(np.uint8))
 
 def apply_depth_based_blur_background(image, mask, strength):