Update app.py

app.py

@@ -69,17 +69,40 @@ from uuid import uuid4
 from sqlalchemy import text, func, or_
 
 # 🗄️ Banco ativo (Produção/Teste/Treinamento)
+# Tentativa de importar utilitários do db_router, com fallback seguro.
 try:
-    from db_router import current_db_choice, bank_label
+    from db_router import current_db_choice, bank_label  # já usados no seu app
     _HAS_ROUTER = True
 except Exception:
     _HAS_ROUTER = False
 
     def current_db_choice() -> str:
-        return "prod"
+        # Fallback simples
+        return st.session_state.get("__db_choice_override__", "prod")
 
     def bank_label(choice: str) -> str:
-        return "🟢 Produção" if choice == "prod" else "🔴 Teste"
+        return {
+            "prod": "🟢 Produção",
+            "test": "🔴 Teste",
+            "train": "🟡 Treinamento",
+        }.get(choice, "🟢 Produção")
+
+# 🔁 Tentativas de funções adicionais do router (se existirem)
+_get_available_choices = None
+_set_db_choice_func = None
+if _HAS_ROUTER:
+    try:
+        from db_router import get_available_choices as _get_available_choices
+    except Exception:
+        _get_available_choices = None
+    # tenta 2 nomes comuns de setter
+    try:
+        from db_router import set_current_db_choice as _set_db_choice_func  # tipo preferido
+    except Exception:
+        try:
+            from db_router import set_db_choice as _set_db_choice_func      # alias
+        except Exception:
+            _set_db_choice_func = None
 
 # =========================================
 # 🔧 STARTUP HOOKS (1x por container / process)
@@ -670,6 +693,70 @@ def exibir_logo_once(top: bool = False, sidebar: bool = False):
             if need_sidebar:
                 state["sidebar"] = True
 
+# ===============================
+# 🔧 Escolha do Banco (UI + setters)
+# ===============================
+def _db_choice_ui():
+    """
+    Renderiza na sidebar a seleção de banco de dados (Produção/Teste/Treinamento).
+    Usa db_router quando disponível; caso não, mantém fallback visual.
+    """
+    with st.sidebar.expander("🗄️ Banco de dados", expanded=True):
+        # Obtém lista de choices
+        if _get_available_choices:
+            try:
+                avail = list(_get_available_choices())  # ex.: ["prod","test","train"]
+                if not avail:
+                    avail = ["prod", "test", "train"]
+            except Exception:
+                avail = ["prod", "test", "train"]
+        else:
+            avail = ["prod", "test", "train"]
+
+        # Label amigável para cada opção
+        def _lbl(c):
+            try:
+                return bank_label(c)
+            except Exception:
+                return {"prod": "🟢 Produção", "test": "🔴 Teste", "train": "🟡 Treinamento"}.get(c, c)
+
+        current = current_db_choice() if callable(current_db_choice) else st.session_state.get("__db_choice_override__", "prod")
+        labels = [_lbl(c) for c in avail]
+        # Mapeia índice atual
+        try:
+            idx = avail.index(current)
+        except Exception:
+            idx = 0
+
+        sel_label = st.selectbox(
+            "Escolha o banco:",
+            options=labels,
+            index=idx,
+            key="__db_choice_select__",
+        )
+        # Resolve escolha → código
+        sel_idx = labels.index(sel_label)
+        sel_code = avail[sel_idx]
+
+        # Aplica se mudou
+        if sel_code != current:
+            # Tenta aplicar no router (se houver)
+            applied = False
+            if _set_db_choice_func:
+                try:
+                    _set_db_choice_func(sel_code)
+                    applied = True
+                except Exception:
+                    applied = False
+
+            # Fallback: guarda override local e exporta env var (se algum loader usar)
+            if not applied:
+                st.session_state["__db_choice_override__"] = sel_code
+                os.environ["DB_CHOICE"] = sel_code  # se o router/engine ler do env
+
+            st.toast(f"Banco alterado para: {_lbl(sel_code)}", icon="🗄️")
+            st.rerun()
+
 # ===============================
 # MAIN
 # ===============================
@@ -688,6 +775,9 @@ def main():
     st.session_state.setdefault("__auto_refresh_interval_sec__", 60)
     st.session_state.setdefault("__auth_diag__", False)  # 🔧 estado do painel de diagnóstico (Admin)
 
+    # ===== Seleção de banco (sempre visível na sidebar) =====
+    _db_choice_ui()
+
     # LOGIN
     if not st.session_state.logado:
         st.session_state.quiz_verificado = False
@@ -695,9 +785,9 @@ def main():
         exibir_logo_once(top=True, sidebar=False)
         login()
 
-        # ⚠️ Opcional: dica breve quando nenhum caminho de auth está definido
-        if not os.getenv("DISABLE_AUTH") and not os.getenv("ALLOW_EMERGENCY_LOGIN"):
-            st.info("🔒 Autenticação ativa. Para testes rápidos no Spaces, defina **DISABLE_AUTH=1** ou habilite o **login emergencial** (ALLOW_EMERGENCY_LOGIN=1 + EMERG_*). Depois, **Restart this Space**.")
+        # 🔕 REMOVIDO: mensagem de dica de autenticação no Spaces (conforme solicitado)
+        # (antes havia um st.info(...) aqui — removemos)
+
         return
 
     # 👥 Heartbeat + Badge de usuários logados (APENAS ADMIN)
@@ -927,11 +1017,9 @@ def main():
 
     st.sidebar.markdown("### Menu | 🎉 Carnaval 🎭 2026!")
 
-    # Banco ativo na sidebar
+    # Banco ativo na sidebar (badge rápido)
     try:
-        _b_label = bank_label(current_db_choice()) if _HAS_ROUTER else (
-            "🟢 Produção" if current_db_choice() == "prod" else "🔴 Teste"
-        )
+        _b_label = bank_label(current_db_choice()) if _HAS_ROUTER else bank_label(current_db_choice())
         st.sidebar.caption(f"🗄️ Banco ativo: {_b_label}")
     except Exception:
         pass
@@ -1039,9 +1127,9 @@ def main():
     termo_busca = st.sidebar.text_input("Pesquisar módulo:").strip().lower()
 
     try:
-        ambiente_atual = current_db_choice() if _HAS_ROUTER else "prod"
+        ambiente_atual = current_db_choice() if _HAS_ROUTER else current_db_choice()
     except Exception:
-        ambiente_atual = "prod"
+        ambiente_atual = current_db_choice()
 
     grupos_disponiveis = obter_grupos_disponiveis(
         MODULES,