fix(bokeh): address review feedback for column-stratigraphic

Attempt 1/3 - fixes based on AI review

Author: github-actions[bot]

plots/column-stratigraphic/implementations/bokeh.py

@@ -1,63 +1,51 @@
-""" pyplots.ai
+"""pyplots.ai
 column-stratigraphic: Stratigraphic Column with Lithology Patterns
 Library: bokeh 3.9.0 | Python 3.14.3
-Quality: 83/100 | Created: 2026-03-15
 """
 
 from bokeh.io import export_png
-from bokeh.models import ColumnDataSource, HoverTool, Label, Legend, LegendItem, Range1d
+from bokeh.models import ColumnDataSource, HoverTool, Label, Legend, LegendItem, Range1d, Span
 from bokeh.plotting import figure, output_file, save
 
 
-# Data: Synthetic sedimentary section with 10 layers
+# Data: Synthetic sedimentary section with 10 layers (varied thicknesses)
 layers = [
-    {"top": 0, "bottom": 15, "lithology": "Sandstone", "formation": "Dakota Fm", "age": "Late Cretaceous"},
-    {"top": 15, "bottom": 35, "lithology": "Shale", "formation": "Mancos Fm", "age": "Late Cretaceous"},
-    {"top": 35, "bottom": 50, "lithology": "Limestone", "formation": "Niobrara Fm", "age": "Late Cretaceous"},
-    {"top": 50, "bottom": 65, "lithology": "Siltstone", "formation": "Pierre Fm", "age": "Late Cretaceous"},
-    {"top": 65, "bottom": 90, "lithology": "Sandstone", "formation": "Fox Hills Fm", "age": "Late Cretaceous"},
-    {"top": 90, "bottom": 110, "lithology": "Conglomerate", "formation": "Dawson Fm", "age": "Paleocene"},
-    {"top": 110, "bottom": 135, "lithology": "Shale", "formation": "Green River Fm", "age": "Eocene"},
-    {"top": 135, "bottom": 155, "lithology": "Limestone", "formation": "Leadville Fm", "age": "Eocene"},
-    {"top": 155, "bottom": 175, "lithology": "Sandstone", "formation": "Wasatch Fm", "age": "Eocene"},
-    {"top": 175, "bottom": 200, "lithology": "Siltstone", "formation": "Uinta Fm", "age": "Eocene"},
+    {"top": 0, "bottom": 12, "lithology": "Sandstone", "formation": "Dakota Fm", "age": "Late Cretaceous"},
+    {"top": 12, "bottom": 38, "lithology": "Shale", "formation": "Mancos Fm", "age": "Late Cretaceous"},
+    {"top": 38, "bottom": 52, "lithology": "Limestone", "formation": "Niobrara Fm", "age": "Late Cretaceous"},
+    {"top": 52, "bottom": 60, "lithology": "Siltstone", "formation": "Pierre Fm", "age": "Late Cretaceous"},
+    {"top": 60, "bottom": 88, "lithology": "Sandstone", "formation": "Fox Hills Fm", "age": "Late Cretaceous"},
+    {"top": 88, "bottom": 112, "lithology": "Conglomerate", "formation": "Dawson Fm", "age": "Paleocene"},
+    {"top": 112, "bottom": 140, "lithology": "Shale", "formation": "Green River Fm", "age": "Eocene"},
+    {"top": 140, "bottom": 158, "lithology": "Limestone", "formation": "Leadville Fm", "age": "Eocene"},
+    {"top": 158, "bottom": 180, "lithology": "Sandstone", "formation": "Wasatch Fm", "age": "Eocene"},
+    {"top": 180, "bottom": 200, "lithology": "Siltstone", "formation": "Uinta Fm", "age": "Eocene"},
 ]
 
-# Lithology style mapping: color, hatch_pattern
+# Lithology style mapping: color, hatch_pattern (improved color differentiation)
 lithology_styles = {
     "Sandstone": {"color": "#F5DEB3", "hatch_pattern": ".", "hatch_color": "#8B7355"},
     "Shale": {"color": "#A9A9A9", "hatch_pattern": "-", "hatch_color": "#4A4A4A"},
     "Limestone": {"color": "#87CEEB", "hatch_pattern": "+", "hatch_color": "#2E5A88"},
-    "Siltstone": {"color": "#C4B7A6", "hatch_pattern": "/", "hatch_color": "#6B5B4A"},
-    "Conglomerate": {"color": "#D2691E", "hatch_pattern": "o", "hatch_color": "#5C2E00"},
+    "Siltstone": {"color": "#8B6F5E", "hatch_pattern": "/", "hatch_color": "#3E2F23"},
+    "Conglomerate": {"color": "#E8923F", "hatch_pattern": "o", "hatch_color": "#6B3A00"},
 }
 
-# Build data arrays
-tops = [layer["top"] for layer in layers]
-bottoms = [layer["bottom"] for layer in layers]
-lithologies = [layer["lithology"] for layer in layers]
-formations = [layer["formation"] for layer in layers]
-ages = [layer["age"] for layer in layers]
-thicknesses = [layer["bottom"] - layer["top"] for layer in layers]
-
-fill_colors = [lithology_styles[lith]["color"] for lith in lithologies]
-hatch_patterns = [lithology_styles[lith]["hatch_pattern"] for lith in lithologies]
-hatch_colors = [lithology_styles[lith]["hatch_color"] for lith in lithologies]
-
-# Column x-position and width
-col_left = 0.0
-col_right = 1.0
+# K-Pg boundary depth (between Fox Hills Fm / Late Cretaceous and Dawson Fm / Paleocene)
+KPG_DEPTH = 88
+
+# Column x-position and width (wider column for better canvas use)
 col_center = 0.5
 col_width = 1.0
 
-# Plot
+# Plot (tighter x_range for better horizontal utilization)
 p = figure(
     width=4800,
     height=2700,
     title="column-stratigraphic · bokeh · pyplots.ai",
     y_axis_label="Depth (m)",
     toolbar_location=None,
-    x_range=Range1d(-0.8, 2.5),
+    x_range=Range1d(-0.7, 2.0),
     y_range=Range1d(210, -10),
 )
 
@@ -96,12 +84,10 @@
         hatch_weight=2,
     )
 
-    # Track renderers for legend (one entry per lithology)
     lith = layer["lithology"]
     if lith not in legend_items_dict:
         legend_items_dict[lith] = renderer
 
-    # Hover tool per layer
     hover = HoverTool(
         renderers=[renderer],
         tooltips=[
@@ -115,11 +101,28 @@
     )
     p.add_tools(hover)
 
+# K-Pg boundary emphasis — bold red dashed line with annotation
+kpg_span = Span(location=KPG_DEPTH, dimension="width", line_color="#CC0000", line_width=4, line_dash="dashed")
+p.add_layout(kpg_span)
+
+kpg_label = Label(
+    x=col_center,
+    y=KPG_DEPTH,
+    text="K-Pg Boundary (~66 Ma)",
+    text_font_size="16pt",
+    text_font_style="bold",
+    text_color="#CC0000",
+    text_align="center",
+    text_baseline="bottom",
+    y_offset=8,
+)
+p.add_layout(kpg_label)
+
 # Formation labels on the right side
 for layer in layers:
     mid_y = (layer["top"] + layer["bottom"]) / 2
     label = Label(
-        x=col_right + 0.08,
+        x=1.08,
         y=mid_y,
         text=layer["formation"],
         text_font_size="18pt",
@@ -129,7 +132,7 @@
     )
     p.add_layout(label)
 
-# Age labels on the left side
+# Age labels on the left side with brackets
 age_groups = {}
 for layer in layers:
     age = layer["age"]
@@ -153,16 +156,15 @@
     )
     p.add_layout(label)
 
-    # Draw bracket line for age span
     p.line(x=[-0.05, -0.05], y=[bounds["top"], bounds["bottom"]], line_color="#2C2C2C", line_width=2)
     p.line(x=[-0.07, -0.05], y=[bounds["top"], bounds["top"]], line_color="#2C2C2C", line_width=2)
     p.line(x=[-0.07, -0.05], y=[bounds["bottom"], bounds["bottom"]], line_color="#2C2C2C", line_width=2)
 
-# Legend
+# Legend — positioned adjacent to column, not in far corner
 legend_items = [LegendItem(label=lith, renderers=[rend]) for lith, rend in legend_items_dict.items()]
 legend = Legend(
     items=legend_items,
-    location="bottom_right",
+    location="top_right",
     label_text_font_size="20pt",
     spacing=12,
     padding=20,
@@ -173,7 +175,7 @@
     title_text_font_size="22pt",
     title_text_font_style="bold",
 )
-p.add_layout(legend)
+p.add_layout(legend, "right")
 
 # Style
 p.title.text_font_size = "36pt"
@@ -187,6 +189,7 @@
 
 p.yaxis.minor_tick_line_color = None
 p.outline_line_color = None
+p.background_fill_color = "#FAFAFA"
 
 # Save
 export_png(p, filename="plot.png")