updates

app.py

@@ -58,7 +58,7 @@ class VideoAnnotator:
                 annotated_files = self.api.list_repo_files(
                     repo_id=self.annotation_repo_id,
                     repo_type=self.annotation_repo_type,
-                    path_in_repo="annotations" # Specify the directory
+                    path_in_repo="annotations"
                 )
                 # Extract base video names from annotation filenames
                 # e.g., "annotations/video1.mp4.jsonl" -> "video1.mp4"
@@ -80,12 +80,10 @@ class VideoAnnotator:
             except Exception as e:
                 # Log error and fallback to using all videos if the check fails
                 logger.error(f"Could not list or process annotation files: {e}. Proceeding with all videos, but conflicts may occur.")
-                self.video_files = all_video_files # Fallback: load all if check fails
-            # --- End Edit ---
+                self.video_files = all_video_files 
 
             if not self.video_files:
                 logger.warning("No videos left to annotate!")
-                # Optionally, display a message in the UI here if possible
 
             return len(self.video_files) > 0
         except Exception as e:
@@ -198,9 +196,7 @@ class VideoAnnotator:
             # --- End Edit ---
         else:
             logger.info("Already at the last video")
-            # --- Start Edit: Return only the save status as the last element ---
             return self.get_current_video(), *[None] * len(ANNOTATION_CATEGORIES), "Already at the last video. " + save_status
-            # --- End Edit ---
 
     def prev_video(self, *current_annotations):
         # Save current annotations before moving to previous video
@@ -233,17 +229,14 @@ def create_interface():
     
     if not success:
         logger.error("Failed to load videos. Interface might not function correctly.")
-        # Handle the error appropriately, maybe show a message in the UI
     
     with gr.Blocks() as demo:
         gr.Markdown("# Cricket Video Annotation Tool")
         
-        # --- Start Edit: Define progress update function ---
         total_categories = len(ANNOTATION_CATEGORIES)
         def update_progress(*annotation_values):
             filled_count = sum(1 for val in annotation_values if val is not None)
             return f"**Progress:** {filled_count} / {total_categories} categories selected"
-        # --- End Edit ---
 
         with gr.Row(): # Main row to hold video and controls side-by-side
             with gr.Column(scale=2): # Column for Video Player and Nav Buttons
@@ -252,9 +245,7 @@ def create_interface():
                     label="Current Video",
                     height=350
                 )
-                # --- Start Edit: Rename status_textbox to status_display ---
                 status_display = gr.Textbox(label="Status", interactive=False) # For save/nav messages
-                # --- End Edit ---
                 with gr.Row():
                     prev_btn = gr.Button("Previous Video")
                     next_btn = gr.Button("Next Video")
@@ -287,15 +278,10 @@ def create_interface():
         current_video = annotator.get_current_video()
         if current_video:
             logger.info(f"Setting initial video player value: {current_video}")
-            # The video_player will call annotator.get_current_video on load,
-            # so explicitly setting value might be redundant unless get_current_video changes state.
-            # video_player.value = current_video # Let's rely on the function loader
 
             existing_annotations = annotator.load_existing_annotation()
             if existing_annotations:
                 logger.info(f"Loading existing annotations: {existing_annotations}")
-                # Need to return initial values for components if loading annotations
-                # This part is tricky with function loading. Let's adjust.
 
                 # We might need a separate function to load initial state
                 # Or adjust how initial values are set.
@@ -307,32 +293,25 @@ def create_interface():
         save_btn.click(
             fn=annotator.save_annotation,
             inputs=annotation_components,
-            # --- Start Edit: Output to status_display ---
             outputs=status_display
-            # --- End Edit ---
         )
 
         next_btn.click(
             fn=annotator.next_video,
             inputs=annotation_components,
-            # --- Start Edit: Output status to status_display. Progress updates via radio changes. ---
             # Outputs: video, clear all radios, update status_display
             outputs=[video_player] + annotation_components + [status_display]
-            # --- End Edit ---
         )
 
         prev_btn.click(
             fn=annotator.prev_video,
             inputs=annotation_components,
-            # --- Start Edit: Output status to status_display. Progress updates via radio changes. ---
             # Outputs: video, clear all radios, update status_display
             outputs=[video_player] + annotation_components + [status_display]
-            # --- End Edit ---
         )
 
     return demo
 
 if __name__ == "__main__":
     demo = create_interface()
-    # Consider adding share=True for easier testing if needed
     demo.launch(allowed_paths=["/"])