Update lineage_visualizer.py

lineage_visualizer.py

@@ -1,26 +1,66 @@
+# lineage_visualizer.py
 # Author: Liam Grinstead
-# Renders agent lineage as an SVG tree diagram
-
-from lineage_tracker import LINEAGE
-
-def render_lineage(agent_id):
-    children = LINEAGE.get(agent_id, [])
-    if not children:
-        return f"<svg width='400' height='100'><text x='10' y='50'>No lineage for {agent_id}</text></svg>"
-
-    svg = "<svg width='600' height='400'>"
-    svg += f"<text x='10' y='30' font-size='16'>Lineage of {agent_id}</text>"
-    svg += f"<circle cx='300' cy='60' r='20' fill='purple'/>"
-    svg += f"<text x='280' y='100' font-size='12'>{agent_id}</text>"
-
-    y_offset = 160
-    x_step = 120
-    for i, child in enumerate(children):
-        x_pos = 100 + i * x_step
-        svg += f"<line x1='300' y1='80' x2='{x_pos}' y2='{y_offset-20}' stroke='black'/>"
-        svg += f"<circle cx='{x_pos}' cy='{y_offset}' r='15' fill='green'>"
-        svg += "<animate attributeName='r' values='15;20;15' dur='2s' repeatCount='indefinite'/>"
-        svg += "</circle>"
-        svg += f"<text x='{x_pos-20}' y='{y_offset+40}' font-size='12'>{child['child_id']}</text>"
-    svg += "</svg>"
-    return svg
+# Renders a multi-generation SVG tree: ancestors (top), root (center), descendants (levels below)
+
+from typing import List
+from lineage_tracker import get_ancestors, get_descendants
+
+def _row_positions(n: int, width: int, y: int) -> List[tuple]:
+    # return evenly spaced (x, y) for n nodes
+    if n <= 0:
+        return []
+    step = width // (n + 1)
+    return [(step * (i + 1), y) for i in range(n)]
+
+def render_lineage_tree(agent_id: str, max_depth: int = 5) -> str:
+    width, height = 900, 600
+    margin_top = 40
+    svg = [f"<svg width='{width}' height='{height}' xmlns='http://www.w3.org/2000/svg'>"]
+    svg.append(f"<style>text{{font-family:monospace}} .node{{filter:drop-shadow(0 1px 1px rgba(0,0,0,.2))}}</style>")
+    svg.append(f"<text x='10' y='25' font-size='16'>Lineage view: {agent_id}</text>")
+
+    # Layout rows
+    root_y = 180
+    ancestor_y = 80
+    base_y = 280
+    level_gap = 70
+
+    # Ancestors row
+    ancestors = get_ancestors(agent_id)  # oldest → newest
+    if ancestors:
+        anc_positions = _row_positions(len(ancestors), width, ancestor_y)
+        for (x, y), aid in zip(anc_positions, ancestors):
+            svg.append(f"<circle class='node' cx='{x}' cy='{y}' r='16' fill='#6c5ce7'/>")
+            svg.append(f"<text x='{x-30}' y='{y+34}' font-size='12'>{aid}</text>")
+        # connect ancestors to root (last ancestor to root)
+        last_x, last_y = anc_positions[-1]
+        svg.append(f"<line x1='{last_x}' y1='{last_y+16}' x2='{width//2}' y2='{root_y-16}' stroke='#444'/>")
+    else:
+        svg.append(f"<text x='10' y='{ancestor_y}' font-size='12' fill='#888'>No ancestors</text>")
+
+    # Root node
+    root_x = width // 2
+    svg.append(f"<circle class='node' cx='{root_x}' cy='{root_y}' r='20' fill='#00b894'>"
+               f"<animate attributeName='r' values='20;24;20' dur='2s' repeatCount='indefinite'/></circle>")
+    svg.append(f"<text x='{root_x-40}' y='{root_y+40}' font-size='12'>{agent_id} (root)</text>")
+
+    # Descendant levels
+    descendants = get_descendants(agent_id, depth=max_depth)  # {'level_1': [...], 'level_2': [...]}
+    current_y = base_y
+    for level_idx in range(1, max_depth + 1):
+        level_key = f"level_{level_idx}"
+        nodes = descendants.get(level_key, [])
+        if not nodes:
+            break
+        positions = _row_positions(len(nodes), width, current_y)
+        svg.append(f"<text x='10' y='{current_y-26}' font-size='12' fill='#888'>Gen {level_idx}</text>")
+        for (x, y), nid in zip(positions, nodes):
+            svg.append(f"<circle class='node' cx='{x}' cy='{y}' r='14' fill='#0984e3'>"
+                       f"<animate attributeName='fill' values='#0984e3;#74b9ff;#0984e3' dur='2s' repeatCount='indefinite'/></circle>")
+            svg.append(f"<text x='{x-28}' y='{y+33}' font-size='12'>{nid}</text>")
+            # draw soft link from approximate parent band to this node (visual continuity)
+            svg.append(f"<path d='M {root_x} {root_y+20} Q {(root_x+x)//2} {(root_y+y)//2} {x} {y-14}' stroke='rgba(0,0,0,.35)' fill='none'/>")
+        current_y += level_gap
+
+    svg.append("</svg>")
+    return "".join(svg)