Update utils/state_management.py

utils/state_management.py

@@ -1,12 +1,11 @@
 import gradio as gr
 
-def start_evaluation(state, ui_components):
+def start_evaluation(state):
     """
     Disables UI components when evaluation starts.
     
     Args:
         state (dict): Current evaluation state.
-        ui_components (dict): Dictionary of UI components to update.
     
     Returns:
         list: List of updated state and UI components.
@@ -14,18 +13,25 @@ def start_evaluation(state, ui_components):
     if state["running"]:
         return [
             state,
-            *[gr.update(interactive=False) for _ in range(8)],  # 8 UI components to disable
-            gr.update(visible=True),  # cancel_button
+            gr.update(interactive=False),  # subject_selection_mode
+            gr.update(interactive=False),  # num_subjects_slider
+            gr.update(interactive=False),  # specific_subjects
+            gr.update(interactive=False),  # all_questions_checkbox
+            gr.update(interactive=False),  # num_questions_slider
+            gr.update(interactive=False),  # model1_dropdown
+            gr.update(interactive=False),  # model2_dropdown
+            gr.update(interactive=False),  # eval_button
+            gr.update(visible=True),       # cancel_button
             "Evaluation already in progress. Please wait.",  # results_output
-            None,  # results_table
-            gr.update(visible=False)  # results_table_container
+            None,                          # results_table
+            gr.update(visible=False)       # results_table_container
         ]
     
     # Update state to running
     state["running"] = True
     
-    # Create updates for UI components
-    updates = [
+    # Return updates for UI components
+    return [
         state,  # Updated state
         gr.update(interactive=False),  # subject_selection_mode
         gr.update(interactive=False),  # num_subjects_slider
@@ -40,8 +46,6 @@ def start_evaluation(state, ui_components):
         None,                          # results_table
         gr.update(visible=False)       # results_table_container
     ]
-    
-    return updates
 
 def finish_evaluation(state):
     """
@@ -56,13 +60,12 @@ def finish_evaluation(state):
     state["running"] = False
     return state
 
-def cancel_evaluation(state, ui_components):
+def cancel_evaluation(state):
     """
     Re-enables UI components when evaluation is canceled.
     
     Args:
         state (dict): Current evaluation state.
-        ui_components (dict): Dictionary of UI components to update.
     
     Returns:
         list: List of updated state and UI components.
@@ -70,8 +73,8 @@ def cancel_evaluation(state, ui_components):
     # Update state to not running
     state["running"] = False
     
-    # Create updates for UI components
-    updates = [
+    # Return updates for UI components
+    return [
         state,  # Updated state
         gr.update(interactive=True),  # subject_selection_mode
         gr.update(interactive=True),  # num_subjects_slider
@@ -86,16 +89,13 @@ def cancel_evaluation(state, ui_components):
         None,                         # results_table
         gr.update(visible=False)      # results_table_container
     ]
-    
-    return updates
 
-def handle_evaluation_results(eval_results, ui_components):
+def handle_evaluation_results(eval_results):
     """
     Updates UI components based on evaluation results.
     
     Args:
         eval_results (dict): Results from evaluation.
-        ui_components (dict): Dictionary of UI components to update.
     
     Returns:
         list: List of updated UI components.
@@ -125,4 +125,4 @@ def handle_evaluation_results(eval_results, ui_components):
             gr.update(interactive=True),  # num_questions_slider
             gr.update(interactive=True),  # model1_dropdown
             gr.update(visible=False)  # results_table_container
-        ]
+        ]