Update app.py

app.py

@@ -71,13 +71,12 @@ 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  # já usados no seu app
+    from db_router import current_db_choice, bank_label
     _HAS_ROUTER = True
 except Exception:
     _HAS_ROUTER = False
 
     def current_db_choice() -> str:
-        # Fallback simples
         return st.session_state.get("__db_choice_override__", "prod")
 
     def bank_label(choice: str) -> str:
@@ -125,7 +124,7 @@ if os.getenv("CLEAR_CACHE_ON_START", "0") == "1":
     except Exception:
         pass
 
-# Rodar init_db.run() uma única vez por start de container
+# Rodar init_db.run() uma única vez por start de container (⚠️ roda no banco padrão do ambiente)
 try:
     import init_db as _init_db
     _HAS_INIT_DB = hasattr(_init_db, "run")
@@ -149,17 +148,15 @@ if os.getenv("INIT_DB_ON_START", "0") == "1" and _HAS_INIT_DB:
                 pass
 
 # ===============================
-# RERUN por querystring (atalho ?rr=1)
+# RERUN por querystring (atalho ?rr=1) e aplicar ?db=teste|treinamento
 # ===============================
 def _get_query_params():
     """Compat: retorna query params como dict (Streamlit novo/antigo)."""
     try:
-        # Streamlit >= 1.32
-        return dict(st.query_params)
+        return dict(st.query_params)  # Streamlit >= 1.32
     except Exception:
-        # Streamlit antigo (experimental)
         try:
-            return dict(st.experimental_get_query_params())
+            return dict(st.experimental_get_query_params())  # Antigo
         except Exception:
             return {}
 
@@ -173,14 +170,32 @@ def _set_query_params(new_params: dict):
         except Exception:
             pass
 
+def _apply_db_choice_from_qs():
+    """Permite selecionar o banco via URL (?db=test|treinamento)."""
+    try:
+        params = _get_query_params()
+        db = params.get("db")
+        if not db:
+            return
+        db = db[0] if isinstance(db, (list, tuple)) else db
+        sel = str(db).strip().lower()
+        if _set_db_choice_func:
+            try:
+                _set_db_choice_func(sel)
+            except Exception:
+                os.environ["DB_CHOICE"] = sel
+                st.session_state["__db_choice_override__"] = sel
+        else:
+            os.environ["DB_CHOICE"] = sel
+            st.session_state["__db_choice_override__"] = sel
+    except Exception:
+        pass
+
 # 🔒 ANTI-TREMOR PATCH: não consumir rr=1 quando login/quiz ainda não concluídos
 def _check_rerun_qs(pagina_atual: str = ""):
     """
     Se a URL contiver rr=1 (ou true), força um rerun e limpa o parâmetro para evitar loop.
-    ✅ Não dispara quando estiver na página 'resposta' (Inbox Admin).
-    ✅ (PATCH) Também não dispara quando estiver em 'outlook_relatorio' para não interromper leitura COM.
-    ✅ (PATCH) Não dispara quando estiver em 'formulario'.
-    ✅ (ANTI-TREMOR) Não dispara enquanto login/quiz não concluídos (para não mexer na logo).
+    ✅ Bloqueios para módulos sensíveis e login/quiz.
     """
     try:
         if st.session_state.get("__qs_rr_consumed__", False):
@@ -192,7 +207,7 @@ def _check_rerun_qs(pagina_atual: str = ""):
 
         # Evita rr=1 em módulos sensíveis a rerun/refresh
         if pagina_atual in ("resposta", "outlook_relatorio", "formulario"):
-            return  # evita 'piscar' e cancelamentos
+            return
 
         params = _get_query_params()
         rr_raw = params.get("rr", ["0"])
@@ -229,15 +244,6 @@ def _get_db_session():
     # 3) Fallback
     return SessionLocal()
 
