Update app.py

app.py

@@ -256,34 +256,25 @@ def build_cb(cost, salv, life, year):
         return pd.DataFrame([{"error": str(e)}])
     return df
 
-def check_cb(cost, salv, life, year, table_docx, table_img):
+def check_cb(cost, salv, life, year, table_state):
+    # expected (numeric)
     exp = build_sl_schedule(float(cost), float(salv), int(life), int(year))
+    exp["Year"] = pd.to_numeric(exp["Year"], errors="coerce")
 
-    # Prefer the image-normalized table if present; else the docx table
-    actual = None
-    for t in (table_img, table_docx):
-        if isinstance(t, pd.DataFrame) and not t.empty:
-            actual = t.copy()
-            break
-
-    if not isinstance(actual, pd.DataFrame) or actual.empty:
+    # nothing to check?
+    if not isinstance(table_state, pd.DataFrame) or table_state.empty:
         return pd.DataFrame(), "No student table found to check."
 
-    # 🔧 CRUCIAL: Re-normalize & force numerics because Gradio returns strings
-    actual = _normalize_depr_columns(actual)
-    for c in ["Year","Begin BV","Depreciation","Accum Dep","End BV"]:
+    # 👇 Gradio returns strings → re-normalize and coerce here every time
+    actual = _normalize_depr_columns(table_state)
+    for c in ["Year", "Begin BV", "Depreciation", "Accum Dep", "End BV"]:
         actual[c] = pd.to_numeric(actual[c], errors="coerce")
     actual = actual.dropna(subset=["Year"]).reset_index(drop=True)
 
-    # Ensure exp Year is numeric too
-    exp["Year"] = pd.to_numeric(exp["Year"], errors="coerce")
-
     deltas, msg = audit_against_expected(exp, actual)
     return deltas, msg
 
 
-
-
 # ---------- UI ----------
 with gr.Blocks(title="Jerry • HW Intake (Echo)") as demo:
     last_params = gr.State({})