Update app.py

app.py

@@ -256,11 +256,27 @@ def build_cb(cost, salv, life, year):
         return pd.DataFrame([{"error": str(e)}])
     return df
 
-def check_cb(cost, salv, life, year, table):
+def check_cb(cost, salv, life, year, table_docx, table_img):
     exp = build_sl_schedule(float(cost), float(salv), int(life), int(year))
-    deltas, msg = audit_against_expected(exp, table if isinstance(table, pd.DataFrame) else pd.DataFrame())
+
+    # 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 columns aren't standardized yet, normalize them
+    if isinstance(actual, pd.DataFrame) and not actual.empty:
+        needed = {"Year","Begin BV","Depreciation","Accum Dep","End BV"}
+        if not needed.issubset(set(actual.columns)):
+            actual = _normalize_depr_columns(actual)
+
+    deltas, msg = audit_against_expected(exp, actual if isinstance(actual, pd.DataFrame) else pd.DataFrame())
     return deltas, msg
 
+
+
 # ---------- UI ----------
 with gr.Blocks(title="Jerry • HW Intake (Echo)") as demo:
     last_params = gr.State({})
@@ -328,7 +344,13 @@ with gr.Blocks(title="Jerry • HW Intake (Echo)") as demo:
         ],
     )
 
-    btn_use.click(fill_from_state, inputs=last_params, outputs=[in_cost, in_salv, in_life, in_year])
+#    btn_use.click(fill_from_state, inputs=last_params, outputs=[in_cost, in_salv, in_life, in_year])
+    btn_check.click(
+        check_cb,
+        [in_cost, in_salv, in_life, in_year, table_df, norm_df],   # <- pass both tables
+        [deltas_df, coach_txt]
+    )
+
     btn_build.click(build_cb, [in_cost, in_salv, in_life, in_year], [expected_df])
     btn_check.click(check_cb, [in_cost, in_salv, in_life, in_year, last_table], [deltas_df, coach_txt])