push app

app.py

@@ -129,8 +129,10 @@ else:
             # Create a red overlay with transparency
             overlay = Image.new("RGBA", img.size, (255, 0, 0, 100))
             base = img.convert("RGBA")
-            mask_for_overlay = combined_mask_img.convert("L")
-            seg_overlay = Image.alpha_composite(overlay, base, mask_for_overlay)
+            mask_alpha = combined_mask_img.point(lambda p: 80 if p > 0 else 0)
+            overlay.putalpha(mask_alpha)
+
+            seg_overlay = Image.alpha_composite(overlay, base, overlay)
         else:
             seg_overlay = img.copy()
     else: