test: try fixing variable error

Changed gr.EventData to gr.SelectData for event handling
Updated event data access from evt._data["index"] to evt.index
Added error handling around array indexing
Made event parameters Optional where appropriate
Added try-except blocks around event processing code

app.py

@@ -120,28 +120,37 @@ def get_data(image_name: str, model_name: str) -> np.ndarray:
 def get_activation_distribution(image_name: str, model_type: str) -> np.ndarray:
     """Get activation distribution with memory optimization."""
     try:
-        activation = get_data(image_name, model_type)[0]
+        data = get_data(image_name, model_type)
+        if isinstance(data, (list, tuple)):
+            activation = data[0]
+        else:
+            activation = data
+            
+        if not isinstance(activation, np.ndarray):
+            activation = np.array(activation)
+            
         mean_acts = _CACHE.get('sae_data_dict', "mean_acts", {}).get("imagenet", np.array([]))
         
-        if mean_acts.size > 0:
-            noisy_features_indices = (mean_acts > 0.1).nonzero()[0]
-            activation[:, noisy_features_indices] = 0
+        if mean_acts.size > 0 and activation.size > 0:
+            noisy_features_indices = np.where(mean_acts > 0.1)[0]
+            if activation.ndim >= 2:
+                activation[:, noisy_features_indices] = 0
             
         return activation
     except Exception as e:
         print(f"Error getting activation distribution: {e}")
         return np.array([])
 
-def get_grid_loc(evt: gr.EventData, image: Image.Image) -> Tuple[int, int, int, int]:
+def get_grid_loc(evt: gr.SelectData, image: Image.Image) -> Tuple[int, int, int, int]:
     """Get grid location from click event."""
-    x, y = evt._data["index"][0], evt._data["index"][1]
+    x, y = evt.index[0], evt.index[1]
     cell_width = image.width // GRID_NUM
     cell_height = image.height // GRID_NUM
     grid_x = x // cell_width
     grid_y = y // cell_height
     return grid_x, grid_y, cell_width, cell_height
 
-def highlight_grid(evt: gr.EventData, image_name: str) -> Image.Image:
+def highlight_grid(evt: gr.SelectData, image_name: str) -> Image.Image:
     """Highlight selected grid cell."""
     image = _CACHE.get('data_dict', image_name, {}).get("image")
     if not image:
@@ -538,7 +547,6 @@ def start_memory_monitor(interval: int = 300):
 data_dict, sae_data_dict = load_all_data(image_root="./data/image", pkl_root=PKL_ROOT)
 default_image_name = "christmas-imagenet"
 
-
 # Create the Gradio interface
 with gr.Blocks(
     theme=gr.themes.Citrus(),