Replace edge hover info with tooltip bubbles on visualization

app.py

@@ -617,21 +617,5 @@ def show_layer_analysis(node_data, activation_data):
     except Exception as e:
         return html.P(f"Error loading analysis: {str(e)}", className="placeholder-text")
 
-# Edge hover callback for token information
-@app.callback(
-    Output('model-status', 'children'),
-    [Input('model-flow-graph', 'mouseoverEdgeData')]
-)
-def show_edge_info(hover_data):
-    """Show token and probability info when hovering over edges."""
-    if hover_data and 'token' in hover_data:
-        token = hover_data['token']
-        prob = hover_data['probability']
-        return html.Div([
-            html.Span(f"Token: {token} | Probability: {prob:.3f}", 
-                     style={'font-size': '14px', 'color': '#495057'})
-        ])
-    return None
-
 if __name__ == '__main__':
     app.run(debug=True, port=8050)

todo.md

@@ -84,4 +84,8 @@ Steps:
 [x] 3. Update run_analysis callback to NOT compute head categorization automatically
 [x] 4. Update node click callback to compute layer-specific categorization and show in Analysis Results
 [x] 5. Add categorize_single_layer_heads to utils/__init__.py exports
-[ ] 6. Test: Click node, verify categorization appears in Analysis Results section
+[ ] 6. Test: Click node, verify categorization appears in Analysis Results section
+
+## Feature Updates:
+[x] Replace edge hover info (displayed above visualization) with tooltip bubbles on edges
+✅ Update complete!

utils/model_patterns.py

@@ -419,6 +419,7 @@ def _create_edge(src_layer: int, tgt_layer: int, token: str, prob: float, rank:
             'target': f'layer_{tgt_layer}' if isinstance(tgt_layer, int) else tgt_layer,
             'token': token,
             'probability': prob,
+            'label': f"{token} | {prob:.3f}",  # Tooltip on hover
             'width': max(2, prob * 10),
             'opacity': max(0.3, prob),
             'color': token_to_color(token)