Add random active neuron discovery feature to Neuroscope

app.py

@@ -123,6 +123,21 @@ default_neuron_index = 1
 default_max_val = 4.0
 default_min_val = 0.0
 
+def get_random_active_neuron(text, threshold=2.0):
+    # Try random layers and neurons until we find one that exceeds threshold
+    import random
+    max_attempts = 100
+    
+    for _ in range(max_attempts):
+        layer = random.randint(0, model.cfg.n_layers - 1)
+        neuron = random.randint(0, model.cfg.d_mlp - 1)
+        acts = get_neuron_acts(text, layer, neuron)
+        if acts.max() > threshold:
+            return layer, neuron
+    
+    # If no neuron found, return default values
+    return 0, 0
+
 with gr.Blocks() as demo:
     gr.HTML(value=f"Neuroscope for {model_name}")
     with gr.Row():
@@ -132,6 +147,7 @@ with gr.Blocks() as demo:
             neuron_index = gr.Number(
                 label="Neuron Index", value=default_neuron_index, precision=0
             )
+            random_btn = gr.Button("Find Random Active Neuron")
             max_val = gr.Number(label="Max Value", value=default_max_val)
             min_val = gr.Number(label="Min Value", value=default_min_val)
             inputs = [text, layer, neuron_index, max_val, min_val]
@@ -146,6 +162,17 @@ with gr.Blocks() as demo:
                     default_min_val,
                 ),
             )
+    
+    def random_neuron_callback(text):
+        layer_num, neuron_num = get_random_active_neuron(text)
+        return layer_num, neuron_num
+    
+    random_btn.click(
+        random_neuron_callback,
+        inputs=[text],
+        outputs=[layer, neuron_index]
+    )
+    
     for inp in inputs:
         inp.change(basic_neuron_vis, inputs, out)