Update app.py

app.py

@@ -774,6 +774,91 @@ def _db_choice_ui():
     except Exception as e:
         st.sidebar.caption(f"⚙️ DEBUG router fail: {e}")
 
+# ===============================
+# 🔴 Zerar banco (somente Admin, e apenas test/treinamento)
+# ===============================
+def _wipe_current_database_safely():
+    """
+    Zera totalmente o banco ATUAL selecionado no router (apenas test/treinamento).
+    - SQLite: remove o arquivo físico e recria o schema.
+    - Postgres/MySQL: drop_all + recreate.
+    """
+    try:
+        from db_router import current_db_choice
+    except Exception:
+        st.error("Router indisponível. Não é possível determinar o banco atual.")
+        return
+
+    from sqlalchemy.engine.url import make_url
+    from banco import get_engine, init_schema, Base
+
+    choice = (current_db_choice() or "").strip().lower()
+    if choice not in ("test", "treinamento", "train"):
+        st.error("Operação bloqueada: só é permitido zerar **Teste** ou **Treinamento**.")
+        return
+
+    eng = get_engine()
+    url = str(eng.url)
+    url_obj = make_url(url)
+
+    try:
+        if url_obj.drivername.startswith("sqlite"):
+            # Caminho do arquivo
+            path = url.replace("sqlite:///", "", 1)
+            if path.startswith("//"):
+                path = path[1:]
+            file_path = os.path.abspath(path)
+
+            # Fecha conexões e apaga arquivo
+            try:
+                eng.dispose()
+            except Exception:
+                pass
+
+            try:
+                if os.path.exists(file_path):
+                    os.remove(file_path)
+                else:
+                    # Pode estar no ~/.ioirun/<arquivo>
+                    home_alt = os.path.join(os.path.expanduser("~"), ".ioirun", os.path.basename(file_path))
+                    if os.path.exists(home_alt):
+                        os.remove(home_alt)
+            except Exception as e_rm:
+                st.error(f"Falha ao remover arquivo SQLite: {e_rm}")
+                return
+
+            # Recria o schema do zero no arquivo recém criado
+            try:
+                init_schema()
+            except Exception as e_schema:
+                st.error(f"Falha ao recriar schema após remoção do arquivo: {e_schema}")
+                return
+
+        else:
+            # Postgres/MySQL: drop_all + create_all
+            try:
+                Base.metadata.drop_all(bind=eng)
+            except Exception as e_drop:
+                st.error(f"Falha no drop de tabelas: {e_drop}")
+                return
+            try:
+                init_schema()
+            except Exception as e_create:
+                st.error(f"Falha ao recriar schema: {e_create}")
+                return
+
+        st.success("✅ Banco zerado e schema recriado com sucesso.")
+        try:
+            st.toast("Banco zerado com sucesso.", icon="🗑️")
+        except Exception:
+            pass
+
+        # Rerun para refletir o novo estado
+        st.rerun()
+
+    except Exception as e:
+        st.error(f"Erro ao zerar o banco: {e}")
+
 # ===============================
 # MAIN
 # ===============================
@@ -798,11 +883,11 @@ def main():
     # ===== Seleção de banco (sempre visível na sidebar) =====
     _db_choice_ui()
 
-    # 🔧 Manutenção do banco atual: Schema + init_db.run()
+    # 🔧 Manutenção do banco atual: Schema + init_db.run() + Zerar banco (Admin)
     try:
         from banco import init_schema
         with st.sidebar.expander("⚙️ Manutenção do banco atual", expanded=False):
-            if st.button("Criar/atualizar schema neste banco", key="__btn_init_schema__"):
+            if st.button("Criar/atualizar schema neste banco", key="__btn_init_schema__", type="secondary"):
                 try:
                     init_schema()
                     st.sidebar.success("Schema criado/atualizado no banco selecionado com sucesso.")
@@ -823,6 +908,47 @@ def main():
                             st.success("init_db.run() executado com sucesso no banco selecionado.")
                         except Exception as e:
                             st.error(f"Falha ao rodar init_db.run(): {e}")
+
+            # 🔴 Zerar banco (somente Admin, e nunca em Produção)
+            try:
+                perfil_atual = (st.session_state.get("perfil") or "").strip().lower()
+                ch = current_db_choice() if _HAS_ROUTER else "prod"
+            except Exception:
+                perfil_atual = (st.session_state.get("perfil") or "").strip().lower()
+                ch = "prod"
+
+            if perfil_atual == "admin":
+                st.markdown("---")
+                st.markdown("### 🧨 Zerar banco (somente Admin)")
+                # Exibir apenas se não for produção
+                if ch in ("test", "treinamento", "train"):
+                    st.info("Esta ação **apaga todo o conteúdo** do banco **selecionado** (Test/Treinamento) e recria o schema. **Irreversível.**")
+                    colz1, colz2 = st.columns(2)
+                    z_conf1 = colz1.checkbox("Confirmo que entendo os riscos", key="__wipe_c1__")
+                    z_conf2 = colz2.checkbox("Estou ciente que é irreversível", key="__wipe_c2__")
+                    safe_phrase = st.text_input("Digite **ZERAR** para habilitar", key="__wipe_phrase__")
+                    # Seleção explícita do banco-alvo (precisa coincidir com o atual)
+                    alvo = st.selectbox(
+                        "Banco a zerar (deve ser o banco atual):",
+                        ["test", "treinamento"],
+                        index=(0 if ch == "test" else 1 if ch in ("treinamento", "train") else 0),
+                        key="__wipe_target__"
+                    )
+
+                    btn_wipe = st.button("🗑️ ZERAR banco **ATUAL**", key="__btn_wipe_db__", type="secondary")
+                    if btn_wipe:
+                        if alvo not in ("test", "treinamento"):
+                            st.error("Selecione um alvo válido (test/treinamento).")
+                        elif alvo != ("treinamento" if ch in ("treinamento", "train") else ch):
+                            st.error("O alvo selecionado **não coincide** com o banco atual. Ajuste a seleção.")
+                        elif not (z_conf1 and z_conf2):
+                            st.error("Marque as duas confirmações para continuar.")
+                        elif safe_phrase.strip().upper() != "ZERAR":
+                            st.error("Frase de segurança incorreta. Digite exatamente **ZERAR**.")
+                        else:
+                            _wipe_current_database_safely()
+                else:
+                    st.warning("Zerar banco indisponível: **Produção** está selecionado. Troque para **Teste** ou **Treinamento** para habilitar.")
     except Exception:
         pass