Update app.py

app.py

@@ -51,6 +51,31 @@ FONT_SZ = 13
 BOLD_FONT = "Arial Black, Arial, sans-serif"
 PLAIN_FONT = "Arial, sans-serif"
 
+# ===== Axis Titles (easy to edit) ===========================================
+# Track plots: set either to a string to override, or leave as None to use defaults.
+X_AXIS_TITLE_OVERRIDE = "Breakout Limit (pcf)"          # e.g., "Mud Weight (pcf)"
+Y_AXIS_TITLE_OVERRIDE = Depth (ft)          # e.g., "Depth (ft)"
+
+# Cross-plot axis titles. You can use {units} and it will fill from meta.
+CROSS_TITLES = {
+    "bo": {"x": "Actual Breakout Limit ({units})", "y": "Predicted Breakout Limit ({units})"},
+    "bd": {"x": "Actual Breakdown Limit ({units})", "y": "Predicted Breakdown Limit ({units})"},
+}
+
+def _track_x_title() -> str:
+    return X_AXIS_TITLE_OVERRIDE or st.session_state.get("X_UNITS", "MW (pcf)")
+
+def _track_y_title(default_ylab: str) -> str:
+    return Y_AXIS_TITLE_OVERRIDE or default_ylab
+
+def _cross_titles(kind: str) -> tuple[str, str]:
+    units = st.session_state.get("X_UNITS", "MW (pcf)")
+    t = CROSS_TITLES.get(kind, {})
+    xlab = (t.get("x") or f"Actual {kind.upper()} ({units})").format(units=units)
+    ylab = (t.get("y") or f"Predicted {kind.upper()} ({units})").format(units=units)
+    return xlab, ylab
+# ===========================================================================
+
 # =========================
 # Page / CSS
 # =========================
@@ -330,11 +355,13 @@ def cross_plot_static(actual, pred, xlabel, ylabel, color="#1f77b4"):
 def _depth_series(df):
     depth_col = next((c for c in df.columns if 'depth' in str(c).lower()), None)
     if depth_col is not None:
-        y = pd.to_numeric(df[depth_col], errors="coerce"); ylab = depth_col
+        y = pd.to_numeric(df[depth_col], errors="coerce"); ylab_default = depth_col
         rng = [float(y.max()), float(y.min())]  # reversed
     else:
-        y = pd.Series(np.arange(1, len(df) + 1)); ylab = "Point Index"
+        y = pd.Series(np.arange(1, len(df) + 1)); ylab_default = "Point Index"
         rng = [float(y.max()), float(y.min())]
+    # apply override for Y-axis title
+    ylab = _track_y_title(ylab_default)
     return y, ylab, rng
 
 def _x_range_for_tracks(df, cols):
@@ -366,7 +393,7 @@ def track_plot_single(df, pred_col, actual_col=None, title_suffix=""):
                     bgcolor="rgba(255,255,255,0.75)", bordercolor="#ccc", borderwidth=1),
         legend_title_text="", title=title_suffix)
     fig.update_xaxes(
-        title_text=st.session_state.get("X_UNITS","MW (pcf)"),
+        title_text=_track_x_title(),
         title_font=dict(size=20, family=BOLD_FONT, color="#000"),
         tickfont=dict(size=15, family=PLAIN_FONT, color="#000"),
         side="top", range=[xmin, xmax], ticks="outside", tickformat=",.2f",
@@ -412,7 +439,7 @@ def track_plot_combined(df):
                     bgcolor="rgba(255,255,255,0.75)", bordercolor="#ccc", borderwidth=1),
         legend_title_text="", title="Combined (Breakout / Breakdown)")
     fig.update_xaxes(
-        title_text=st.session_state.get("X_UNITS","MW (pcf)"),
+        title_text=_track_x_title(),
         title_font=dict(size=20, family=BOLD_FONT, color="#000"),
         tickfont=dict(size=15, family=PLAIN_FONT, color="#000"),
         side="top", range=[xmin, xmax], ticks="outside", tickformat=",.2f",
@@ -649,7 +676,6 @@ if st.session_state.app_step == "dev":
         _metrics_block("BO", mbo); _metrics_block("BD", mbd)
         st.markdown("<div style='text-align:left;font-size:0.8em;color:#6b7280;margin-top:-16px;margin-bottom:8px;'><strong>R</strong> = Pearson correlation • <strong>RMSE/MAE</strong> in MW (pcf)</div>", unsafe_allow_html=True)
         t1, t2, t3 = st.tabs(["Breakout", "Breakdown", "Combined"])
-        units = st.session_state.get("X_UNITS","MW (pcf)")
         with t1:
             left, right = st.columns([3,1], gap="large")
             with left:
@@ -658,11 +684,9 @@ if st.session_state.app_step == "dev":
                     use_container_width=False, config={"displayModeBar": False, "scrollZoom": True}
                 )
             with right:
+                xlab, ylab = _cross_titles("bo")
                 st.pyplot(
-                    cross_plot_static(df[TARGET_BO], df[PRED_BO],
-                                      f"Actual {TARGET_BO} ({units})",
-                                      f"Predicted {TARGET_BO} ({units})",
-                                      COLORS["pred_bo"]),
+                    cross_plot_static(df[TARGET_BO], df[PRED_BO], xlab, ylab, COLORS["pred_bo"]),
                     use_container_width=False
                 )
         with t2:
@@ -673,11 +697,9 @@ if st.session_state.app_step == "dev":
                     use_container_width=False, config={"displayModeBar": False, "scrollZoom": True}
                 )
             with right:
+                xlab, ylab = _cross_titles("bd")
                 st.pyplot(
-                    cross_plot_static(df[TARGET_BD], df[PRED_BD],
-                                      f"Actual {TARGET_BD} ({units})",
-                                      f"Predicted {TARGET_BD} ({units})",
-                                      COLORS["pred_bd"]),
+                    cross_plot_static(df[TARGET_BD], df[PRED_BD], xlab, ylab, COLORS["pred_bd"]),
                     use_container_width=False
                 )
         with t3:
@@ -741,11 +763,10 @@ if st.session_state.app_step == "validate":
         c1,c2,c3 = st.columns(3)
         c1.metric("R (BO)", f"{m_bo['R']:.3f}"); c2.metric("RMSE (BO)", f"{m_bo['RMSE']:.2f}"); c3.metric("MAE (BO)", f"{m_bo['MAE']:.2f}")
         c1,c2,c3 = st.columns(3)
-        c1.metric("R (BD)", f"{m_bd['R']:.3f}"); c2.metric("RMSE (BD)", f"{m_bd['RMSE']:.2f}"); c3.metric("MAE (BD)", f"{m_bd['MAE']:.2f}")
+        c1.metric("R (BD)", f"{m_bd['R']:.3f}"); c2.metric("RMSE (BD)", f"{m_bd['RMSE']:.2f}"); c3.metric(f"MAE (BD)", f"{m_bd['MAE']:.2f}")
         st.markdown("<div style='text-align:left;font-size:0.8em;color:#6b7280;margin-top:-16px;margin-bottom:8px;'>R = Pearson correlation</div>", unsafe_allow_html=True)
 
         t1, t2, t3 = st.tabs(["Breakout", "Breakdown", "Combined"])
-        units = st.session_state.get("X_UNITS","MW (pcf)")
         with t1:
             left, right = st.columns([3,1], gap="large")
             with left:
@@ -754,11 +775,9 @@ if st.session_state.app_step == "validate":
                     use_container_width=False, config={"displayModeBar": False, "scrollZoom": True}
                 )
             with right:
+                xlab, ylab = _cross_titles("bo")
                 st.pyplot(
-                    cross_plot_static(df[TARGET_BO], df[PRED_BO],
-                                      f"Actual {TARGET_BO} ({units})",
-                                      f"Predicted {TARGET_BO} ({units})",
-                                      COLORS["pred_bo"]),
+                    cross_plot_static(df[TARGET_BO], df[PRED_BO], xlab, ylab, COLORS["pred_bo"]),
                     use_container_width=False
                 )
         with t2:
@@ -769,11 +788,9 @@ if st.session_state.app_step == "validate":
                     use_container_width=False, config={"displayModeBar": False, "scrollZoom": True}
                 )
             with right:
+                xlab, ylab = _cross_titles("bd")
                 st.pyplot(
-                    cross_plot_static(df[TARGET_BD], df[PRED_BD],
-                                      f"Actual {TARGET_BD} ({units})",
-                                      f"Predicted {TARGET_BD} ({units})",
-                                      COLORS["pred_bd"]),
+                    cross_plot_static(df[TARGET_BD], df[PRED_BD], xlab, ylab, COLORS["pred_bd"]),
                     use_container_width=False
                 )
         with t3:
@@ -864,6 +881,7 @@ if st.session_state.show_preview_modal:
             for t, name in zip(tabs, names):
                 with t:
                     df = book_to_preview[name]
+                    # show a strict check summary, but do not stop
                     missing = [c for c in st.session_state["FEATURES"] if c not in df.columns]
                     st.write("**Missing vs required features:**", missing if missing else "None ✅")
                     t1, t2 = st.tabs(["Tracks", "Summary"])