Update app.py

app.py

@@ -84,6 +84,7 @@ except Exception:
         return {
             "prod": "🟢 Produção",
             "test": "🔴 Teste",
+            "treinamento": "🟡 Treinamento",
             "train": "🟡 Treinamento",
         }.get(choice, "🟢 Produção")
 
@@ -705,20 +706,20 @@ def _db_choice_ui():
         # Obtém lista de choices
         if _get_available_choices:
             try:
-                avail = list(_get_available_choices())  # ex.: ["prod","test","train"]
+                avail = list(_get_available_choices())  # ex.: ["prod","test","treinamento"]
                 if not avail:
-                    avail = ["prod", "test", "train"]
+                    avail = ["prod", "test", "treinamento"]
             except Exception:
-                avail = ["prod", "test", "train"]
+                avail = ["prod", "test", "treinamento"]
         else:
-            avail = ["prod", "test", "train"]
+            avail = ["prod", "test", "treinamento"]
 
         # 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)
+                return {"prod": "🟢 Produção", "test": "🔴 Teste", "treinamento": "🟡 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]
@@ -778,6 +779,20 @@ 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)
+    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__"):
+                try:
+                    init_schema()
+                    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}")
+    except Exception:
+        # Se o módulo não expor init_schema, apenas ignora silenciosamente
+        pass
+
     # LOGIN
     if not st.session_state.logado:
         st.session_state.quiz_verificado = False
@@ -786,8 +801,6 @@ def main():
         login()
 
         # 🔕 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)