Enhance ball rings with neon glow effect (additive blending)

app.py

@@ -969,6 +969,7 @@ def compose_frame(state: AppState, frame_idx: int, remove_bg: bool = False) -> I
 
     current_union_mask: np.ndarray | None = None
     focus_mask: np.ndarray | None = None
+    ball_mask_main: np.ndarray | None = None
 
     for obj_id, mask in masks.items():
         if mask is None:
@@ -984,6 +985,8 @@ def compose_frame(state: AppState, frame_idx: int, remove_bg: bool = False) -> I
             if focus_mask is None:
                 focus_mask = np.zeros_like(mask_np, dtype=np.float32)
             focus_mask = np.maximum(focus_mask, mask_np)
+        if obj_id == BALL_OBJECT_ID:
+            ball_mask_main = mask
 
     ghost_mask = _build_ball_trail_mask(state, frame_idx)
     ring_result = _build_ball_ring_mask(state, frame_idx)
@@ -1006,11 +1009,8 @@ def compose_frame(state: AppState, frame_idx: int, remove_bg: bool = False) -> I
             result_np = _apply_cutout_fx(state, frame_np, combined_mask)
             out_img = Image.fromarray(result_np)
         else:
-            overlay_masks = masks
-            if (ghost_mask is not None or ring_result is not None) and BALL_OBJECT_ID in masks:
-                overlay_masks = {oid: mask for oid, mask in masks.items() if oid != BALL_OBJECT_ID}
-            if overlay_masks:
-                out_img = overlay_masks_on_frame(out_img, overlay_masks, state.color_by_obj, alpha=0.65)
+            if masks:
+                out_img = overlay_masks_on_frame(out_img, masks, state.color_by_obj, alpha=0.65)
             # Overlay feathered ball on top
             if BALL_OBJECT_ID in masks:
                 ball_mask = masks[BALL_OBJECT_ID]
@@ -1044,7 +1044,12 @@ def compose_frame(state: AppState, frame_idx: int, remove_bg: bool = False) -> I
         ring_presence = np.clip(ring_presence.astype(np.float32), 0.0, 1.0)
         ring_color_map = np.clip(ring_color_map.astype(np.float32), 0.0, 1.0)
         if current_union_mask is not None:
-            mask_keep = np.clip(1.0 - current_union_mask, 0.0, 1.0)
+            if ball_mask_main is not None:
+                ball_np = np.clip(ball_mask_main.astype(np.float32), 0.0, 1.0)
+                mask_block = np.maximum(current_union_mask - ball_np, 0.0)
+            else:
+                mask_block = current_union_mask
+            mask_keep = np.clip(1.0 - mask_block, 0.0, 1.0)
             ring_presence = ring_presence * mask_keep
             ring_color_map = ring_color_map * mask_keep[..., None]
         if ring_presence.max() > FX_EPS and ring_color_map.max() > FX_EPS: