Update app.py

app.py

@@ -54,14 +54,19 @@ def process_image(image: Image.Image):
     # two extremes: no blur for near, heavy blur for far
     near_blur = img_np                          
     far_blur  = cv2.GaussianBlur(img_np, (81,81), 20)
-    # invert so 0→near, 1→far
+    
+    # invert so 0 -> near, 1 -> far
     inv_d    = 1.0 - depth_norm
     alpha    = inv_d[...,None]
+    
     # now near (inv_d≈1) stays sharp, far (inv_d≈0) becomes far_blur
     combined = img_np * alpha + far_blur * (1.0 - alpha)
     lens_blur = Image.fromarray(np.clip(combined,0,255).astype(np.uint8))
 
-    return image, output_blur, lens_blur
+    # 6) composite to keep foreground sharp, invert the binary segmentation mask:
+    bg_mask   = Image.fromarray((mask==0).astype(np.uint8)*255)
+    final_lens_blur = Image.composite(lens_blur, image, mask=bg_mask)
+    return image, output_blur, final_lens_blur
 
 iface = gr.Interface(
     fn=process_image,