Function write_svg

experiments/human_eval/draw_llm_token_cost_tree.py:180–225

()

def write_svg() -> None:
    root_y = 72
    closed_group_y = 170
    open_group_y = 610
    closed_ys = [220 + idx * 44 for idx in range(len(CLOSED_ROWS))]
    open_ys = [660 + idx * 44 for idx in range(len(OPEN_ROWS))]
    stem_x = GROUP_X + 84
    leaf_join_x = LEAF_X - 26
    bar_x = VALUE_X - MAX_BAR_W - 20

    parts = [
        f'<svg xmlns="http://www.w3.org/2000/svg" width="{W}" height="{H}" viewBox="0 0 {W} {H}">',
        '<rect width="100%" height="100%" fill="white"/>',
        f'<line x1="{ROOT_X + 72}" y1="{root_y + 27}" x2="{ROOT_X + 72}" y2="{open_group_y}" stroke="{rgb(LINE)}" stroke-width="2"/>',
    ]

    for group_y, leaf_ys, color in [(closed_group_y, closed_ys, CLOSED), (open_group_y, open_ys, OPEN)]:
        parts.append(f'<line x1="{ROOT_X + 72}" y1="{group_y}" x2="{GROUP_X - 64}" y2="{group_y}" stroke="{rgb(LINE)}" stroke-width="2"/>')
        parts.append(f'<line x1="{GROUP_X + 72}" y1="{group_y}" x2="{stem_x}" y2="{group_y}" stroke="{rgb(LINE)}" stroke-width="2"/>')
        parts.append(f'<line x1="{stem_x}" y1="{min(leaf_ys)}" x2="{stem_x}" y2="{max(leaf_ys)}" stroke="{rgb(LINE)}" stroke-width="2"/>')
        for y in leaf_ys:
            parts.append(f'<line x1="{stem_x}" y1="{y}" x2="{leaf_join_x}" y2="{y}" stroke="{rgb(LINE)}" stroke-width="2"/>')
            parts.append(f'<circle cx="{leaf_join_x}" cy="{y}" r="5" fill="{rgb(color)}"/>')

    parts.extend(svg_node(ROOT_X + 72, root_y, "TokenCost", 30, TEXT))
    parts.extend(svg_node(GROUP_X, closed_group_y, "Closed API", 27, CLOSED))
    parts.extend(svg_node(GROUP_X, open_group_y, "Open / open-access", 27, OPEN))

    for y, (name, value) in zip(closed_ys, CLOSED_ROWS):
        color = PAPERFLOW if name == "PaperFlow" else CLOSED
        label_color = PAPERFLOW if name == "PaperFlow" else TEXT
        bar_w = round((value / MAX_VALUE) * MAX_BAR_W)
        parts.append(f'<text x="{LEAF_X}" y="{y + 8}" font-family="Times New Roman" font-size="24" font-weight="700" fill="{rgb(label_color)}">{name}</text>')
        parts.append(f'<rect x="{bar_x}" y="{y - 9}" width="{MAX_BAR_W}" height="18" rx="5" fill="{rgb(BAR_BG)}"/>')
        parts.append(f'<rect x="{bar_x}" y="{y - 9}" width="{bar_w}" height="18" rx="5" fill="{rgb(color)}"/>')
        parts.append(f'<text x="{VALUE_X}" y="{y + 8}" font-family="Times New Roman" font-size="22" font-weight="700" fill="{rgb(color)}">{value / 1_000_000:.1f}M</text>')

    for y, (name, value) in zip(open_ys, OPEN_ROWS):
        bar_w = round((value / MAX_VALUE) * MAX_BAR_W)
        parts.append(f'<text x="{LEAF_X}" y="{y + 8}" font-family="Times New Roman" font-size="24" font-weight="700" fill="{rgb(TEXT)}">{name}</text>')
        parts.append(f'<rect x="{bar_x}" y="{y - 9}" width="{MAX_BAR_W}" height="18" rx="5" fill="{rgb(BAR_BG)}"/>')
        parts.append(f'<rect x="{bar_x}" y="{y - 9}" width="{bar_w}" height="18" rx="5" fill="{rgb(OPEN)}"/>')
        parts.append(f'<text x="{VALUE_X}" y="{y + 8}" font-family="Times New Roman" font-size="22" font-weight="700" fill="{rgb(OPEN)}">{value / 1_000_000:.1f}M</text>')

    parts.append("</svg>")
    SVG_OUT.write_text("\n".join(parts), encoding="utf-8")


def main() -> None: