Upload app.py

app.py

@@ -45,19 +45,20 @@ def apply_depth_aware_blur(
 
     blurred_image = np.copy(np.array(original_image))
 
-    if foreground_blur > 0:
+    # Apply blur, ensuring kernel size is valid
+    if foreground_blur > 0 and foreground_blur % 2 == 1:
         blurred_image = np.where(
             (foreground_mask[..., None] == 255),
             cv2.GaussianBlur(blurred_image, (foreground_blur, foreground_blur), 10),
             blurred_image,
         )
-    if midground_blur > 0:
+    if midground_blur > 0 and midground_blur % 2 == 1:
         blurred_image = np.where(
             (midground_mask[..., None] == 255),
             cv2.GaussianBlur(blurred_image, (midground_blur, midground_blur), 8),
             blurred_image,
         )
-    if background_blur > 0:
+    if background_blur > 0 and background_blur % 2 == 1:
         blurred_image = np.where(
             (background_mask[..., None] == 255),
             cv2.GaussianBlur(blurred_image, (background_blur, background_blur), 20),
@@ -71,8 +72,8 @@ def apply_depth_aware_blur(
 example_image = np.zeros((512, 512, 3), dtype=np.uint8)  # Placeholder for an image
 example_inputs = [
     example_image,
-    0,  # foreground_blur
-    0,  # midground_blur
+    15,  # foreground_blur
+    7,  # midground_blur
     35,  # background_blur (default)
     0.2,  # foreground_threshold (default)
     0.2,  # midground_lower (default)
@@ -84,9 +85,9 @@ iface = gr.Interface(
     fn=apply_depth_aware_blur,
     inputs=[
         gr.Image(label="Input Image"),
-        gr.Slider(0, 51, step=2, label="Foreground Blur Kernel Size"),
-        gr.Slider(0, 51, step=2, label="Midground Blur Kernel Size"),
-        gr.Slider(0, 51, step=2, label="Background Blur Kernel Size"),
+        gr.Slider(1, 51, step=2, label="Foreground Blur Kernel Size"),
+        gr.Slider(1, 51, step=2, label="Midground Blur Kernel Size"),
+        gr.Slider(1, 51, step=2, label="Background Blur Kernel Size"),
         gr.Slider(0, 1, label="Foreground Threshold"),
         gr.Slider(0, 1, label="Midground Lower Threshold"),
         gr.Slider(0, 1, label="Midground Upper Threshold"),