updates

app.py

@@ -224,7 +224,6 @@ class VideoAnnotator:
             # --- End Edit ---
 
 def create_interface():
-    # Move the VideoAnnotator creation inside a function that will be called for each session
     def get_session_annotator():
         annotator = VideoAnnotator()
         success = annotator.load_videos_from_hf()
@@ -279,11 +278,15 @@ def create_interface():
 
                 save_btn = gr.Button("Save Annotations", variant="primary")
 
-        # Add this after all components are defined but before the event handlers
+        # Modify the load event handler to properly initialize and return all outputs
         demo.load(
-            fn=lambda annotator: annotator.get_current_video(),
+            fn=lambda annotator: (
+                annotator.get_current_video(),  # video player
+                *[None] * len(ANNOTATION_CATEGORIES),  # annotation components
+                ""  # status display
+            ),
             inputs=[annotator_state],
-            outputs=video_player
+            outputs=[video_player] + annotation_components + [status_display]
         )
 
         # Modify the event handlers to use the session-specific annotator