Update app.py

app.py

@@ -288,6 +288,15 @@ def cross_plot_static(actual, pred):
 # Track plot (Plotly)
 # =========================
 def track_plot(df, include_actual=True, pred_col="GR_Pred", actual_col="GR"):
+    # --- helper to guarantee a 1D series even if there are duplicate column names
+    def _col_1d(frame: pd.DataFrame, col: str) -> pd.Series:
+        if col not in frame.columns:
+            return pd.Series(dtype=float)
+        v = frame[col]
+        if isinstance(v, pd.DataFrame):  # happens when there are duplicate column names
+            v = v.iloc[:, 0]
+        return pd.Series(v, dtype=float)
+
     depth_col = next((c for c in df.columns if 'depth' in str(c).lower()), None)
     if depth_col is not None:
         y = pd.Series(df[depth_col]).astype(float)
@@ -299,9 +308,14 @@ def track_plot(df, include_actual=True, pred_col="GR_Pred", actual_col="GR"):
         y_range = [float(y.max()), float(y.min())]
 
     # X (GR) range & ticks
-    x_series = pd.Series(df.get(pred_col, pd.Series(dtype=float))).astype(float)
+    x_pred = _col_1d(df, pred_col)
     if include_actual and actual_col in df.columns:
-        x_series = pd.concat([x_series, pd.Series(df[actual_col]).astype(float)], ignore_index=True)
+        x_act = _col_1d(df, actual_col)
+        x_series = pd.concat([x_pred, x_act], ignore_index=True)
+    else:
+        x_act = None
+        x_series = x_pred
+
     x_lo, x_hi = float(x_series.min()), float(x_series.max())
     x_pad = 0.03 * (x_hi - x_lo if x_hi > x_lo else 1.0)
     xmin, xmax = x_lo - x_pad, x_hi + x_pad
@@ -309,14 +323,14 @@ def track_plot(df, include_actual=True, pred_col="GR_Pred", actual_col="GR"):
 
     fig = go.Figure()
     fig.add_trace(go.Scatter(
-        x=df[pred_col], y=y, mode="lines",
+        x=x_pred, y=y, mode="lines",
         line=dict(color=COLORS["pred"], width=1.8),
-        name="GR_Pred",
+        name=pred_col,
         hovertemplate="GR_Pred: %{x:.0f}<br>"+ylab+": %{y}<extra></extra>"
     ))
-    if include_actual and actual_col in df.columns:
+    if include_actual and x_act is not None:
         fig.add_trace(go.Scatter(
-            x=df[actual_col], y=y, mode="lines",
+            x=x_act, y=y, mode="lines",
             line=dict(color=COLORS["actual"], width=2.0, dash="dot"),
             name="GR (actual)",
             hovertemplate="GR (actual): %{x:.0f}<br>"+ylab+": %{y}<extra></extra>"
@@ -352,6 +366,7 @@ def track_plot(df, include_actual=True, pred_col="GR_Pred", actual_col="GR"):
     )
     return fig
 
+
 # ---------- Preview modal (matplotlib) ----------
 def preview_tracks(df: pd.DataFrame, cols: list[str]):
     cols = [c for c in cols if c in df.columns]
@@ -566,12 +581,11 @@ if st.session_state.app_step == "dev":
 
         col_track, col_cross = st.columns([2, 3], gap="large")
         with col_track:
-            df_for_plot = df.rename(columns={"GR_Actual":"GR"})
-            st.plotly_chart(
-                track_plot(df_for_plot, include_actual=True, pred_col="GR_Pred", actual_col="GR"),
-                use_container_width=False,
-                config={"displayModeBar": False, "scrollZoom": True}
-            )
+    st.plotly_chart(
+        track_plot(df, include_actual=True, pred_col="GR_Pred", actual_col="GR_Actual"),
+        use_container_width=False,
+        config={"displayModeBar": False, "scrollZoom": True}
+    )
         with col_cross:
             st.pyplot(cross_plot_static(df["GR_Actual"], df["GR_Pred"]), use_container_width=False)
 
@@ -653,11 +667,11 @@ if st.session_state.app_step == "validate":
 
         col_track, col_cross = st.columns([2, 3], gap="large")
         with col_track:
-            st.plotly_chart(
-                track_plot(st.session_state.results["Validate"].rename(columns={"GR_Actual":"GR"}),
-                           include_actual=True, pred_col="GR_Pred", actual_col="GR"),
-                use_container_width=False, config={"displayModeBar": False, "scrollZoom": True}
-            )
+    st.plotly_chart(
+        track_plot(st.session_state.results["Validate"],
+                   include_actual=True, pred_col="GR_Pred", actual_col="GR_Actual"),
+        use_container_width=False, config={"displayModeBar": False, "scrollZoom": True}
+    )
         with col_cross:
             st.pyplot(
                 cross_plot_static(st.session_state.results["Validate"]["GR_Actual"],