Update app.py

app.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 import os
+import io
 import streamlit as st
 from dotenv import load_dotenv
 from datetime import date, datetime, time
@@ -80,6 +81,49 @@ except Exception:
     def bank_label(choice: str) -> str:
         return "🟢 Produção" if choice == "prod" else "🔴 Teste"
 
+# =========================================
+# 🔧 STARTUP HOOKS (1x por container / process)
+# =========================================
+# - CLEAR_CACHE_ON_START=1 → limpa caches do Streamlit
+# - INIT_DB_ON_START=1 → tenta rodar init_db.run() e grava ~/.ioirun/.init_done
+_MARKER_DIR = os.path.join(os.path.expanduser("~"), ".ioirun")
+os.makedirs(_MARKER_DIR, exist_ok=True)
+_INIT_MARK = os.path.join(_MARKER_DIR, ".init_done")
+
+# Limpa cache no start (não depende de marker; roda a cada start quando flag=1)
+if os.getenv("CLEAR_CACHE_ON_START", "0") == "1":
+    try:
+        st.cache_data.clear()
+    except Exception:
+        pass
+    try:
+        st.cache_resource.clear()
+    except Exception:
+        pass
+
+# Rodar init_db.run() uma única vez por start de container
+try:
+    import init_db as _init_db
+    _HAS_INIT_DB = hasattr(_init_db, "run")
+except Exception:
+    _HAS_INIT_DB = False
+
+if os.getenv("INIT_DB_ON_START", "0") == "1" and _HAS_INIT_DB:
+    # Só roda se marker não existir OU se variável forçada estiver setada
+    force = os.getenv("FORCE_INIT_DB_EVERY_START", "0") == "1"
+    should_run = (not os.path.exists(_INIT_MARK)) or force
+    if should_run:
+        try:
+            _init_db.run()
+            with open(_INIT_MARK, "w", encoding="utf-8") as f:
+                f.write(f"init at {datetime.now().isoformat()}\n")
+        except Exception as e:
+            # no start, se falhar, apenas registra visualmente
+            try:
+                st.sidebar.warning(f"INIT_DB_ON_START falhou: {e}")
+            except Exception:
+                pass
+
 # ===============================
 # RERUN por querystring (atalho ?rr=1)
 # ===============================
@@ -537,7 +581,6 @@ def _show_birthday_banner_if_needed():
             unsafe_allow_html=True
         )
 
-
 # ===============================
 # 🔎 Painel de Diagnóstico de Autenticação (Admin)
 # ===============================
@@ -576,7 +619,6 @@ def _render_auth_diag_panel():
     except Exception as e:
         st.sidebar.error(f"Falha ao importar login.py: {e}")
 
-
 # ===============================
 # MAIN
 # ===============================
@@ -682,7 +724,6 @@ def main():
 
                         # 6) Amostra de usuários (sem senhas)
                         try:
-                            # tenta nomes comuns de colunas
                             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"))
@@ -700,7 +741,6 @@ def main():
 
                         if st.button("Testar bcrypt com este usuário", key="__btn_bcrypt_test__"):
                             try:
-                                # detectar nome da coluna de senha
                                 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]
                                 if not pass_cols or not user_cols:
@@ -735,6 +775,41 @@ def main():
                     except Exception:
                         pass
 
+            # ============================
+            # 🧰 Ações de manutenção (somente Admin)
+            # ============================
+            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):
+                    try:
+                        st.cache_data.clear()
+                    except Exception:
+                        pass
+                    try:
+                        st.cache_resource.clear()
+                    except Exception:
+                        pass
+                    st.sidebar.success("Caches limpos.")
+                    st.rerun()
+
+                # Rodar init_db.run() com dupla confirmação
+                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):
+                        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.")
+                            except Exception as e:
+                                st.error(f"Falha no init_db.run(): {e}")
+
     # 🔄 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])
@@ -974,7 +1049,7 @@ def main():
                 usuario=st.session_state.get("usuario"),
                 ambiente=ambiente_atual
             )
-            ja_nas_opcoes_r = any(mid == "repositorio_load" for mid, _ in (opcoes or []))
+            ja_nas_opcoes_r = any(mid == "repositorio_load" for mid, _ in (opcoes ou []))
             passa_busca_r = (not termo_busca) or (termo_busca in mod_repo.get("label", "").strip().lower())
             if mesmo_grupo_r and perfil_ok_r and not ja_nas_opcoes_r and passa_busca_r:
                 opcoes = (opcoes or []) + [("repositorio_load", mod_repo.get("label", "Repositório Load"))]