Add missing type annotation

app.py

@@ -1017,7 +1017,7 @@ with gr.Blocks(title="SAM3", theme=Soft(primary_hue="blue", secondary_hue="rose"
         show_progress=True,
     )
 
-    def _sync_frame_idx_pointbox(state_in: AppState, idx: int):
+    def _sync_frame_idx_pointbox(state_in: AppState, idx: int) -> Image.Image:
         if state_in is not None:
             state_in.current_frame_idx = int(idx)
         return update_frame_display(state_in, int(idx))
@@ -1035,7 +1035,7 @@ with gr.Blocks(title="SAM3", theme=Soft(primary_hue="blue", secondary_hue="rose"
         show_progress=True,
     )
 
-    def _sync_frame_idx_text(state_in: AppState, idx: int):
+    def _sync_frame_idx_text(state_in: AppState, idx: int) -> Image.Image:
         if state_in is not None:
             state_in.current_frame_idx = int(idx)
         return update_frame_display(state_in, int(idx))
@@ -1046,28 +1046,33 @@ with gr.Blocks(title="SAM3", theme=Soft(primary_hue="blue", secondary_hue="rose"
         outputs=preview_text,
     )
 
-    def _sync_obj_id(s: AppState, oid):
+    def _sync_obj_id(s: AppState, oid) -> None:
         if s is not None and oid is not None:
             s.current_obj_id = int(oid)
 
-    obj_id_inp.change(fn=_sync_obj_id, inputs=[app_state, obj_id_inp])
+    obj_id_inp.change(
+        fn=_sync_obj_id,
+        inputs=[app_state, obj_id_inp],
+    )
 
-    def _sync_label(s: AppState, lab: str):
+    def _sync_label(s: AppState, lab: str) -> None:
         if s is not None and lab is not None:
             s.current_label = str(lab)
 
-    label_radio.change(fn=_sync_label, inputs=[app_state, label_radio])
+    label_radio.change(
+        fn=_sync_label,
+        inputs=[app_state, label_radio],
+    )
 
-    def _sync_prompt_type(s: AppState, val: str):
+    def _sync_prompt_type(s: AppState, val: str) -> tuple[dict, dict]:
         if s is not None and val is not None:
             s.current_prompt_type = str(val)
             s.pending_box_start = None
         is_points = str(val).lower() == "points"
-        updates = [
+        return (
             gr.update(visible=is_points),
             gr.update(interactive=is_points) if is_points else gr.update(value=True, interactive=False),
-        ]
-        return updates
+        )
 
     prompt_type.change(
         fn=_sync_prompt_type,
@@ -1081,7 +1086,7 @@ with gr.Blocks(title="SAM3", theme=Soft(primary_hue="blue", secondary_hue="rose"
         outputs=preview_pointbox,
     )
 
-    def _on_text_apply(state: AppState, frame_idx: int, text: str):
+    def _on_text_apply(state: AppState, frame_idx: int, text: str) -> tuple[Image.Image, str, str]:
         img, status, active_prompts = on_text_prompt(state, frame_idx, text)
         return img, status, active_prompts
 
@@ -1093,11 +1098,11 @@ with gr.Blocks(title="SAM3", theme=Soft(primary_hue="blue", secondary_hue="rose"
 
     reset_prompts_btn.click(
         fn=reset_prompts,
-        inputs=[app_state],
+        inputs=app_state,
         outputs=[app_state, preview_text, text_status, active_prompts_display],
     )
 
-    def _render_video(s: AppState):
+    def _render_video(s: AppState) -> str:
         if s is None or s.num_frames == 0:
             raise gr.Error("Load a video first.")
         fps = s.video_fps if s.video_fps and s.video_fps > 0 else 12
@@ -1123,12 +1128,20 @@ with gr.Blocks(title="SAM3", theme=Soft(primary_hue="blue", secondary_hue="rose"
             print(f"Failed to render video with cv2: {e}")
             raise gr.Error(f"Failed to render video: {e}")
 
-    render_btn_pointbox.click(_render_video, inputs=[app_state], outputs=[playback_video_pointbox])
-    render_btn_text.click(_render_video, inputs=[app_state], outputs=[playback_video_text])
+    render_btn_pointbox.click(
+        fn=_render_video,
+        inputs=app_state,
+        outputs=playback_video_pointbox,
+    )
+    render_btn_text.click(
+        fn=_render_video,
+        inputs=app_state,
+        outputs=playback_video_text,
+    )
 
     propagate_btn_pointbox.click(
         fn=propagate_masks,
-        inputs=[app_state],
+        inputs=app_state,
         outputs=[app_state, propagate_status_pointbox, frame_slider_pointbox],
     )