Update app.py

app.py

@@ -38,15 +38,16 @@ BOLD_FONT = "Arial Black, Arial, sans-serif"  # used for bold axis titles & tick
 st.set_page_config(page_title="ST_GeoMech_UCS", page_icon="logo.png", layout="wide")
 st.markdown("""
 <style>
-  /* Reusable logo style */
-  .brand-logo { width: 16px; height: auto; object-fit: contain; }
+  /* Reusable logo style */
+  .brand-logo { width: 16px; height: auto; object-fit: contain; }
 
-  /* Sidebar header layout */
-  .sidebar-header { display:flex; align-items:center; gap:12px; }
-  .sidebar-header .text h1 { font-size: 1.05rem; margin:0; line-height:1.1; }
-  .sidebar-header .text .tag { font-size: .85rem; color:#6b7280; margin:2px 0 0; }
+  /* Sidebar header layout */
+  .sidebar-header { display:flex; align-items:center; gap:12px; }
+  .sidebar-header .text h1 { font-size: 1.05rem; margin:0; line-height:1.1; }
+  .sidebar-header .text .tag { font-size: .85rem; color:#6b7280; margin:2px 0 0; }
 </style>
 """, unsafe_allow_html=True)
+
 # Hide uploader helper text ("Drag and drop file here", limits, etc.)
 st.markdown("""
 <style>
@@ -198,14 +199,9 @@ def cross_plot_static(actual, pred):
     return fig
 
 # =========================
-# Track plot (Plotly)
+# Track plot (Plotly) — FIXED
 # =========================
-
-    ticks="outside",
-    showline=True, linewidth=1.2, linecolor="#444", mirror=True,
-    showgrid=True, gridcolor="rgba(0,0,0,0.12)", automargin=True
-)
-    return figdef track_plot(df, include_actual=True):
+def track_plot(df, include_actual=True):
     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)
@@ -230,14 +226,14 @@ def cross_plot_static(actual, pred):
         x=df["UCS_Pred"], y=y, mode="lines",
         line=dict(color=COLORS["pred"], width=1.8),
         name="UCS_Pred",
-        hovertemplate="UCS_Pred: %{x:.0f}<br>" + ylab + ": %{y}<extra></extra>"
+        hovertemplate="UCS_Pred: %{x:.0f}<br>"+ylab+": %{y}<extra></extra>"
     ))
     if include_actual and TARGET in df.columns:
         fig.add_trace(go.Scatter(
             x=df[TARGET], y=y, mode="lines",
             line=dict(color=COLORS["actual"], width=2.0, dash="dot"),
             name="UCS (actual)",
-            hovertemplate="UCS (actual): %{x:.0f}<br>" + ylab + ": %{y}<extra></extra>"
+            hovertemplate="UCS (actual): %{x:.0f}<br>"+ylab+": %{y}<extra></extra>"
         ))
 
     fig.update_layout(
@@ -252,7 +248,7 @@ def cross_plot_static(actual, pred):
         legend_title_text=""
     )
 
-    # ✅ These must stay inside the function (properly indented)
+    # Bold axis titles & tick labels
     fig.update_xaxes(
         title_text="UCS (psi)",
         title_font=dict(size=16, family=BOLD_FONT),
@@ -279,8 +275,7 @@ def cross_plot_static(actual, pred):
 
     return fig
 
-
-# ---------- Preview modal (matplotlib) ----------
+# ---------- Preview modal (matplotlib)
 def preview_tracks(df: pd.DataFrame, cols: list[str]):
     cols = [c for c in cols if c in df.columns]
     n = len(cols)
@@ -470,7 +465,7 @@ if st.session_state.app_step == "dev":
         c1,c2,c3 = st.columns(3)
         c1.metric("R", f"{m['R']:.2f}"); c2.metric("RMSE", f"{m['RMSE']:.2f}"); c3.metric("MAE", f"{m['MAE']:.2f}")
 
-        # NEW: 2-column layout, big gap, no nested columns, no 0-weights.
+        # clean 2-column layout
         col_cross, col_track = st.columns([3, 2], gap="large")
         with col_cross:
             st.pyplot(cross_plot_static(df[TARGET], df["UCS_Pred"]), use_container_width=True)
@@ -524,7 +519,9 @@ if st.session_state.app_step == "validate":
                 tbl = df.loc[any_viol, FEATURES].copy()
                 for c in FEATURES:
                     if pd.api.types.is_numeric_dtype(tbl[c]): tbl[c] = tbl[c].round(2)
-                tbl["Violations"] = pd.DataFrame({f:(df[f]<ranges[f][0])|(df[f]>ranges[f][1]) for f in FEATURES}).loc[any_viol].apply(lambda r:", ".join([c for c,v in r.items() if v]), axis=1)
+                tbl["Violations"] = pd.DataFrame({f:(df[f]<ranges[f][0])|(df[f]>ranges[f][1]) for f in FEATURES}).loc[any_viol].apply(
+                    lambda r:", ".join([c for c,v in r.items() if v]), axis=1
+                )
         st.session_state.results["m_val"]={
             "R": pearson_r(df[TARGET], df["UCS_Pred"]),
             "RMSE": rmse(df[TARGET], df["UCS_Pred"]),