Enhance ball rings with neon glow effect (additive blending)

app.py

@@ -577,6 +577,7 @@ class AppState:
         self.fx_ring_thickness: float = BALL_RING_THICKNESS_PX
         self.fx_ring_alpha: float = BALL_RING_ALPHA
         self.fx_ring_feather: float = BALL_RING_FEATHER_SIGMA
+        self.fx_ring_gamma: float = BALL_RING_INTENSITY_GAMMA
 
     def __repr__(self):
         return f"AppState(video_frames={self.video_frames}, inference_session={self.inference_session is not None}, model={self.model is not None}, processor={self.processor is not None}, device={self.device}, dtype={self.dtype}, video_fps={self.video_fps}, masks_by_frame={self.masks_by_frame}, color_by_obj={self.color_by_obj}, clicks_by_frame_obj={self.clicks_by_frame_obj}, boxes_by_frame_obj={self.boxes_by_frame_obj}, composited_frames={self.composited_frames}, current_frame_idx={self.current_frame_idx}, current_obj_id={self.current_obj_id}, current_label={self.current_label}, current_clear_old={self.current_clear_old}, current_prompt_type={self.current_prompt_type}, pending_box_start={self.pending_box_start}, pending_box_start_frame_idx={self.pending_box_start_frame_idx}, pending_box_start_obj_id={self.pending_box_start_obj_id}, is_switching_model={self.is_switching_model}, model_repo_key={self.model_repo_key}, model_repo_id={self.model_repo_id}, session_repo_id={self.session_repo_id})"
@@ -1108,6 +1109,7 @@ def _update_fx_controls(
     ring_thickness: float,
     ring_alpha: float,
     ring_feather: float,
+    ring_gamma: float,
 ) -> Image.Image:
     if state is None:
         return None
@@ -1129,6 +1131,7 @@ def _update_fx_controls(
     state.fx_ring_thickness = max(1.0, float(ring_thickness))
     state.fx_ring_alpha = max(0.0, float(ring_alpha))
     state.fx_ring_feather = max(0.0, float(ring_feather))
+    state.fx_ring_gamma = max(0.1, float(ring_gamma))
 
     state.composited_frames.clear()
     idx = int(getattr(state, "current_frame_idx", 0))
@@ -1285,8 +1288,9 @@ def _build_ball_ring_mask(state: AppState, frame_idx: int) -> np.ndarray | None:
     max_val = float(ring_mask.max())
     if max_val > FX_EPS:
         ring_mask = ring_mask / max_val
-        if abs(BALL_RING_INTENSITY_GAMMA - 1.0) > 1e-6:
-            ring_mask = np.power(np.clip(ring_mask, 0.0, 1.0), BALL_RING_INTENSITY_GAMMA)
+        gamma_val = getattr(state, "fx_ring_gamma", BALL_RING_INTENSITY_GAMMA)
+        if abs(gamma_val - 1.0) > 1e-6:
+            ring_mask = np.power(np.clip(ring_mask, 0.0, 1.0), gamma_val)
     return np.clip(ring_mask, 0.0, 1.0)
 
 
@@ -2840,7 +2844,7 @@ with gr.Blocks(title="SAM2 Video (Transformers) - Interactive Segmentation", the
         with gr.Column(scale=1):
             ball_ring_chk = gr.Checkbox(
                 label="Ball rings (future)",
-                value=False,
+                value=True,
                 info="Replace the ghost trail fill with magenta rings at future ball positions.",
             )
         with gr.Column(scale=1):
@@ -2954,7 +2958,15 @@ with gr.Blocks(title="SAM2 Video (Transformers) - Interactive Segmentation", the
                     minimum=0.0,
                     maximum=5.0,
                     step=0.1,
-                    value=1.5, # Updated default from 0.75->1.5
+                    value=1.5,
+                )
+            with gr.Column(scale=1):
+                ring_gamma = gr.Slider(
+                    label="Ring Gamma (Contrast)",
+                    minimum=0.1,
+                    maximum=2.0,
+                    step=0.05,
+                    value=0.65, # Default gamma
                 )
     
     with gr.Row():
@@ -2978,6 +2990,7 @@ with gr.Blocks(title="SAM2 Video (Transformers) - Interactive Segmentation", the
         ring_thickness,
         ring_alpha,
         ring_feather,
+        ring_gamma,
     ]
 
     for comp in fx_inputs: