Upload app_simple_ui.py

app_simple_ui.py

@@ -0,0 +1,768 @@
+"""
+Subdivision Cluster / FIPS-Validation pipeline — Hugging Face Space (Gradio).
+
+Faithful port of the Colab notebook
+`LatestUndeterminedZipCityCensusTract70_Threshold_95_Clusters.ipynb`.
+
+Colab-specific I/O replaced:
+  * Drive input folder      -> file upload (multiple .xlsx/.xls)
+  * google.colab.files      -> Gradio file outputs (written to a temp dir)
+  * Drive safety-list path  -> get_safety_list_path() (repo file OR private dataset)
+
+All processing logic (FIPS validation, RawFile snapshot, RapidFuzz clustering,
+wide tables, majority-vote replacement, column reorder, OnlyBlanks warnings,
+chunked CSV export, self-test) matches the notebook.
+"""
+
+import os
+import re
+import gc
+import math
+import tempfile
+from pathlib import Path
+from collections import defaultdict
+
+import numpy as np
+import pandas as pd
+import gradio as gr
+from rapidfuzz import fuzz
+
+
+# =====================================================================
+# REFERENCE FILE LOADER  (the Subdivision Program Safety List)
+# ---------------------------------------------------------------------
+# 1. A file committed in the Space repo (override name via SAFETY_LIST_FILE).
+#    Use ONLY if the Space is PRIVATE.
+# 2. A private HF *dataset* repo, downloaded with the HF_TOKEN secret. Set:
+#       SAFETY_LIST_REPO        e.g. "your-username/subdivision-safety-list"
+#       SAFETY_LIST_REPO_FILE   filename inside that dataset (optional)
+#       HF_TOKEN                read token, stored as a Space Secret
+# =====================================================================
+DEFAULT_SAFETY_LIST_NAME = "Subdivision_Program_Safety_List.xlsx"
+EXCEL_ROW_LIMIT = 1_048_575   # matches the notebook
+
+# Fixed pipeline settings (notebook values). Not exposed in the UI on purpose:
+# users should not be able to deviate from the notebook's behavior.
+STRICT_BLANK_CITIES = True    # blank cities that are not in the safety list
+CLUSTER_THRESHOLD = 95        # RapidFuzz similarity for LegalSubDivision clusters
+REPLACE_THRESHOLD = 70        # majority-vote % required to fill a blank
+FILTER_DOMINANT_COUNTY = False  # process ALL counties (multi-county / full-state files)
+
+
+def get_safety_list_path() -> str:
+    local_name = os.environ.get("SAFETY_LIST_FILE", DEFAULT_SAFETY_LIST_NAME)
+    if Path(local_name).exists():
+        return local_name
+
+    repo_id = os.environ.get("SAFETY_LIST_REPO")
+    token = os.environ.get("HF_TOKEN")
+    if repo_id and token:
+        from huggingface_hub import hf_hub_download
+        filename = os.environ.get("SAFETY_LIST_REPO_FILE", local_name)
+        return hf_hub_download(
+            repo_id=repo_id, filename=filename, repo_type="dataset", token=token
+        )
+
+    raise FileNotFoundError(
+        "Safety list not found.\n"
+        f"  - Commit '{local_name}' into this Space (keep the Space private), OR\n"
+        "  - Set SAFETY_LIST_REPO + HF_TOKEN to pull it from a private dataset."
+    )
+
+
+_SAFETY_CACHE: dict | None = None
+
+
+def build_whitelist_from_path(path):
+    """Parse a safety-list file (xlsx or csv) into {pairs, cities, rows}."""
+    if str(path).lower().endswith(".csv"):
+        safety_df = pd.read_csv(path)
+    else:
+        safety_df = pd.read_excel(path)
+    acceptable_cols = [c for c in safety_df.columns if str(c).startswith("Acceptable FIPS")]
+
+    valid_pairs, valid_cities = set(), set()
+    for _, r in safety_df.iterrows():
+        c = _norm_city(r["City Name"])
+        if c is None:
+            continue
+        valid_cities.add(c)
+        for col in acceptable_cols:
+            f = _norm_fips(r[col])
+            if f:
+                valid_pairs.add(c + "||" + f)
+
+    return {"pairs": valid_pairs, "cities": valid_cities, "rows": len(safety_df)}
+
+
+def load_whitelist():
+    """Server-side safety list (repo file or private dataset). Cached per process."""
+    global _SAFETY_CACHE
+    if _SAFETY_CACHE is None:
+        _SAFETY_CACHE = build_whitelist_from_path(get_safety_list_path())
+    return _SAFETY_CACHE
+
+
+# =====================================================================
+# NORMALIZATION HELPERS  (identical to the notebook)
+# =====================================================================
+PLACEHOLDERS = {"UNDETERMINED CITY", "UNDETERMINED ZIP", "NULL", "NAN", "NONE", ""}
+
+
+def _norm_city(x):
+    if pd.isna(x):
+        return None
+    return str(x).strip().upper()
+
+
+def _norm_fips(x):
+    """5-digit FIPS string, or None. Handles 28071, 28071.0, '01067', '28071 - LAFAYETTE'."""
+    if pd.isna(x):
+        return None
+    m = re.search(r"\d+", str(x))
+    return str(int(m.group(0))).zfill(5) if m else None
+
+
+# ---------------------------------------------------------------------
+# FIPS-COUNTY COLUMN DETECTION (by CONTENT, not header name)
+# ---------------------------------------------------------------------
+# The source header has changed before ("** REC FIPS-COUNTY" ->
+# "** REC FIPS-COUNTY NAME"), so we identify the column by what's IN it:
+# values shaped like "05045-FAULKNER" (FIPS digits, a dash, a county name).
+# Header names are only used as a last-resort fallback.
+FIPS_COUNTY_PATTERN = r"^\s*\d{1,5}\s*-\s*[A-Za-z]"
+FALLBACK_FIPS_HEADERS = ["** REC FIPS-COUNTY NAME", "** REC FIPS-COUNTY", "County"]
+
+
+def detect_fips_county_column(df, sample_size=1000, min_score=0.5):
+    """Return (column_name, reason). Picks the column whose values best match
+    the 'FIPS-COUNTYNAME' shape; falls back to known headers, then to any
+    header mentioning FIPS."""
+    best_col, best_score = None, 0.0
+    for col in df.columns:
+        s = df[col].dropna()
+        if s.empty:
+            continue
+        sample = s.head(sample_size).astype(str)
+        try:
+            score = float(sample.str.match(FIPS_COUNTY_PATTERN).mean())
+        except Exception:
+            continue
+        if score > best_score:
+            best_col, best_score = col, score
+
+    if best_col is not None and best_score >= min_score:
+        return best_col, f"content match {best_score:.0%} of sampled rows"
+
+    for name in FALLBACK_FIPS_HEADERS:
+        if name in df.columns:
+            return name, "fallback: known header name"
+
+    for col in df.columns:
+        if "FIPS" in str(col).upper():
+            return col, "fallback: header mentions FIPS"
+
+    return None, "not found"
+
+
+# =====================================================================
+# SELF-TEST  (startup health check — same logic as the notebook's cell)
+# =====================================================================
+def run_self_test() -> str:
+    path = get_safety_list_path()
+    st = pd.read_excel(path)
+    fc = [c for c in st.columns if str(c).startswith("Acceptable FIPS")]
+    pairs = set()
+    for _, r in st.iterrows():
+        c = _norm_city(r["City Name"])
+        if c is None:
+            continue
+        for col in fc:
+            f = _norm_fips(r[col])
+            if f:
+                pairs.add(c + "||" + f)
+
+    good_city = st.iloc[0]["City Name"]
+    good_fips = st.iloc[0]["Acceptable FIPS 1"]
+    test = pd.DataFrame({
+        "DDPPropertyCityFinal": [good_city, good_city],
+        "** REC FIPS-COUNTY": [good_fips, 99999],   # row 2 = wrong FIPS on purpose
+    })
+    cn = test["DDPPropertyCityFinal"].map(_norm_city)
+    fn = test["** REC FIPS-COUNTY"].map(_norm_fips)
+    key = cn.fillna("") + "||" + fn.fillna("")
+    ok = key.isin(pairs)
+    assert bool(ok.iloc[0]) and not bool(ok.iloc[1]), \
+        "SELF-TEST FAILED -- matching logic is not working as expected."
+    return f"SELF-TEST PASSED: known-good ({good_city}) kept, wrong FIPS flagged."
+
+
+# =====================================================================
+# STEP 1 — FIPS SAFETY-LIST VALIDATION
+# =====================================================================
+def fips_validate(df: pd.DataFrame, blank_cities_not_in_list: bool, log, wl=None,
+                  fips_col=None):
+    if wl is None:
+        wl = load_whitelist()
+    valid_pairs, valid_cities = wl["pairs"], wl["cities"]
+
+    if fips_col is None:
+        fips_col, _reason = detect_fips_county_column(df)
+    if fips_col is None:
+        raise ValueError(
+            "Could not find a FIPS-county column. Expected values like "
+            "'05045-FAULKNER' in one of the columns."
+        )
+    FIPS_COL = fips_col
+    CITY_COL, ZIP_COL, TRACT_COL = (
+        "DDPPropertyCityFinal", "DDPPropertyZIPFinal", "DDPRevisedCensusTract",
+    )
+
+    city_norm = df[CITY_COL].map(_norm_city)
+    fips_norm = df[FIPS_COL].map(_norm_fips)
+    row_key = city_norm.fillna("") + "||" + fips_norm.fillna("")
+
+    is_real_city = df[CITY_COL].notna() & (~city_norm.fillna("").isin(PLACEHOLDERS))
+    city_in_list = city_norm.isin(valid_cities)
+    pair_ok = row_key.isin(valid_pairs)
+
+    if blank_cities_not_in_list:
+        fails_mask = is_real_city & (~pair_ok)
+    else:
+        fails_mask = is_real_city & city_in_list & (~pair_ok)
+
+    n_real = int(is_real_city.sum())
+    n_inlist = int((is_real_city & city_in_list).sum())
+    n_valid = int((is_real_city & pair_ok).sum())
+    n_fail = int(fails_mask.sum())
+    log.append(
+        f"FIPS validation — list: {wl['rows']:,} rows | {len(valid_cities):,} cities | "
+        f"{len(valid_pairs):,} pairs"
+    )
+    log.append(
+        f"  real:{n_real:,} in-list:{n_inlist:,} valid-kept:{n_valid:,} blanked:{n_fail:,}"
+    )
+    if n_real > 0 and n_inlist == 0:
+        log.append("  !! 0 cities matched the list — likely a format/spelling mismatch.")
+
+    # Audit built BEFORE overwrite (insert position matches the notebook).
+    key_col = "RecordKey" if "RecordKey" in df.columns else None
+    base_cols = [c for c in [key_col, CITY_COL, FIPS_COL, ZIP_COL, TRACT_COL] if c]
+    audit = df.loc[fails_mask, base_cols].copy()
+    audit.insert(
+        len(base_cols) if key_col is None else 2,
+        "NormalizedFIPS", fips_norm[fails_mask].values,
+    )
+    audit["Reason"] = np.where(
+        city_in_list[fails_mask].values, "FIPS does not match city", "City not in safety list"
+    )
+    audit = audit.rename(
+        columns={CITY_COL: "OLD_City", ZIP_COL: "OLD_ZIP", TRACT_COL: "OLD_CensusTract"}
+    )
+
+    # Recent pandas reads all-numeric Excel columns as float64 and refuses to
+    # accept a string into them; Colab's older pandas silently upcast to object.
+    # Cast first so the placeholder write reproduces the notebook's result.
+    df[CITY_COL] = df[CITY_COL].astype(object)
+    df[ZIP_COL] = df[ZIP_COL].astype(object)
+    df.loc[fails_mask, CITY_COL] = "UNDETERMINED CITY"
+    df.loc[fails_mask, ZIP_COL] = "UNDETERMINED ZIP"
+    df.loc[fails_mask, TRACT_COL] = np.nan
+    return df, audit
+
+
+# =====================================================================
+# STEP 2 — RAPIDFUZZ CLUSTERING  (identical to the notebook)
+# =====================================================================
+def group_similar_strings_optimized(strings, threshold=95, prefix_len=3, max_len_diff=3):
+    cleaned, seen = [], set()
+    for s in strings:
+        if pd.isna(s):
+            continue
+        s = str(s).strip()
+        if s and s not in seen:
+            cleaned.append(s)
+            seen.add(s)
+
+    visited, groups = set(), []
+    buckets = defaultdict(list)
+    for s in cleaned:
+        buckets[s.upper()[:prefix_len]].append(s)
+
+    for bucket in buckets.values():
+        for i, base in enumerate(bucket):
+            if base in visited:
+                continue
+            group = [base]
+            visited.add(base)
+            base_upper = base.upper()
+            base_len = len(base_upper)
+            for j in range(i + 1, len(bucket)):
+                compare = bucket[j]
+                if compare in visited:
+                    continue
+                comp_upper = compare.upper()
+                if abs(base_len - len(comp_upper)) > max_len_diff:
+                    continue
+                if fuzz.ratio(base_upper, comp_upper) >= threshold:
+                    group.append(compare)
+                    visited.add(compare)
+            groups.append(group)
+    return groups
+
+
+def build_clusters(city_df, zip_df, census_tracts_df, cluster_threshold):
+    all_legal = pd.concat(
+        [
+            city_df[["County", "LegalSubDivision"]],
+            zip_df[["County", "LegalSubDivision"]],
+            census_tracts_df[["County", "LegalSubDivision"]],
+        ],
+        ignore_index=True,
+    ).dropna(subset=["County", "LegalSubDivision"])
+    all_legal["County"] = all_legal["County"].astype(str).str.strip()
+    all_legal["LegalSubDivision"] = all_legal["LegalSubDivision"].astype(str).str.strip()
+    all_legal = all_legal[(all_legal["County"] != "") & (all_legal["LegalSubDivision"] != "")]
+    all_legal = all_legal[["County", "LegalSubDivision"]].drop_duplicates()
+
+    groups = []
+    for county in all_legal["County"].unique():
+        strings = all_legal[all_legal["County"] == county]["LegalSubDivision"].tolist()
+        groups.extend(
+            group_similar_strings_optimized(
+                strings, threshold=cluster_threshold, prefix_len=3, max_len_diff=3
+            )
+        )
+
+    rows = []
+    for cid, group in enumerate(groups, start=1):
+        for sub in group:
+            rows.append({"Cluster_ID": cid, "LegalSubDivision": sub})
+    return pd.DataFrame(rows, columns=["Cluster_ID", "LegalSubDivision"])
+
+
+# =====================================================================
+# STEP 3 — WIDE TABLES + MAJORITY-VOTE "REPLACE WITH"
+# =====================================================================
+def build_cluster_wide_table(source_df, cluster_df, value_col, prefix):
+    lk = cluster_df[["LegalSubDivision", "Cluster_ID"]].drop_duplicates("LegalSubDivision")
+    cluster_map = dict(zip(lk["LegalSubDivision"], lk["Cluster_ID"]))
+
+    w = source_df[["LegalSubDivision", value_col]].copy()
+    w["LegalSubDivision"] = w["LegalSubDivision"].astype(str).str.strip()
+    w[value_col] = w[value_col].replace("", np.nan)
+    w["Cluster_ID"] = w["LegalSubDivision"].map(cluster_map)
+    w = w.dropna(subset=["Cluster_ID", value_col])
+    if w.empty:
+        return pd.DataFrame(columns=["Cluster_ID"])
+    w["Cluster_ID"] = w["Cluster_ID"].astype(int)
+    w[value_col] = w[value_col].astype(str).str.strip()
+    w = w[w[value_col] != ""]
+
+    counts = w.groupby(["Cluster_ID", value_col], observed=True).size().unstack(fill_value=0)
+    totals = counts.sum(axis=1)
+    pct = counts.div(totals.replace(0, np.nan), axis=0).mul(100).fillna(0)
+    counts.columns = [f"{prefix}{c}_Count" for c in counts.columns]
+    pct.columns = [f"{prefix}{c}_Percentage" for c in pct.columns]
+    wide = pd.concat([counts, pct], axis=1).reset_index()
+
+    # Match the notebook: Cluster_ID first, all other columns sorted.
+    other = sorted([c for c in wide.columns if c != "Cluster_ID"])
+    return wide[["Cluster_ID"] + other]
+
+
+def build_replace_with(wide_df, prefix_to_remove, threshold):
+    pct_cols = [c for c in wide_df.columns if c.endswith("_Percentage")]
+    if not pct_cols:
+        res = wide_df[["Cluster_ID"]].copy()
+        res["Replace with"] = None
+        return res.drop_duplicates(subset=["Cluster_ID"])
+    labels = np.array(
+        [c.replace(prefix_to_remove, "").replace("_Percentage", "") for c in pct_cols],
+        dtype=object,
+    )
+    values = wide_df[pct_cols].to_numpy(copy=False)
+    out = []
+    for row in values:
+        matched = labels[row >= threshold]
+        out.append(", ".join(matched) if len(matched) else None)
+    res = wide_df[["Cluster_ID"]].copy()
+    res["Replace with"] = out
+    return res.drop_duplicates(subset=["Cluster_ID"])
+
+
+def clean_census_tract_suffix(value):
+    if pd.isna(value):
+        return np.nan
+    try:
+        f = float(value)
+        return str(int(f)) if f == int(f) else str(f)
+    except ValueError:
+        return str(value)
+
+
+def clean_zip(value):
+    """Normalize ZIPs read from Excel as numbers: drop the float '.0' artifact and
+    zero-pad to 5 digits (recovering leading zeros lost on numeric import, e.g.
+    7221.0 -> 07221). Handles comma-joined majority-vote results. ZIP+4 left as-is."""
+    if pd.isna(value):
+        return np.nan
+    out = []
+    for part in str(value).split(","):
+        p = part.strip()
+        if p.endswith(".0"):
+            p = p[:-2]
+        digits = re.sub(r"\D", "", p)
+        if not digits:
+            continue
+        out.append(digits.zfill(5) if len(digits) <= 5 else digits)
+    return ", ".join(out) if out else np.nan
+
+
+# =====================================================================
+# CHUNKED CSV WRITER  (faithful port of save_csv_in_parts)
+# =====================================================================
+def save_csv_in_parts(df, base_filename, out_dir, row_limit=1_000_000):
+    base_name = base_filename.replace(".csv", "")
+    created = []
+    total = len(df)
+    num_parts = math.ceil(total / row_limit)
+    for i in range(num_parts):
+        start = i * row_limit
+        end = min((i + 1) * row_limit, total)
+        part = f"{base_name}_part_{i + 1}.csv" if num_parts > 1 else base_filename
+        p = out_dir / part
+        df.iloc[start:end].to_csv(p, index=False)
+        created.append(str(p))
+        gc.collect()
+    return created
+
+
+# =====================================================================
+# MAIN PIPELINE
+# =====================================================================
+def run_pipeline(uploaded_files, reference_file, blank_cities_not_in_list,
+                 filter_dominant_county, cluster_threshold, replace_threshold):
+    log = []
+    if not uploaded_files:
+        return "Please upload at least one Excel file.", []
+
+    out_dir = Path(tempfile.mkdtemp(prefix="subdiv_"))
+    outputs = []
+
+    # ---- Resolve the reference (safety) list -------------------------
+    # Default: server-side list (repo file or private dataset). If the user
+    # uploads an override on the sidebar, use it for THIS run only.
+    try:
+        if reference_file is not None:
+            wl = build_whitelist_from_path(reference_file.name)
+            log.append(f"Using UPLOADED reference list: {Path(reference_file.name).name}")
+        else:
+            wl = load_whitelist()
+            log.append("Using server-side safety list.")
+    except FileNotFoundError as e:
+        return "\n".join(log) + f"\n\nREFERENCE FILE ERROR:\n{e}", []
+    except Exception as e:
+        return "\n".join(log) + f"\n\nCould not parse reference list: {e}", []
+
+    # ---- Load & merge uploads ----------------------------------------
+    dfs = []
+    for f in uploaded_files:
+        try:
+            dfs.append(pd.read_excel(f.name))
+            log.append(f"Loaded {Path(f.name).name}")
+        except Exception as e:
+            log.append(f"ERROR loading {Path(f.name).name}: {e}")
+    if not dfs:
+        return "\n".join(log) + "\nNo files could be read.", []
+    df = pd.concat(dfs, ignore_index=True)
+    log.append(f"Merged shape: {df.shape}")
+
+    # ---- Locate the FIPS-county column by CONTENT --------------------
+    fips_col, fips_reason = detect_fips_county_column(df)
+    if fips_col is None:
+        return ("\n".join(log) +
+                "\n\nCOLUMN ERROR: could not find a FIPS-county column. Expected "
+                "values like '05045-FAULKNER' in one of the columns."), []
+    log.append(f"FIPS-county column: '{fips_col}' ({fips_reason})")
+
+    # ---- FIPS validation (uses the reference list) -------------------
+    try:
+        df, audit = fips_validate(df, blank_cities_not_in_list, log, wl=wl, fips_col=fips_col)
+    except FileNotFoundError as e:
+        return "\n".join(log) + f"\n\nREFERENCE FILE ERROR:\n{e}", []
+
+    # ---- Download verification files (audit + RawFile_FIPSValidated) -
+    if len(audit) > 0:
+        ap = out_dir / "FIPS_Validation_Audit_ChangedRecords.xlsx"
+        audit.to_excel(ap, index=False)
+        outputs.append(str(ap))
+
+    if len(df) <= EXCEL_ROW_LIMIT:
+        rp = out_dir / "RawFile_FIPSValidated.xlsx"
+        df.to_excel(rp, index=False)
+    else:
+        rp = out_dir / "RawFile_FIPSValidated.csv"
+        df.to_csv(rp, index=False)
+    outputs.append(str(rp))
+
+    # ---- UNDETERMINED/NULL -> NaN ------------------------------------
+    for col in ["DDPPropertyCityFinal", "DDPPropertyZIPFinal",
+                "DDPRevisedCensusTract", "LegalSubDivision"]:
+        if col in df.columns:
+            df[col] = df[col].replace(["UNDETERMINED CITY", "UNDETERMINED ZIP", "NULL"], np.nan)
+
+    # ---- Rename detected FIPS col -> County --------------------------
+    if fips_col != "County":
+        if "County" in df.columns:
+            df = df.drop(columns=["County"])
+        df = df.rename(columns={fips_col: "County"})
+
+    # ---- Filter to single most-dominant county -----------------------
+    # DISABLED: we now process big multi-county files (up to a full state) in
+    # one run, so we keep every county. Clustering already runs per-county.
+    # To re-enable single-county mode, uncomment the block below.
+    # if filter_dominant_county:
+    #     dominant = df["County"].value_counts().idxmax()
+    #     df = df[df["County"] == dominant].copy()
+    #     log.append(f"Filtered to dominant county: {dominant} -> {df.shape}")
+
+    # ---- Source frames -----------------------------------------------
+    city_df = df.loc[df["DDPPropertyCityFinal"].notna()].copy()
+    zip_df = df.loc[df["DDPPropertyZIPFinal"].notna()].copy()
+    census_tracts_df = df.loc[df["DDPRevisedCensusTract"].notna()].copy()
+    original = df.loc[df["LegalSubDivision"].notna()].copy()
+
+    # ---- Cluster ------------------------------------------------------
+    cluster_df = build_clusters(city_df, zip_df, census_tracts_df, cluster_threshold)
+    log.append(f"Clusters: {cluster_df['Cluster_ID'].nunique():,}")
+
+    # ---- Replace-with (majority vote) --------------------------------
+    zip_w = build_replace_with(
+        build_cluster_wide_table(zip_df, cluster_df, "DDPPropertyZIPFinal", "ZIP_"),
+        "ZIP_", replace_threshold).rename(columns={"Replace with": "Replace with Zip Codes"})
+    city_w = build_replace_with(
+        build_cluster_wide_table(city_df, cluster_df, "DDPPropertyCityFinal", "City_"),
+        "City_", replace_threshold).rename(columns={"Replace with": "Replace with Cities"})
+    cen_w = build_replace_with(
+        build_cluster_wide_table(census_tracts_df, cluster_df, "DDPRevisedCensusTract", "CensusTract_"),
+        "CensusTract_", replace_threshold).rename(columns={"Replace with": "Replace with Census Tracts"})
+
+    # ---- Map back onto original --------------------------------------
+    lk = cluster_df[["LegalSubDivision", "Cluster_ID"]].drop_duplicates("LegalSubDivision")
+    cmap = dict(zip(lk["LegalSubDivision"], lk["Cluster_ID"]))
+    original_final = original.copy()
+    original_final["Cluster_ID"] = original_final["LegalSubDivision"].map(cmap)
+    original_final["Replace with Zip Codes"] = original_final["Cluster_ID"].map(
+        dict(zip(zip_w["Cluster_ID"], zip_w["Replace with Zip Codes"])))
+    original_final["Replace with Cities"] = original_final["Cluster_ID"].map(
+        dict(zip(city_w["Cluster_ID"], city_w["Replace with Cities"])))
+    original_final["Replace with Census Tracts"] = original_final["Cluster_ID"].map(
+        dict(zip(cen_w["Cluster_ID"], cen_w["Replace with Census Tracts"])))
+
+    # Normalize ZIPs (drop float '.0', restore 5-digit zero-padding).
+    original_final["Replace with Zip Codes"] = (
+        original_final["Replace with Zip Codes"].apply(clean_zip))
+
+    # ---- Reorder columns (desired_leading_columns) -------------------
+    desired_leading = [
+        "RecordKey", "ComesFromMtg2", "IsFlagged", "Cluster_ID", "LegalSubDivision",
+        "County", "Replace with Zip Codes", "Replace with Cities", "Replace with Census Tracts",
+        "DDPPropertyZIPFinal", "DDPPropertyCityFinal", "DDPRevisedCensusTract",
+        "LotBlockSubdivisionTract", "DDPRevisedLatitude", "DDPRevisedLongitude",
+    ]
+    existing = original_final.columns.tolist()
+    lead = [c for c in desired_leading if c in existing]
+    other = [c for c in existing if c not in lead]
+    original_final = original_final[lead + other]
+
+    original_final_output = original_final.copy()   # full output snapshot
+
+    # ---- Split into final cities / zips / census tracts --------------
+    keep = ["RecordKey", "ComesFromMtg2", "IsFlagged",
+            "Replace with Zip Codes", "Replace with Cities", "Replace with Census Tracts",
+            "DDPPropertyZIPFinal", "DDPPropertyCityFinal", "DDPRevisedCensusTract"]
+    keep = [c for c in keep if c in original_final.columns]
+    of = original_final[keep].rename(columns={
+        "DDPPropertyZIPFinal": "DDPPropertyZIPFinalToRemove",
+        "DDPPropertyCityFinal": "DDPPropertyCityFinalToRemove",
+        "DDPRevisedCensusTract": "DDPRevisedCensusTractToRemove",
+        "Replace with Zip Codes": "DDPPropertyZIPFinal",
+        "Replace with Cities": "DDPPropertyCityFinal",
+        "Replace with Census Tracts": "DDPRevisedCensusTract",
+    })
+    of["DDPRevisedCensusTract"] = of["DDPRevisedCensusTract"].apply(clean_census_tract_suffix)
+
+    idcols = [c for c in ["RecordKey", "ComesFromMtg2", "IsFlagged"] if c in of.columns]
+    final_cities = of[idcols + ["DDPPropertyCityFinalToRemove", "DDPPropertyCityFinal"]].copy()
+    final_zips = of[idcols + ["DDPPropertyZIPFinalToRemove", "DDPPropertyZIPFinal"]].copy()
+    final_ct = of[idcols + ["DDPRevisedCensusTractToRemove", "DDPRevisedCensusTract"]].copy()
+
+    # keep only rows whose ORIGINAL value was blank (the ones we filled)
+    final_cities = final_cities[final_cities["DDPPropertyCityFinalToRemove"].isna()].copy()
+    final_zips = final_zips[final_zips["DDPPropertyZIPFinalToRemove"].isna()].copy()
+    final_ct = final_ct[final_ct["DDPRevisedCensusTractToRemove"].isna()].copy()
+
+    final_cities = final_cities.drop(columns=["DDPPropertyCityFinalToRemove"])
+    final_zips = final_zips.drop(columns=["DDPPropertyZIPFinalToRemove"])
+    final_ct = final_ct.drop(columns=["DDPRevisedCensusTractToRemove"])
+
+    # ---- OnlyBlanks warnings (column entirely null) ------------------
+    def only_blanks_warning(d, col, prefix):
+        if col in d.columns and d[col].isnull().all():
+            p = out_dir / f"OnlyBlanks_{prefix}.xlsx"
+            d.to_excel(p, index=False)
+            outputs.append(str(p))
+            log.append(f"WARNING: {prefix} has only blanks in '{col}' -> {p.name}")
+
+    only_blanks_warning(final_cities, "DDPPropertyCityFinal", "final_cities")
+    only_blanks_warning(final_zips, "DDPPropertyZIPFinal", "final_zips")
+    only_blanks_warning(final_ct, "DDPRevisedCensusTract", "final_census_tracts")
+
+    # ---- Drop NaNs so only program assignments remain ----------------
+    final_cities = final_cities[final_cities["DDPPropertyCityFinal"].notna()].copy()
+    final_zips = final_zips[final_zips["DDPPropertyZIPFinal"].notna()].copy()
+    final_ct = final_ct[final_ct["DDPRevisedCensusTract"].notna()].copy()
+
+    # ---- Write the per-category + cluster outputs --------------------
+    for d, name in [(final_cities, "final_cities.xlsx"),
+                    (final_zips, "final_zips.xlsx"),
+                    (final_ct, "final_census_tracts.xlsx"),
+                    (cluster_df, "cluster_df.xlsx")]:
+        p = out_dir / name
+        d.to_excel(p, index=False)
+        outputs.append(str(p))
+
+    # ---- Full output CSV (chunked at 1,000,000 rows) -----------------
+    # ---- Full output CSV (chunked at 1,000,000 rows) -----------------
+    # NB: no '%' in the filename -- '%' is the URL-encoding character and makes
+    # the browser download fail with "unknown server error".
+    csv_name = f"OutputFile_{replace_threshold}Threshold_{cluster_threshold}ClustersRapidFuzz.csv"
+    outputs.extend(save_csv_in_parts(original_final_output, csv_name, out_dir, row_limit=1_000_000))
+
+    # ---- Summary sheet: ONE ROW PER COUNTY ---------------------------
+    # Computed entirely from output_df (which carries County, the original
+    # post-validation values, and the "Replace with" fills), so the other
+    # output files are unchanged and still hold all counties' results.
+    output_df = original_final_output
+    city_df, zips_df, census_df = final_cities, final_zips, final_ct
+
+    def _fips_name(label):
+        raw = str(label)
+        parts = raw.split("-")
+        fips = parts[0].strip()
+        name = parts[1].strip().title() if len(parts) > 1 else raw.strip().title()
+        return fips, name
+
+    RW_CITY, RW_ZIP, RW_TRACT = ("Replace with Cities", "Replace with Zip Codes",
+                                 "Replace with Census Tracts")
+    ORIG_CITY, ORIG_ZIP, ORIG_TRACT = ("DDPPropertyCityFinal", "DDPPropertyZIPFinal",
+                                       "DDPRevisedCensusTract")
+
+    def _blank(g, col):
+        return g[col].isna() if col in g.columns else pd.Series(False, index=g.index)
+
+    def _fill(g, col):
+        return g[col].notna() if col in g.columns else pd.Series(False, index=g.index)
+
+    summary_rows = []
+    if "County" in output_df.columns and len(output_df):
+        for county_label, g in output_df.groupby("County", dropna=False):
+            fips, name = _fips_name(county_label)
+            city_blank, zip_blank, tract_blank = (_blank(g, ORIG_CITY),
+                                                  _blank(g, ORIG_ZIP), _blank(g, ORIG_TRACT))
+            city_fill, zip_fill, tract_fill = (_fill(g, RW_CITY),
+                                               _fill(g, RW_ZIP), _fill(g, RW_TRACT))
+            summary_rows.append({
+                "County": name,
+                "FIPS": fips,
+                "Total Records in County": len(g),
+                "Total Subdivisions": int(g["LegalSubDivision"].count())
+                                      if "LegalSubDivision" in g.columns else 0,
+                "Total Clusters Made": int(g["Cluster_ID"].nunique())
+                                       if "Cluster_ID" in g.columns else 0,
+                "Total Blank Cities": int(city_blank.sum()),
+                "Total Cities Assignment": int((city_blank & city_fill).sum()),
+                "Total Blank Zips": int(zip_blank.sum()),
+                "Total Zips Assignment": int((zip_blank & zip_fill).sum()),
+                "Total Blank Census": int(tract_blank.sum()),
+                "Total Census Assignment": int((tract_blank & tract_fill).sum()),
+            })
+
+    summary_cols = ["County", "FIPS", "Total Records in County", "Total Subdivisions",
+                    "Total Clusters Made", "Total Blank Cities", "Total Cities Assignment",
+                    "Total Blank Zips", "Total Zips Assignment", "Total Blank Census",
+                    "Total Census Assignment"]
+    summary_df = pd.DataFrame(summary_rows, columns=summary_cols)
+    log.append(f"Summary: {len(summary_df)} county row(s).")
+
+    summary_path = out_dir / "Summary.xlsx"
+    summary_df.to_excel(summary_path, index=False)
+    outputs.append(str(summary_path))
+
+    # ---- Master workbook: every output as a sheet in one file --------
+    master_path = out_dir / "Master_Dataset.xlsx"
+    with pd.ExcelWriter(master_path, engine="openpyxl") as writer:
+        output_df.to_excel(writer, sheet_name="Output_70_95_RapidFuzz", index=False)
+        # full_validation_df.to_excel(writer, sheet_name="RawFile_FIPSValidated", index=False)
+        audit.to_excel(writer, sheet_name="ValidationAuditResults", index=False)
+        cluster_df.to_excel(writer, sheet_name="cluster_df", index=False)
+        city_df.to_excel(writer, sheet_name="final_cities", index=False)
+        zips_df.to_excel(writer, sheet_name="final_zips", index=False)
+        census_df.to_excel(writer, sheet_name="final_census_tracts", index=False)
+        summary_df.to_excel(writer, sheet_name="Summary", index=False)
+    outputs.append(str(master_path))
+
+    log.append(f"Done. {len(outputs)} files ready.")
+    gc.collect()
+    return "\n".join(log), outputs
+
+
+# =====================================================================
+# STARTUP HEALTH CHECK
+# =====================================================================
+def _startup_health_check():
+    if os.environ.get("SELF_TEST_ON_STARTUP", "1") != "1":
+        return
+    try:
+        print("[startup] " + run_self_test())
+    except FileNotFoundError as e:
+        # Reference file not configured yet — let the UI load so the user can fix it.
+        print(f"[startup] Self-test skipped — safety list unavailable:\n{e}")
+    # AssertionError intentionally propagates: a failed self-test fails the boot.
+
+
+_startup_health_check()
+
+
+# =====================================================================
+# GRADIO UI  (simple vertical layout — previous design)
+# =====================================================================
+with gr.Blocks() as demo:
+    gr.Markdown(
+        "## Subdivision Cluster + FIPS Validation\n"
+        "Upload the Excel files to process. Records are validated against the "
+        "Subdivision Program Safety List, then `LegalSubDivision` values are "
+        "fuzzy-clustered and city/ZIP/census-tract are filled by majority vote.\n\n"
+        "*Runs with the notebook's fixed settings.*"
+    )
+    files_in = gr.File(file_count="multiple", file_types=[".xlsx", ".xls"],
+                       label="Excel files to process")
+    run_btn = gr.Button("Run pipeline", variant="primary")
+    status = gr.Textbox(label="Log", lines=14)
+    files_out = gr.File(label="Output files", file_count="multiple")
+
+    def _run_simple(files):
+        # No reference upload in this layout -> always the server-side list.
+        # All settings are fixed (see constants at top). No user-facing options.
+        return run_pipeline(files, None, STRICT_BLANK_CITIES, FILTER_DOMINANT_COUNTY,
+                            CLUSTER_THRESHOLD, REPLACE_THRESHOLD)
+
+    run_btn.click(
+        _run_simple,
+        inputs=[files_in],
+        outputs=[status, files_out],
+    )
+
+if __name__ == "__main__":
+    demo.launch()