Update app.py

app.py

@@ -141,7 +141,8 @@ except AttributeError:
             return wrapper
         return deco
 
-def rmse(y_true, y_pred): return float(np.sqrt(mean_squared_error(y_true, y_pred)))
+def rmse(y_true, y_pred):
+    return float(np.sqrt(mean_squared_error(y_true, y_pred)))
 
 @st.cache_resource(show_spinner=False)
 def load_model(model_path: str):
@@ -153,7 +154,8 @@ def parse_excel(data_bytes: bytes):
     xl = pd.ExcelFile(bio)
     return {sh: xl.parse(sh) for sh in xl.sheet_names}
 
-def read_book_bytes(b: bytes): return parse_excel(b) if b else {}
+def read_book_bytes(b: bytes):
+    return parse_excel(b) if b else {}
 
 def ensure_cols(df, cols):
     miss = [c for c in cols if c not in df.columns]
@@ -165,25 +167,23 @@ def ensure_cols(df, cols):
 def find_sheet(book, names):
     low2orig = {k.lower(): k for k in book.keys()}
     for nm in names:
-        if nm.lower() in low2orig: return low2orig[nm.lower()]
+        if nm.lower() in low2orig:
+            return low2orig[nm.lower()]
     return None
 
 def _nice_tick0(xmin: float, step: int = 100) -> float:
-    """Round xmin down to a sensible multiple so the first tick sits at the left edge."""
-    if not np.isfinite(xmin):
-        return xmin
-    return step * math.floor(xmin / step)
+    return step * math.floor(xmin / step) if np.isfinite(xmin) else xmin
 
-# ---------- Plot builders ----------
+# ---------- cross_plot ----------
 def cross_plot(actual, pred):
     a = pd.Series(actual).astype(float)
     p = pd.Series(pred).astype(float)
 
     fixed_min = 6000
     fixed_max = 10000
-# Choose tick spacing and first tick
-tick_spacing = 500
-tick_start = 6000
+    tick_spacing = 500
+    tick_start = 6000
+
     fig = go.Figure()
 
     # Scatter points
@@ -194,12 +194,12 @@ tick_start = 6000
         showlegend=False
     ))
 
-    # 1:1 reference line from bottom-left to top-right
-fig.add_trace(go.Scatter(
-    x=[x0, x1], y=[x0, x1], mode="lines",
-    line=dict(color=COLORS["ref"], width=1.2, dash="dash"),
-    hoverinfo="skip", showlegend=False
-))
+    # 1:1 diagonal line
+    fig.add_trace(go.Scatter(
+        x=[fixed_min, fixed_max], y=[fixed_min, fixed_max], mode="lines",
+        line=dict(color=COLORS["ref"], width=1.2, dash="dash"),
+        hoverinfo="skip", showlegend=False
+    ))
 
     fig.update_layout(
         width=CROSS_W, height=CROSS_H,
@@ -207,52 +207,48 @@ fig.add_trace(go.Scatter(
         margin=dict(l=64, r=18, t=10, b=48),
         hovermode="closest",
         font=dict(size=FONT_SZ),
-        dragmode=False  # disables zooming
+        dragmode=False
     )
 
-fig.update_xaxes(
-    title_text="<b>Actual UCS (psi)</b>",
-    title_font=dict(size=18, family="Arial", color="#000"),
-    range=[x0, x1],
-    tick0=tick_start,
-    dtick=tick_spacing,
-    ticks="outside", tickformat=",.0f",
-    showline=True, linewidth=1.2, linecolor="#444", mirror=True,
-    showgrid=True, gridcolor="rgba(0,0,0,0.12)", automargin=True
-)
+    fig.update_xaxes(
+        title_text="<b>Actual UCS (psi)</b>",
+        title_font=dict(size=18, family="Arial", color="#000"),
+        range=[fixed_min, fixed_max],
+        tick0=tick_start, dtick=tick_spacing,
+        ticks="outside", tickformat=",.0f",
+        showline=True, linewidth=1.2, linecolor="#444", mirror=True,
+        showgrid=True, gridcolor="rgba(0,0,0,0.12)", automargin=True
+    )
 
-fig.update_yaxes(
-    title_text="<b>Predicted UCS (psi)</b>",
-    title_font=dict(size=18, family="Arial", color="#000"),
-    range=[x0, x1],
-    tick0=tick_start,
-    dtick=tick_spacing,
-    ticks="outside", tickformat=",.0f",
-    showline=True, linewidth=1.2, linecolor="#444", mirror=True,
-    showgrid=True, gridcolor="rgba(0,0,0,0.12)",
-    scaleanchor="x", scaleratio=1, automargin=True
-)
+    fig.update_yaxes(
+        title_text="<b>Predicted UCS (psi)</b>",
+        title_font=dict(size=18, family="Arial", color="#000"),
+        range=[fixed_min, fixed_max],
+        tick0=tick_start, dtick=tick_spacing,
+        ticks="outside", tickformat=",.0f",
+        showline=True, linewidth=1.2, linecolor="#444", mirror=True,
+        showgrid=True, gridcolor="rgba(0,0,0,0.12)",
+        scaleanchor="x", scaleratio=1, automargin=True
+    )
 
     return fig
 
-
+# ---------- track_plot ----------
 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:
+    if depth_col:
         y = pd.Series(df[depth_col]).astype(float)
         ylab = depth_col
-        y_min, y_max = float(y.min()), float(y.max())
-        y_range = [y_max, y_min]   # reversed for log profile style
     else:
         y = pd.Series(np.arange(1, len(df) + 1))
         ylab = "Point Index"
-        y_min, y_max = float(y.min()), float(y.max())
-        y_range = [y_max, y_min]
 
-    # X (UCS) range & ticks
+    y_range = [float(y.max()), float(y.min())]
+
     x_series = pd.Series(df.get("UCS_Pred", pd.Series(dtype=float))).astype(float)
     if include_actual and TARGET in df.columns:
         x_series = pd.concat([x_series, pd.Series(df[TARGET]).astype(float)], ignore_index=True)
+
     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
@@ -292,8 +288,8 @@ def track_plot(df, include_actual=True):
         title_text="<b>UCS (psi)</b>",
         title_font=dict(size=18, family="Arial", color="#000"),
         side="top", range=[xmin, xmax],
-        ticks="outside", tickformat=",.0f",
-        tickmode="auto", tick0=tick0,
+        tick0=tick0, tickmode="auto", tickformat=",.0f",
+        ticks="outside",
         showline=True, linewidth=1.2, linecolor="#444", mirror=True,
         showgrid=True, gridcolor="rgba(0,0,0,0.12)", automargin=True
     )
@@ -308,9 +304,6 @@ def track_plot(df, include_actual=True):
     )
 
     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]