-# ===============================
-# CONFIGURAÇÃO INICIAL
-# ===============================
-# Criação do schema (sem derrubar a app se o banco ainda não estiver pronto)
-try:
-    Base.metadata.create_all(bind=engine)
-except Exception as e:
-    st.sidebar.warning(f"Schema não foi criado automaticamente: {e}")
-
 def quiz_respondido_hoje(usuario: str) -> bool:
     # ✅ Usar sessão ciente do ambiente
     db = _get_db_session()
@@ -758,10 +764,23 @@ def _db_choice_ui():
             st.toast(f"Banco alterado para: {_lbl(sel_code)}", icon="🗄️")
             st.rerun()
 
+    # 🔎 DEBUG da escolha/URL (ajuda a diagnosticar se o router está aplicando)
+    try:
+        from db_router import current_db_choice as _cur, bank_label as _lbl2, get_engine as _eng
+        ch = _cur()
+        eng = _eng()
+        st.sidebar.caption(f"⚙️ DEBUG • Banco atual: {_lbl2(ch)} ({ch})")
+        st.sidebar.caption(f"⚙️ DEBUG • URL: {eng.url}")
+    except Exception as e:
+        st.sidebar.caption(f"⚙️ DEBUG router fail: {e}")
+
 # ===============================
 # MAIN
 # ===============================
 def main():
+    # Aplicar escolha via URL (ex.: ?db=test)
+    _apply_db_choice_from_qs()
+
     # Estados iniciais
     if "logado" not in st.session_state:
         st.session_state.logado = False
@@ -779,7 +798,7 @@ def main():
     # ===== Seleção de banco (sempre visível na sidebar) =====
     _db_choice_ui()
 
-    # 🔧 Criar/atualizar schema no banco selecionado (útil após trocar p/ Teste/Treinamento)
+    # 🔧 Manutenção do banco atual: Schema + init_db.run()
     try:
         from banco import init_schema
         with st.sidebar.expander("⚙️ Manutenção do banco atual", expanded=False):
@@ -789,18 +808,37 @@ def main():
                     st.sidebar.success("Schema criado/atualizado no banco selecionado com sucesso.")
                 except Exception as e:
                     st.sidebar.error(f"Falha ao criar/atualizar schema: {e}")
+
+            # Botão para rodar init_db.run() no banco atual
+            if _HAS_INIT_DB:
+                c1, c2 = st.columns(2)
+                sure1 = c1.checkbox("Confirmo", key="__init_confirm1__")
+                sure2 = c2.checkbox("Estou ciente", key="__init_confirm2__")
+                if st.button("Rodar init_db.run()", key="__btn_run_initdb__", help="Cria usuários padrão e garante schema", type="secondary"):
+                    if not (sure1 and sure2):
+                        st.warning("Marque as duas confirmações para executar.")
+                    else:
+                        try:
+                            _init_db.run()
+                            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}")
     except Exception:
-        # Se o módulo não expor init_schema, apenas ignora silenciosamente
         pass
 
+    # ⚙️ Criação automática do schema (leve) após seleção do banco (rodará no banco correto)
+    try:
+        Base.metadata.create_all(bind=engine)
+    except Exception as e:
+        st.sidebar.warning(f"Schema não foi criado automaticamente: {e}")
+
     # LOGIN
     if not st.session_state.logado:
         st.session_state.quiz_verificado = False
         # 🔒 ANTI-TREMOR: renderiza logo top apenas 1x
         exibir_logo_once(top=True, sidebar=False)
         login()
-
-        # 🔕 REMOVIDO: mensagem de dica de autenticação no Spaces (conforme solicitado)
+        # 🔕 REMOVIDO: mensagem de dica de autenticação no Spaces
         return
 
     # 👥 Heartbeat + Badge de usuários logados (APENAS ADMIN)
@@ -824,11 +862,11 @@ def main():
         # 🔘 Botão Admin: habilitar/desabilitar painel de diagnóstico
         st.sidebar.markdown("---")
         if st.session_state.get("__auth_diag__"):
-            if st.sidebar.button("🧪 Desativar diagnóstico de login (Admin)"):
+            if st.sidebar.button("🧪 Desativar diagnóstico de login (Admin)", key="__btn_auth_diag_off__", type="secondary"):
                 st.session_state["__auth_diag__"] = False
                 st.rerun()
         else:
