Update app.py

app.py

@@ -131,8 +131,8 @@ def create_arena_interface():
         )
 
         # State variables
-        current_video = gr.State("")
-        current_models = gr.State([])
+        current_video = gr.State()
+        current_models = gr.State()
 
         with gr.Row():
             # Display two videos side by side
@@ -163,18 +163,25 @@ def create_arena_interface():
             video_left.label = model_pair[0]["name"]
             video_right.label = model_pair[1]["name"]
 
-            return {
-                video_left: video_paths[0],
-                video_right: video_paths[1],
-                current_video: video_name,
-                current_models: [model["name"] for model in model_pair],
-                rankings_table: manager.get_rankings(),
-            }
+            current_video_value = video_name
+            current_models_value = [model["name"] for model in model_pair]
 
-        def handle_choice(choice_index: int, video_name: str, current_models: List[str]) -> Dict:
+            return (
+                video_paths[0],  # video_left
+                video_paths[1],  # video_right
+                current_video_value,  # current_video
+                current_models_value,  # current_models
+                manager.get_rankings(),  # rankings_table
+            )
+
+        def handle_choice(choice_index: int, video_name: str, current_models_value: List[str]):
             """Handle the user's choice and update rankings."""
-            winner = current_models[choice_index]
-            loser = current_models[1 - choice_index]
+            if not current_models_value or len(current_models_value) < 2:
+                print("Error: current_models is invalid:", current_models_value)
+                return gr.update(visible=False), gr.update(visible=False), "", [], manager.get_rankings()
+
+            winner = current_models_value[choice_index]
+            loser = current_models_value[1 - choice_index]
 
             # Save the comparison result
             manager.save_comparison(video_name, winner, loser)
@@ -210,7 +217,7 @@ def create_arena_interface():
         # Initial load
         demo.load(
             fn=load_new_comparison,
-            inputs=None,
+            inputs=[],
             outputs=[
                 video_left,
                 video_right,