Update app.py

app.py

@@ -69,7 +69,7 @@ def _norm_key(s: Optional[str]) -> str:
 @st.cache_data(show_spinner=False)
 def load_csv(upload) -> pd.DataFrame:
     """
-    PASSO A: garante a coluna 'escritorio_cat' SEMPRE.
+    A: garante 'escritorio_cat' SEMPRE:
     - se não existir no CSV, cria;
     - se vier nula, preenche com 'escritorio_responsavel'.
     """
@@ -162,6 +162,16 @@ mapping_str = st.sidebar.text_area(
 base = load_csv(up)
 base = apply_mapping(base, mapping_str)
 
+# B: reforço pós-mapeamento + extra para CSVs antigos
+if "escritorio_cat" not in base.columns:
+    base["escritorio_cat"] = base.get("escritorio_responsavel", "")
+base["escritorio_cat"] = base["escritorio_cat"].fillna(base.get("escritorio_responsavel", "")).astype(str)
+
+for col in ["tribunal", "natureza", "acao", "data_ajuizamento", "valor_causa"]:
+    if col not in base.columns:
+        base[col] = None
+
+# C: guard-clause antes de montar filtros
 if base.empty:
     st.info("Faça upload do CSV para começar.")
     st.stop()