Update app.py

app.py

@@ -262,58 +262,65 @@ with gr.Blocks(css="""
                     visible=True
                 )
     
-    # Track evaluation state and preview state
+    # Track evaluation state 
     evaluation_state = gr.State({"running": False})
-    preview_state = gr.State({"visible": False})
+    
+    # Track preview visibility state
+    preview_visibility = gr.State(False)
     
     # Function to show/hide configuration based on selected dataset
-    def update_interface_based_on_dataset(dataset):
+    def update_interface_based_on_dataset(dataset, current_visibility):
         if dataset == "MMLU-Pro":
             return (
                 gr.update(visible=True),  # mmlu_config_container
                 gr.update(visible=True),  # results_container
-                gr.update(interactive=True)  # preview_toggle
+                gr.update(interactive=True),  # preview_toggle
+                gr.update(visible=False),  # dataset_preview_container - hide it initially
+                False,  # Reset preview_visibility to False
+                gr.update(value="Show Preview")  # Reset button text
             )
         else:
             return (
                 gr.update(visible=False),  # mmlu_config_container
                 gr.update(visible=False),  # results_container
-                gr.update(interactive=False)  # preview_toggle
+                gr.update(interactive=False),  # preview_toggle
+                gr.update(visible=False),  # dataset_preview_container - hide when no dataset
+                False,  # Reset preview_visibility to False
+                gr.update(value="Show Preview")  # Reset button text
             )
     
     # Connect dataset dropdown to show/hide appropriate configuration
     dataset_dropdown.change(
         fn=update_interface_based_on_dataset,
-        inputs=[dataset_dropdown],
-        outputs=[mmlu_config_container, results_container, preview_toggle]
+        inputs=[dataset_dropdown, preview_visibility],
+        outputs=[mmlu_config_container, results_container, preview_toggle, dataset_preview_container, preview_visibility, preview_toggle]
     )
     
     # Function to toggle dataset preview visibility
-    def toggle_preview(state, dataset):
-        # Toggle visibility state
-        new_visible = not state["visible"]
-        state["visible"] = new_visible
+    def toggle_preview(dataset, preview_visibility):
+        # Toggle the visibility state
+        is_visible = not preview_visibility
+        
+        # Update button text based on new state
+        button_text = "Hide Preview" if is_visible else "Show Preview"
         
-        # If becoming visible, get the preview data
-        if new_visible and dataset == "MMLU-Pro":
+        # Get preview data if becoming visible
+        if is_visible and dataset == "MMLU-Pro":
             preview_data = mmlupro_dataset_preview()
             formatted_preview = enhanced_format_preview_for_display(preview_data)
-            button_text = "Hide Preview"
-            return state, gr.update(visible=True), formatted_preview, gr.update(value=button_text)
-        elif new_visible:
+            return is_visible, gr.update(visible=True), formatted_preview, gr.update(value=button_text)
+        elif is_visible:
             # For other datasets (not implemented yet)
-            button_text = "Hide Preview"
-            return state, gr.update(visible=True), None, gr.update(value=button_text)
+            return is_visible, gr.update(visible=True), None, gr.update(value=button_text)
         else:
             # Hiding the preview
-            button_text = "Show Preview"
-            return state, gr.update(visible=False), None, gr.update(value=button_text)
+            return is_visible, gr.update(visible=False), None, gr.update(value=button_text)
     
     # Connect preview toggle to show/hide dataset information
     preview_toggle.click(
         fn=toggle_preview,
-        inputs=[preview_state, dataset_dropdown],
-        outputs=[preview_state, dataset_preview_container, preview_output, preview_toggle]
+        inputs=[dataset_dropdown, preview_visibility],
+        outputs=[preview_visibility, dataset_preview_container, preview_output, preview_toggle]
     )
     
     # Update num_subjects_slider interactivity based on all_subjects checkbox