MarkusNeusinger
diff --git a/‎plots/funnel-basic/implementations/python/letsplot.py‎
Lines changed: 82 additions & 73 deletions b/‎plots/funnel-basic/implementations/python/letsplot.py‎
Lines changed: 82 additions & 73 deletions
@@ -1,95 +1,104 @@
-""" pyplots.ai
+""" anyplot.ai
 funnel-basic: Basic Funnel Chart
-Library: letsplot 4.8.2 | Python 3.13.11
-Quality: 92/100 | Created: 2025-12-23
+Library: letsplot 4.9.0 | Python 3.14.4
+Quality: 86/100 | Updated: 2026-04-26
 """
 
+import os
+
 import pandas as pd
-from lets_plot import *  # noqa: F403
-from lets_plot.export import ggsave as export_ggsave
+from lets_plot import (
+    LetsPlot,
+    aes,
+    element_blank,
+    element_rect,
+    element_text,
+    geom_polygon,
+    geom_text,
+    ggplot,
+    ggsave,
+    ggsize,
+    labs,
+    scale_color_identity,
+    scale_fill_manual,
+    scale_y_reverse,
+    theme,
+    theme_void,
+)
+
 
+LetsPlot.setup_html()
 
-LetsPlot.setup_html()  # noqa: F405
+# Theme tokens
+THEME = os.getenv("ANYPLOT_THEME", "light")
+PAGE_BG = "#FAF8F1" if THEME == "light" else "#1A1A17"
+INK = "#1A1A17" if THEME == "light" else "#F0EFE8"
 
-# Data - Sales funnel example
+# Okabe-Ito palette — first stage is brand green (#009E73)
+OKABE_ITO = ["#009E73", "#D55E00", "#0072B2", "#CC79A7", "#E69F00"]
+# Light orange (#E69F00) needs dark text for contrast; others use white.
+TEXT_ON_FILL = ["white", "white", "white", "white", INK]
+
+# Data — sales funnel example from specification
 stages = ["Awareness", "Interest", "Consideration", "Intent", "Purchase"]
 values = [1000, 600, 400, 200, 100]
-
-# Calculate percentages and build funnel geometry
-max_value = values[0]
 n_stages = len(stages)
+max_value = values[0]
 
-# Build polygon data for trapezoid shapes
+# Build trapezoid polygon vertices for each stage
+gap = 0.06
 polygons = []
-y_top = 0
-stage_height = 1.0
-
-for i, (stage, value) in enumerate(zip(stages, values, strict=True)):
-    # Calculate widths proportional to max value
-    width = value / max_value
-
-    # Next stage width (minimum 15% to keep readable shape)
-    if i < n_stages - 1:
-        next_width = max(values[i + 1] / max_value, 0.15)
-    else:
-        next_width = max(width * 0.5, 0.10)  # Wider base for final stage
-
-    y_bottom = y_top + stage_height
-
-    # Trapezoid vertices (clockwise from top-left)
-    half_w_top = width / 2
-    half_w_bottom = next_width / 2
-
-    polygons.append({"stage": stage, "x": -half_w_top, "y": y_top, "order": 0})
-    polygons.append({"stage": stage, "x": half_w_top, "y": y_top, "order": 1})
-    polygons.append({"stage": stage, "x": half_w_bottom, "y": y_bottom, "order": 2})
-    polygons.append({"stage": stage, "x": -half_w_bottom, "y": y_bottom, "order": 3})
-
-    y_top = y_bottom
-
-df_poly = pd.DataFrame(polygons)
-
-# Label positions (center of each trapezoid)
 labels = []
-y_pos = 0.5
-for stage, value in zip(stages, values, strict=True):
+for i, (stage, value) in enumerate(zip(stages, values, strict=True)):
+    w_top = value / max_value
+    w_bot = (values[i + 1] / max_value) if i + 1 < n_stages else w_top
+    y_top = i + gap / 2
+    y_bot = (i + 1) - gap / 2
+    polygons.extend(
+        [
+            {"stage": stage, "x": -w_top / 2, "y": y_top, "order": 0},
+            {"stage": stage, "x": w_top / 2, "y": y_top, "order": 1},
+            {"stage": stage, "x": w_bot / 2, "y": y_bot, "order": 2},
+            {"stage": stage, "x": -w_bot / 2, "y": y_bot, "order": 3},
+        ]
+    )
     pct = value / max_value * 100
-    labels.append({"stage": stage, "x": 0, "y": y_pos, "label": f"{stage}\n{value:,} ({pct:.0f}%)"})
-    y_pos += stage_height
+    labels.append(
+        {
+            "stage": stage,
+            "x": 0,
+            "y": (y_top + y_bot) / 2,
+            "label": f"{stage}\n{value:,}  ·  {pct:.0f}%",
+            "text_color": TEXT_ON_FILL[i],
+        }
+    )
 
+df_poly = pd.DataFrame(polygons)
+df_poly["stage"] = pd.Categorical(df_poly["stage"], categories=stages, ordered=True)
 df_labels = pd.DataFrame(labels)
 
-# Colors for each stage - Python blue as primary with variations
-colors = ["#306998", "#4A90D9", "#FFD43B", "#F5A623", "#D45D00"]
-
-# Create plot
+# Plot
 plot = (
-    ggplot()  # noqa: F405
-    + geom_polygon(  # noqa: F405
-        aes(x="x", y="y", fill="stage", group="stage"),  # noqa: F405
-        data=df_poly,
-        color="white",
-        size=2,
-        alpha=0.9,
-    )
-    + geom_text(  # noqa: F405
-        aes(x="x", y="y", label="label"),  # noqa: F405
-        data=df_labels,
-        size=12,
-        color="white",
-        fontface="bold",
-    )
-    + scale_fill_manual(values=colors)  # noqa: F405
-    + scale_y_reverse()  # noqa: F405  # Top to bottom
-    + labs(title="funnel-basic · letsplot · pyplots.ai", x="", y="")  # noqa: F405
-    + theme_void()  # noqa: F405
-    + theme(  # noqa: F405
-        plot_title=element_text(size=24, hjust=0.5, face="bold"),  # noqa: F405
+    ggplot()
+    + geom_polygon(aes(x="x", y="y", fill="stage", group="stage"), data=df_poly, color=PAGE_BG, size=2)
+    + geom_text(aes(x="x", y="y", label="label", color="text_color"), data=df_labels, size=12, fontface="bold")
+    + scale_fill_manual(values=OKABE_ITO[:n_stages], guide="none")
+    + scale_color_identity()
+    + scale_y_reverse()
+    + labs(title="funnel-basic · letsplot · anyplot.ai", x="", y="")
+    + theme_void()
+    + theme(
+        plot_background=element_rect(fill=PAGE_BG, color=PAGE_BG),
+        panel_background=element_rect(fill=PAGE_BG, color=PAGE_BG),
+        plot_title=element_text(size=24, color=INK, face="bold", hjust=0.5),
+        axis_text=element_blank(),
+        axis_ticks=element_blank(),
+        panel_grid=element_blank(),
         legend_position="none",
     )
-    + ggsize(1600, 900)  # noqa: F405
+    + ggsize(1600, 900)
 )
 
-# Save outputs to current directory
-export_ggsave(plot, "plot.png", path=".", scale=3)
-export_ggsave(plot, "plot.html", path=".")
+# Save
+ggsave(plot, f"plot-{THEME}.png", path=".", scale=3)
+ggsave(plot, f"plot-{THEME}.html", path=".")