-            if st.sidebar.button("🧪 Ativar diagnóstico de login (Admin)"):
+            if st.sidebar.button("🧪 Ativar diagnóstico de login (Admin)", key="__btn_auth_diag_on__", type="secondary"):
                 st.session_state["__auth_diag__"] = True
                 st.rerun()
 
@@ -841,14 +879,14 @@ def main():
             # ============================
             from sqlalchemy import inspect  # import local para evitar custo desnecessário
             with st.sidebar.expander("🧪 Diagnóstico Profundo do Banco", expanded=False):
-                db = _get_db_session()
+                dbx = _get_db_session()
                 try:
-                    eng = db.bind
+                    eng = dbx.bind
                     st.caption(f"Engine URL: {eng.url}")
 
                     # 1) SELECT 1
                     try:
-                        db.execute(text("SELECT 1"))
+                        dbx.execute(text("SELECT 1"))
                         st.success("SELECT 1 OK")
                     except Exception as e:
                         st.error(f"SELECT 1 falhou: {e}")
@@ -872,7 +910,7 @@ def main():
 
                         # 5) Contagem
                         try:
-                            cnt = db.execute(text(f"SELECT COUNT(*) FROM {target_table}")).fetchone()[0]
+                            cnt = dbx.execute(text(f"SELECT COUNT(*) FROM {target_table}")).fetchone()[0]
                             st.write("Quantidade de registros:", cnt)
                         except Exception as e:
                             st.error(f"COUNT(*) falhou: {e}")
@@ -881,7 +919,7 @@ def main():
                         try:
                             sel_cols = [c for c in ["usuario", "email", "perfil", "nome"] if c in cols]
                             sel_expr = ", ".join(sel_cols) if sel_cols else "*"
-                            amostra = db.execute(text(f"SELECT {sel_expr} FROM {target_table} LIMIT 5"))
+                            amostra = dbx.execute(text(f"SELECT {sel_expr} FROM {target_table} LIMIT 5"))
                             rows = [dict(r._mapping) for r in amostra]
                             st.write("Amostra:", rows)
                         except Exception as e:
@@ -894,7 +932,7 @@ def main():
                         test_user = st.text_input("Usuário para testar hash", "", key="__bcrypt_user__")
                         test_pass = st.text_input("Senha para testar hash", "", type="password", key="__bcrypt_pass__")
 
-                        if st.button("Testar bcrypt com este usuário", key="__btn_bcrypt_test__"):
+                        if st.button("Testar bcrypt com este usuário", key="__btn_bcrypt_test__", type="secondary"):
                             try:
                                 pass_cols = [c for c in ["senha_hash", "password_hash", "senha", "hash"] if c in cols]
                                 user_cols = [c for c in ["usuario", "username", "login"] if c in cols]
@@ -903,7 +941,7 @@ def main():
                                 else:
                                     c_pass = pass_cols[0]
                                     c_user = user_cols[0]
