Update lineage_visualizer.py

lineage_visualizer.py

@@ -4,22 +4,24 @@
 from lineage_tracker import get_ancestors, get_descendants
 
 def _row_positions(n: int, width: int, y: int):
-    if n <= 0: return []
+    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, overlay_color_hue: int = 260, max_depth: int = 6) -> str:
-    width, height = 980, 640
+    width, height = 980, 480  # reduced height to avoid cutoff
     svg = [f"<svg width='{width}' height='{height}' xmlns='http://www.w3.org/2000/svg'>"]
     svg.append("<style>text{font-family:monospace}</style>")
     svg.append(f"<text x='12' y='24' font-size='16'>RFT Lineage: {agent_id}</text>")
 
-    root_x, root_y = width//2, 170
+    root_x, root_y = width // 2, 170
 
+    # Ancestors
     ancestors = get_ancestors(agent_id)
     if ancestors:
         pos = _row_positions(len(ancestors), width, 80)
-        for (x,y), aid in zip(pos, ancestors):
+        for (x, y), aid in zip(pos, ancestors):
             svg.append(f"<circle cx='{x}' cy='{y}' r='14' fill='hsl({overlay_color_hue},60%,50%)'/>")
             svg.append(f"<text x='{x-28}' y='{y+32}' font-size='11'>{aid}</text>")
         x_last, y_last = pos[-1]
@@ -27,26 +29,42 @@ def render_lineage_tree(agent_id: str, overlay_color_hue: int = 260, max_depth:
     else:
         svg.append("<text x='12' y='80' font-size='12' fill='#888'>No ancestors</text>")
 
-    svg.append(f"<circle cx='{root_x}' cy='{root_y}' r='18' fill='#00b894'>"
-               "<animate attributeName='r' values='18;22;18' dur='2s' repeatCount='indefinite'/></circle>")
+    # Root agent
+    svg.append(
+        f"<circle cx='{root_x}' cy='{root_y}' r='18' fill='#00b894'>"
+        "<animate attributeName='r' values='18;22;18' dur='2s' repeatCount='indefinite'/>"
+        "</circle>"
+    )
     svg.append(f"<text x='{root_x-40}' y='{root_y+40}' font-size='12'>{agent_id}</text>")
 
+    # Descendants
     levels = get_descendants(agent_id, depth=max_depth)
     y = 260
     gap = 90
     for d in range(1, max_depth + 1):
         nodes = levels.get(d, [])
-        if not nodes: break
+        if not nodes:
+            break
         pos = _row_positions(len(nodes), width, y)
         svg.append(f"<text x='12' y='{y-26}' font-size='12' fill='#888'>Gen {d}</text>")
         for (x, yy), item in zip(pos, nodes):
             hue = 40 + (15 * d)
-            svg.append(f"<circle cx='{x}' cy='{yy}' r='14' fill='hsl({hue},70%,55%)'>"
-                       "<animate attributeName='fill' values='hsl({h},70%,55%);hsl({h},70%,65%);hsl({h},70%,55%)' dur='2s' repeatCount='indefinite'/>"
-                       .replace("{h}", str(hue)) + "</circle>")
-            label = f"{item['child_id']} · {item.get('tier','?')} · {item.get('overlay','?')}"
+            child_id = item.get("child_id", "?")
+            tier = item.get("tier", "?")
+            overlay = item.get("overlay", "?")
+            svg.append(
+                f"<circle cx='{x}' cy='{yy}' r='14' fill='hsl({hue},70%,55%)'>"
+                "<animate attributeName='fill' "
+                f"values='hsl({hue},70%,55%);hsl({hue},70%,65%);hsl({hue},70%,55%)' "
+                "dur='2s' repeatCount='indefinite'/>"
+                "</circle>"
+            )
+            label = f"{child_id} · {tier} · {overlay}"
             svg.append(f"<text x='{x-60}' y='{yy+32}' font-size='11'>{label}</text>")
-            svg.append(f"<path d='M {root_x} {root_y+20} Q {(root_x+x)//2} {(root_y+yy)//2} {x} {yy-14}' stroke='rgba(0,0,0,.35)' fill='none'/>")
+            svg.append(
+                f"<path d='M {root_x} {root_y+20} Q {(root_x+x)//2} {(root_y+yy)//2} {x} {yy-14}' "
+                "stroke='rgba(0,0,0,.35)' fill='none'/>"
+            )
         y += gap
 
     svg.append("</svg>")