Remove cross-dataset comparison panel

scripts/explorer_app.py

@@ -211,7 +211,7 @@ def _apply_dataset_globals(idx):
     global live_mask, live_indices, dict_live_mask, dict_live_indices
     global umap_backup
     global _clip_scores, _clip_vocab, _clip_embeds, _clip_scores_f32, HAS_CLIP
-    global _compare_datasets, HAS_COMPARE
+    global _compare_datasets
     global feature_names, _names_file
 
     ds = _all_datasets[idx]
@@ -240,7 +240,6 @@ def _apply_dataset_globals(idx):
     _clip_scores_f32   = ds['clip_scores_f32']
     HAS_CLIP           = _clip_scores is not None and _clip_vocab is not None
     _compare_datasets  = [d for i, d in enumerate(_all_datasets) if i != idx]
-    HAS_COMPARE        = len(_compare_datasets) > 0
     feature_names      = ds['feature_names']
     _names_file        = ds['names_file']
 
@@ -577,7 +576,7 @@ def _on_dataset_switch(attr, old, new):
     status_div.text = _status_html('idle', 'Model switched — select a feature to explore.')
     for div in [top_heatmap_div, top_zoom_div,
                 mean_heatmap_div, mean_zoom_div,
-                p75_heatmap_div, p75_zoom_div, cross_dataset_div]:
+                p75_heatmap_div, p75_zoom_div]:
         div.text = ""
 
 
@@ -596,7 +595,6 @@ mean_heatmap_div = Div(text="", width=900)
 mean_zoom_div = Div(text="", width=900)
 p75_heatmap_div = Div(text="", width=900)
 p75_zoom_div = Div(text="", width=900)
-cross_dataset_div = Div(text="", width=900)   # cross-dataset comparison rows
 
 # Name editing widget (defined here so update_feature_display can reference it)
 name_input = TextInput(
@@ -754,74 +752,6 @@ def update_feature_display(feature_idx):
         p75_zoom_div.text = make_image_grid_html(
             p75_zm_infos, f"75th Percentile — Zoomed to peak patch (feature {feat})")
 
-        # --- Cross-dataset comparison ---
-        if HAS_COMPARE:
-            COMP_THUMB = min(THUMB, 160)
-            N_COMP = 4
-            primary_label = _all_datasets[_active[0]]['label']
-
-            def _row_html(label, imgs_info, token_type='spatial'):
-                tag = (" <span style='font-size:10px;color:#888'>(CLS — no heatmap)</span>"
-                       if token_type == 'cls' else "")
-                html = (f'<div style="margin-bottom:6px">'
-                        f'<b style="font-size:12px">{label}{tag}</b>'
-                        f'<div style="display:flex;gap:4px;margin-top:3px">')
-                for img, cap in imgs_info:
-                    url = pil_to_data_url(img)
-                    html += (f'<div style="text-align:center">'
-                             f'<img src="{url}" width="{COMP_THUMB}" height="{COMP_THUMB}"'
-                             f' style="border:1px solid #ccc;border-radius:3px"/>'
-                             f'<div style="font-size:9px;color:#666">{cap}</div></div>')
-                html += '</div></div>'
-                return html
-
-            # Primary dataset row
-            primary_infos = []
-            for j in range(min(N_COMP, top_img_idx.shape[1])):
-                img_i = top_img_idx[feat, j].item()
-                if img_i < 0:
-                    break
-                try:
-                    pil = load_image(img_i).resize((COMP_THUMB, COMP_THUMB), Image.BILINEAR)
-                    act = top_img_act[feat, j].item()
-                    primary_infos.append((pil, f"{act:.3f}"))
-                except Exception:
-                    primary_infos.append((Image.new("RGB", (COMP_THUMB, COMP_THUMB), (180,180,180)), "err"))
-
-            html = ('<div style="border-top:2px solid #e0e0e0;padding-top:8px;margin-top:6px">'
-                    '<h3 style="margin:0 0 8px 0;color:#333">Cross-dataset comparison'
-                    ' — top images by max activation</h3>')
-            html += _row_html(primary_label, primary_infos, 'spatial')
-
-            # Comparison dataset rows
-            for cd in _compare_datasets:
-                if cd.get('_lazy', False):
-                    html += (f'<div style="margin-bottom:6px">'
-                             f'<b style="font-size:12px">{cd["label"]}</b>'
-                             f' <span style="font-size:10px;color:#888">'
-                             f'(not yet loaded — select this model to load it)</span></div>')
-                    continue
-                cd_infos = []
-                if feat < cd['d_model']:
-                    for j in range(min(N_COMP, cd['top_img_idx'].shape[1])):
-                        img_i = cd['top_img_idx'][feat, j].item()
-                        if img_i < 0:
-                            break
-                        try:
-                            path = cd['image_paths'][img_i]
-                            pil = load_compare_image(path, COMP_THUMB)
-                            act = cd['top_img_act'][feat, j].item()
-                            cd_infos.append((pil, f"{act:.3f}"))
-                        except Exception:
-                            cd_infos.append((Image.new("RGB", (COMP_THUMB, COMP_THUMB), (180,180,180)), "err"))
-                else:
-                    cd_infos = [(Image.new("RGB", (COMP_THUMB, COMP_THUMB), (220,220,220)),
-                                 "feat idx\nout of range")]
-                html += _row_html(cd['label'], cd_infos, cd['token_type'])
-
-            html += '</div>'
-            cross_dataset_div.text = html
-
         status_div.text = _status_html('ok', f'&#x2713; Feature {feat} ready.')
         _update_view_visibility()
 
@@ -1520,7 +1450,6 @@ right_panel = column(
     top_heatmap_div, top_zoom_div,
     mean_heatmap_div, mean_zoom_div,
     p75_heatmap_div, p75_zoom_div,
-    cross_dataset_div,
 )
 
 layout = row(left_panel, middle_panel, right_panel)