-                                    res = db.execute(
+                                    res = dbx.execute(
                                         text(f"SELECT {c_user} AS u, {c_pass} AS h FROM {target_table} WHERE {c_user} = :u LIMIT 1"),
                                         {"u": test_user.strip()}
                                     ).fetchone()
@@ -926,7 +964,7 @@ def main():
                                 st.error(f"Erro no teste: {e}")
                 finally:
                     try:
-                        db.close()
+                        dbx.close()
                     except Exception:
                         pass
 
@@ -935,7 +973,7 @@ def main():
             # ============================
             with st.sidebar.expander("🧰 Manutenção (Admin)", expanded=False):
                 col_m1, col_m2 = st.columns(2)
-                if col_m1.button("🧹 Limpar cache agora", use_container_width=True):
+                if col_m1.button("🧹 Limpar cache agora", key="__btn_clear_cache__", help="Limpa cache de dados/recursos", type="secondary"):
                     try:
                         st.cache_data.clear()
                     except Exception:
@@ -947,18 +985,17 @@ def main():
                     st.sidebar.success("Caches limpos.")
                     st.rerun()
 
-                # Rodar init_db.run() com dupla confirmação
+                # Rodar init_db.run() com dupla confirmação (Admin)
                 if _HAS_INIT_DB:
                     c1, c2 = st.columns(2)
-                    sure1 = c1.checkbox("Confirmo")
-                    sure2 = c2.checkbox("Estou ciente")
-                    if st.button("🧬 Rodar init_db.run()", use_container_width=True):
+                    sure1 = c1.checkbox("Confirmo", key="__btn_run_initdb_admin_c1__")
+                    sure2 = c2.checkbox("Estou ciente", key="__btn_run_initdb_admin_c2__")
+                    if st.button("🧬 Rodar init_db.run()", key="__btn_run_initdb_admin__", type="secondary"):
                         if not (sure1 and sure2):
                             st.warning("Marque as duas confirmações para executar.")
                         else:
                             try:
                                 _init_db.run()
-                                # Atualiza marker
                                 with open(_INIT_MARK, "w", encoding="utf-8") as f:
                                     f.write(f"init manual at {datetime.now().isoformat()}\n")
                                 st.success("init_db.run() executado com sucesso.")
@@ -968,7 +1005,7 @@ def main():
     # 🔄 Botão de Recarregar (mantém a sessão ativa) + ⏱️ Controle do intervalo
     st.sidebar.markdown("---")
     col_reload, col_interval = st.sidebar.columns([1, 1])
-    if col_reload.button("🔄 Recarregar (sem sair)", key="__btn_reload_now__"):
+    if col_reload.button("🔄 Recarregar (sem sair)", key="__btn_reload_now__", type="secondary"):
         st.rerun()
 
     if hasattr(st, "popover"):
@@ -979,7 +1016,7 @@ def main():
                 value=int(st.session_state["__auto_refresh_interval_sec__"]),
                 step=5, key="__auto_refresh_input__"
             )
-            if st.button("Aplicar intervalo", key="__btn_apply_auto_refresh__"):
+            if st.button("Aplicar intervalo", key="__btn_apply_auto_refresh__", type="secondary"):
                 st.session_state["__auto_refresh_interval_sec__"] = int(new_val)
                 try:
                     if int(new_val) > 0:
@@ -997,7 +1034,7 @@ def main():
                 value=int(st.session_state["__auto_refresh_interval_sec__"]),
                 step=5, key="__auto_refresh_input__"
             )
-            if st.button("Aplicar intervalo", key="__btn_apply_auto_refresh__"):
+            if st.button("Aplicar intervalo", key="__btn_apply_auto_refresh__", type="secondary"):
                 st.session_state["__auto_refresh_interval_sec__"] = int(new_val)
                 try:
                     if int(new_val) > 0:
@@ -1067,7 +1104,7 @@ def main():
             )
 
         # 👉 Direciona para o MESMO módulo do menu (resposta.main())
-        if st.sidebar.button("📬 Abrir Caixa de Entrada (Admin)"):
+        if st.sidebar.button("📬 Abrir Caixa de Entrada (Admin)", key="__btn_open_inbox__", type="secondary"):
             st.session_state.nav_target = "resposta"
             st.rerun()
 
@@ -1129,7 +1166,7 @@ def main():
                     pass
                 st.session_state["__user_toast_shown__"] = True
 
-            if st.sidebar.button("📥 Ver respostas"):
+            if st.sidebar.button("📥 Ver respostas", key="__btn_view_answers__", type="secondary"):
                 st.session_state.nav_target = "sugestoes_ioirun"
                 st.session_state.user_responses_viewed = True
                 st.rerun()
@@ -1274,7 +1311,7 @@ def main():
     # Logout
     st.sidebar.markdown("---")
     if st.session_state.get("logado"):
-        if st.sidebar.button("🚪 Sair (Logout)"):
+        if st.sidebar.button("🚪 Sair (Logout)", key="__btn_logout__", type="primary"):
             logout()
 
     st.divider()