fixing batching for loading data

components/review_dashboard_page.py

@@ -479,10 +479,17 @@ class ReviewDashboardPage:
                 gr.update(value="❌ Reject")  # Reset reject button text
             )
 
+        def update_review_info_fn(items, total_count):
+            """Update the review info banner with current loaded items count"""
+            if items:
+                return f"🔍 **Phase 2 Review Mode** - Reviewing assigned annotations. Loaded {len(items)} of {total_count} total items."
+            else:
+                return f"🔍 **Phase 2 Review Mode** - No annotations found for review."
+
         def navigate_and_load_fn(items, current_idx, direction, session):
             """Combined navigation and loading function"""
             if not items:
-                return items, 0
+                return items, 0, ""
             
             # Navigate
             if direction == "next":
@@ -492,13 +499,15 @@ class ReviewDashboardPage:
                 if should_load_more:
                     log.info(f"User reached end of loaded items ({new_idx}/{len(items)}), will load more items")
                     # Load more items
-                    updated_items = load_more_items_fn(items, session, current_batch_size=10)
-                    return updated_items, new_idx
+                    updated_items, total_count = load_more_items_fn(items, session, current_batch_size=10)
+                    # Update review info with new count
+                    review_info = update_review_info_fn(updated_items, total_count)
+                    return updated_items, new_idx, review_info
                 else:
-                    return items, new_idx
+                    return items, new_idx, ""  # No review info update needed
             else:  # prev
                 new_idx = max(current_idx - 1, 0)
-                return items, new_idx
+                return items, new_idx, ""  # No review info update needed
 
         def save_validation_fn(items, idx, session, approved: bool, rejection_reason: str = ""):
             if not items or idx >= len(items):
@@ -615,7 +624,7 @@ class ReviewDashboardPage:
             username = session.get("username")
             
             if not user_id or not username:
-                return items  # Return existing items if no user session
+                return items, 0  # Return existing items if no user session
                 
             # Find target annotator
             target_annotator = None
@@ -625,12 +634,17 @@ class ReviewDashboardPage:
                     break
                     
             if not target_annotator:
-                return items
+                return items, 0
                 
             with get_db() as db:
                 target_annotator_obj = db.query(Annotator).filter_by(name=target_annotator).first()
                 if not target_annotator_obj:
-                    return items
+                    return items, 0
+                    
+                # Get total count for updated review info
+                total_count = db.query(Annotation).filter(
+                    Annotation.annotator_id == target_annotator_obj.id
+                ).count()
                     
                 # Load next batch starting from where we left off
                 offset = len(items)
@@ -670,7 +684,7 @@ class ReviewDashboardPage:
                 # Combine with existing items
                 all_items = items + new_items
                 log.info(f"Loaded {len(new_items)} more items, total now: {len(all_items)}")
-                return all_items
+                return all_items, total_count
 
         # Output definitions
         review_display_outputs = [
@@ -715,7 +729,7 @@ class ReviewDashboardPage:
             ).then(
                 fn=lambda items, idx, session, dir=direction: navigate_and_load_fn(items, idx, dir, session),
                 inputs=[self.items_state, self.idx_state, session_state],
-                outputs=[self.items_state, self.idx_state]
+                outputs=[self.items_state, self.idx_state, self.review_info]
             ).then(
                 fn=show_current_review_item_fn,
                 inputs=[self.items_state, self.idx_state, session_state],
@@ -751,7 +765,7 @@ class ReviewDashboardPage:
         ).then(
             fn=lambda items, idx, session: navigate_and_load_fn(items, idx, "next", session),
             inputs=[self.items_state, self.idx_state, session_state],
-            outputs=[self.items_state, self.idx_state]
+            outputs=[self.items_state, self.idx_state, self.review_info]
         ).then(
             fn=show_current_review_item_fn,
             inputs=[self.items_state, self.idx_state, session_state],
@@ -772,9 +786,9 @@ class ReviewDashboardPage:
             inputs=[self.items_state, self.idx_state, session_state, self.rejection_mode_active],
             outputs=[self.header.progress_display]
         ).then(
-            fn=lambda items, idx, session, rejection_mode: navigate_and_load_fn(items, idx, "next", session) if not rejection_mode else (items, idx),
+            fn=lambda items, idx, session, rejection_mode: navigate_and_load_fn(items, idx, "next", session) if not rejection_mode else (items, idx, ""),
             inputs=[self.items_state, self.idx_state, session_state, self.rejection_mode_active],
-            outputs=[self.items_state, self.idx_state]
+            outputs=[self.items_state, self.idx_state, self.review_info]
         ).then(
             fn=lambda items, idx, session, rejection_mode: show_current_review_item_fn(items, idx, session) if not rejection_mode else (
                 str(items[idx]["tts_id"]) if items and idx < len(items) else "",
@@ -802,7 +816,7 @@ class ReviewDashboardPage:
         self.btn_skip.click(
             fn=lambda items, idx, session: navigate_and_load_fn(items, idx, "next", session),
             inputs=[self.items_state, self.idx_state, session_state],
-            outputs=[self.items_state, self.idx_state]
+            outputs=[self.items_state, self.idx_state, self.review_info]
         ).then(
             fn=show_current_review_item_fn,
             inputs=[self.items_state, self.idx_state, session_state],