Function draw_legend

experiments/human_eval/draw_model_eval_ggbench_style.py:158–166

(draw: ImageDraw.ImageDraw)

def draw_legend(draw: ImageDraw.ImageDraw) -> None:
    x, y = 1605, 245
    for label, fill, edge in [
        ("Closed API", COLORS["closed"], COLORS["closed_edge"]),
        ("Open / open-access", COLORS["open"], COLORS["open_edge"]),
    ]:
        draw.ellipse((x, y - 15, x + 30, y + 15), fill=fill, outline=edge, width=3)
        draw.text((x + 42, y - 17), label, font=F_LEGEND, fill=COLORS["text"])
        y += 48


def draw_plot(rows: list[dict[str, object]]) -> Image.Image: