Spaces:
Sleeping
Sleeping
| # Author: Liam Grinstead | |
| # SVG tree with tier and overlay labels | |
| from lineage_tracker import get_ancestors, get_descendants | |
| def _row_positions(n: int, width: int, y: int): | |
| 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, 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 | |
| # Ancestors | |
| ancestors = get_ancestors(agent_id) | |
| if ancestors: | |
| pos = _row_positions(len(ancestors), width, 80) | |
| 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] | |
| svg.append(f"<line x1='{x_last}' y1='{y_last+14}' x2='{root_x}' y2='{root_y-20}' stroke='#555'/>") | |
| else: | |
| svg.append("<text x='12' y='80' font-size='12' fill='#888'>No ancestors</text>") | |
| # 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 | |
| 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) | |
| 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'/>" | |
| ) | |
| y += gap | |
| svg.append("</svg>") | |
| return "".join(svg) | |