Update modules/database.py

modules/database.py

@@ -1,4 +1,4 @@
-# modules/database.py — AKIRA V21 ULTIMATE (Dezembro 2025) - COMPLETO
+# modules/database.py — AKIRA V21 ULTIMATE (Dezembro 2025) - COMPLETO E ADAPTADO
 """
 Banco de dados SQLite para Akira IA - COMPLETAMENTE ATUALIZADO
 ✅ Sistema de usuários privilegiados completo
@@ -7,7 +7,9 @@ Banco de dados SQLite para Akira IA - COMPLETAMENTE ATUALIZADO
 ✅ Transições graduais de humor (3 níveis)
 ✅ Memória emocional BERT GoEmotions
 ✅ Backup e otimização automática
-✅ MÉTODOS FALTANTES ADICIONADOS: salvar_training_example, registrar_tom_usuario, salvar_aprendizado_detalhado (com numero_usuario)
+✅ ADAPTADO ao payload do index.js
+✅ NOVAS TABELAS: audio_transcricoes, grupo_contexto
+✅ NOVAS COLUNAS: tipo_mensagem, reply_info_json
 """
 import sqlite3
 import time
@@ -35,7 +37,7 @@ class Database:
         self._ensure_all_columns_and_indexes()
         self._sincronizar_usuarios_privilegiados() # Sincroniza com config
       
-        logger.info(f"✅ Banco V21 inicializado: {self.db_path}")
+        logger.info(f"✅ Banco V21 ADAPTADO inicializado: {self.db_path}")
 
     # ================================================================
     # ISOLAMENTO CRIPTOGRÁFICO (V21 MELHORADO)
@@ -43,7 +45,7 @@ class Database:
   
     def _gerar_contexto_id(self, numero: str, tipo: str = 'auto') -> str:
         if tipo == 'auto':
-            if "@g.us" in str(numero) or "120363" in str(numero):
+            if "@g.us" in str(numero) or "120363" in str(numero) or "grupo_" in str(numero):
                 tipo = "grupo"
             else:
                 tipo = "pv"
@@ -117,17 +119,48 @@ class Database:
             with self._get_connection() as conn:
                 c = conn.cursor()
               
+                # NOVAS COLUNAS PARA TABELA MENSAGENS
+                try:
+                    c.execute("ALTER TABLE mensagens ADD COLUMN tipo_mensagem TEXT DEFAULT 'texto'")
+                except sqlite3.OperationalError:
+                    pass  # Coluna já existe
+                
+                try:
+                    c.execute("ALTER TABLE mensagens ADD COLUMN reply_info_json TEXT")
+                except sqlite3.OperationalError:
+                    pass  # Coluna já existe
+                
+                try:
+                    c.execute("ALTER TABLE mensagens ADD COLUMN usuario_nome TEXT DEFAULT ''")
+                except sqlite3.OperationalError:
+                    pass  # Coluna já existe
+                
+                try:
+                    c.execute("ALTER TABLE mensagens ADD COLUMN grupo_id TEXT DEFAULT ''")
+                except sqlite3.OperationalError:
+                    pass  # Coluna já existe
+                
+                try:
+                    c.execute("ALTER TABLE mensagens ADD COLUMN grupo_nome TEXT DEFAULT ''")
+                except sqlite3.OperationalError:
+                    pass  # Coluna já existe
+
                 indexes = [
                     "CREATE INDEX IF NOT EXISTS idx_mensagens_numero ON mensagens(numero, deletado)",
                     "CREATE INDEX IF NOT EXISTS idx_mensagens_contexto ON mensagens(contexto_id)",
                     "CREATE INDEX IF NOT EXISTS idx_mensagens_timestamp ON mensagens(timestamp)",
+                    "CREATE INDEX IF NOT EXISTS idx_mensagens_tipo ON mensagens(tipo_mensagem)",
+                    "CREATE INDEX IF NOT EXISTS idx_mensagens_grupo ON mensagens(grupo_id, deletado)",
                     "CREATE INDEX IF NOT EXISTS idx_contexto_numero ON contexto(numero)",
+                    "CREATE INDEX IF NOT EXISTS idx_contexto_tipo ON contexto(tipo_contexto)",
                     "CREATE INDEX IF NOT EXISTS idx_usuarios_numero ON usuarios_privilegiados(numero)",
                     "CREATE INDEX IF NOT EXISTS idx_transicoes_numero ON transicoes_humor(numero, timestamp)",
                     "CREATE INDEX IF NOT EXISTS idx_girias_numero ON girias_aprendidas(numero, giria)",
                     "CREATE INDEX IF NOT EXISTS idx_training_qualidade ON training_examples(qualidade_score, usado)",
                     "CREATE INDEX IF NOT EXISTS idx_comandos_numero ON comandos_executados(numero, timestamp)",
-                    "CREATE INDEX IF NOT EXISTS idx_reset_numero ON reset_log(numero, timestamp)"
+                    "CREATE INDEX IF NOT EXISTS idx_reset_numero ON reset_log(numero, timestamp)",
+                    "CREATE INDEX IF NOT EXISTS idx_audio_usuario ON audio_transcricoes(numero_usuario, sucesso)",
+                    "CREATE INDEX IF NOT EXISTS idx_grupo_contexto_id ON grupo_contexto(grupo_id, contexto_id)"
                 ]
               
                 for idx_query in indexes:
@@ -143,7 +176,7 @@ class Database:
             logger.error(f"Erro ao verificar colunas/índices: {e}")
 
     # ================================================================
-    # INICIALIZAÇÃO COMPLETA
+    # INICIALIZAÇÃO COMPLETA COM NOVAS TABELAS
     # ================================================================
   
     def _init_db(self):
@@ -151,23 +184,28 @@ class Database:
             with self._get_connection() as conn:
                 c = conn.cursor()
               
-                # TABELA PRINCIPAL DE MENSAGENS
+                # TABELA PRINCIPAL DE MENSAGENS (ATUALIZADA)
                 c.execute('''
                     CREATE TABLE IF NOT EXISTS mensagens (
                         id INTEGER PRIMARY KEY AUTOINCREMENT,
                         usuario TEXT NOT NULL,
+                        usuario_nome TEXT DEFAULT '',
                         mensagem TEXT NOT NULL,
                         resposta TEXT NOT NULL,
                         numero TEXT NOT NULL,
                         contexto_id TEXT NOT NULL,
                         tipo_contexto TEXT DEFAULT 'pv',
+                        tipo_mensagem TEXT DEFAULT 'texto',
                         is_reply BOOLEAN DEFAULT 0,
                         mensagem_original TEXT,
                         reply_to_bot BOOLEAN DEFAULT 0,
+                        reply_info_json TEXT,
                         humor TEXT DEFAULT 'normal_ironico',
                         modo_resposta TEXT DEFAULT 'normal_ironico',
                         emocao_detectada TEXT,
                         confianca_emocao REAL DEFAULT 0.5,
+                        grupo_id TEXT DEFAULT '',
+                        grupo_nome TEXT DEFAULT '',
                         timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
                         deletado BOOLEAN DEFAULT 0
                     )
@@ -193,7 +231,7 @@ class Database:
                     )
                 ''')
               
-                # TABELA DE CONTEXTO
+                # TABELA DE CONTEXTO (ATUALIZADA)
                 c.execute('''
                     CREATE TABLE IF NOT EXISTS contexto (
                         id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -210,6 +248,26 @@ class Database:
                         tom TEXT DEFAULT 'normal',
                         emocao_tendencia TEXT DEFAULT 'neutral',
                         volatilidade REAL DEFAULT 0.5,
+                        nome_usuario TEXT DEFAULT '',
+                        ultimo_contato DATETIME,
+                        data_criacao DATETIME DEFAULT CURRENT_TIMESTAMP,
+                        data_atualizacao DATETIME DEFAULT CURRENT_TIMESTAMP
+                    )
+                ''')
+              
+                # NOVA TABELA: CONTEXTO DE GRUPO
+                c.execute('''
+                    CREATE TABLE IF NOT EXISTS grupo_contexto (
+                        id INTEGER PRIMARY KEY AUTOINCREMENT,
+                        grupo_id TEXT UNIQUE NOT NULL,
+                        contexto_id TEXT NOT NULL,
+                        grupo_nome TEXT DEFAULT '',
+                        numero_participantes INTEGER DEFAULT 0,
+                        historico_grupo TEXT,
+                        humor_medio TEXT DEFAULT 'normal_ironico',
+                        tom_medio TEXT DEFAULT 'normal',
+                        atividade_score REAL DEFAULT 0.5,
+                        ultima_atividade DATETIME DEFAULT CURRENT_TIMESTAMP,
                         data_criacao DATETIME DEFAULT CURRENT_TIMESTAMP,
                         data_atualizacao DATETIME DEFAULT CURRENT_TIMESTAMP
                     )
@@ -262,6 +320,21 @@ class Database:
                     )
                 ''')
               
+                # NOVA TABELA: TRANSCRIÇÕES DE ÁUDIO
+                c.execute('''
+                    CREATE TABLE IF NOT EXISTS audio_transcricoes (
+                        id INTEGER PRIMARY KEY AUTOINCREMENT,
+                        numero_usuario TEXT NOT NULL,
+                        audio_hash TEXT NOT NULL,
+                        texto_transcrito TEXT NOT NULL,
+                        fonte TEXT DEFAULT 'deepgram',
+                        sucesso BOOLEAN DEFAULT 1,
+                        confianca REAL DEFAULT 0.5,
+                        duracao_segundos REAL,
+                        timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
+                    )
+                ''')
+              
                 # TABELA DE COMANDOS EXECUTADOS
                 c.execute('''
                     CREATE TABLE IF NOT EXISTS comandos_executados (
@@ -290,7 +363,7 @@ class Database:
               
                 conn.commit()
               
-            logger.info("✅ Todas as tabelas criadas/verificadas")
+            logger.info("✅ Todas as tabelas criadas/verificadas (ADAPTADO)")
           
         except Exception as e:
             logger.error(f"❌ Erro ao criar tabelas: {e}")
@@ -526,7 +599,8 @@ class Database:
             tabelas_para_resetar = [
                 "mensagens", "contexto", "transicoes_humor",
                 "girias_aprendidas", "training_examples",
-                "comandos_executados", "reset_log"
+                "comandos_executados", "reset_log",
+                "audio_transcricoes", "grupo_contexto"
             ]
           
             for tabela in tabelas_para_resetar:
@@ -549,7 +623,7 @@ class Database:
             }
 
     # ================================================================
-    # MÉTODOS DE RECUPERAÇÃO (COM ISOLAMENTO)
+    # MÉTODOS DE RECUPERAÇÃO (COM ISOLAMENTO) - ADAPTADOS
     # ================================================================
   
     def recuperar_mensagens(self, numero: str, limite: int = 10) -> List[Tuple]:
@@ -557,7 +631,8 @@ class Database:
             results = self._execute_with_retry(
                 """
                 SELECT mensagem, resposta, is_reply, mensagem_original, reply_to_bot,
-                       humor, modo_resposta, emocao_detectada, confianca_emocao
+                       humor, modo_resposta, emocao_detectada, confianca_emocao,
+                       tipo_mensagem, reply_info_json
                 FROM mensagens
                 WHERE numero = ? AND deletado = 0
                 ORDER BY timestamp DESC
@@ -568,13 +643,18 @@ class Database:
             )
           
             if results:
-                return [
-                    (
+                mensagens = []
+                for r in results[::-1]:
+                    try:
+                        reply_info = json.loads(r[10]) if r[10] else None
+                    except:
+                        reply_info = None
+                    
+                    mensagens.append((
                         r[0], r[1], bool(r[2]), r[3], bool(r[4]),
-                        r[5], r[6], r[7], r[8]
-                    )
-                    for r in results[::-1]
-                ]
+                        r[5], r[6], r[7], r[8], r[9], reply_info
+                    ))
+                return mensagens
             return []
           
         except Exception as e:
@@ -609,7 +689,7 @@ class Database:
                 """
                 SELECT contexto_id, tipo_contexto, historico, humor_atual,
                        modo_resposta, nivel_transicao, humor_alvo, termos,
-                       girias, tom, emocao_tendencia, volatilidade
+                       girias, tom, emocao_tendencia, volatilidade, nome_usuario
                 FROM contexto
                 WHERE numero = ?
                 """,
@@ -631,7 +711,8 @@ class Database:
                     "girias": row[8],
                     "tom": row[9] or "normal",
                     "emocao_tendencia": row[10] or "neutral",
-                    "volatilidade": row[11] or 0.5
+                    "volatilidade": row[11] or 0.5,
+                    "nome_usuario": row[12] or ""
                 }
           
             return {
@@ -646,7 +727,8 @@ class Database:
                 "girias": "",
                 "tom": "normal",
                 "emocao_tendencia": "neutral",
-                "volatilidade": 0.5
+                "volatilidade": 0.5,
+                "nome_usuario": ""
             }
           
         except Exception as e:
@@ -663,7 +745,8 @@ class Database:
                 "girias": "",
                 "tom": "normal",
                 "emocao_tendencia": "neutral",
-                "volatilidade": 0.5
+                "volatilidade": 0.5,
+                "nome_usuario": ""
             }
 
     def recuperar_historico_humor(self, numero: str, limite: int = 5) -> List[Dict]:
@@ -768,7 +851,7 @@ class Database:
             logger.error(f"Erro ao marcar exemplos: {e}")
 
     # ================================================================
-    # MÉTODOS DE SALVAMENTO (COM CONTEXTO)
+    # MÉTODOS DE SALVAMENTO (COM CONTEXTO) - ADAPTADOS
     # ================================================================
   
     def salvar_mensagem(self, usuario: str, mensagem: str, resposta: str,
@@ -777,42 +860,102 @@ class Database:
                        humor: str = 'normal_ironico',
                        modo_resposta: str = 'normal_ironico',
                        emocao_detectada: str = None,
-                       confianca_emocao: float = 0.5):
+                       confianca_emocao: float = 0.5,
+                       tipo_mensagem: str = 'texto',
+                       reply_info: Dict = None,
+                       usuario_nome: str = '',
+                       grupo_id: str = '',
+                       grupo_nome: str = ''):
+        """Método adaptado para o payload do index.js"""
         try:
             numero_final = str(numero or usuario).strip()
             contexto_id = self._gerar_contexto_id(numero_final, 'auto')
-            tipo_contexto = "grupo" if ("@g.us" in numero_final or "120363" in numero_final) else "pv"
+            tipo_contexto = "grupo" if ("@g.us" in numero_final or "120363" in numero_final or grupo_id) else "pv"
+            
+            # Converte reply_info para JSON
+            reply_info_json = json.dumps(reply_info) if reply_info else None
+            
+            # Se for grupo, atualiza grupo_contexto
+            if grupo_id and tipo_contexto == "grupo":
+                self._atualizar_contexto_grupo(grupo_id, grupo_nome or "Sem nome", usuario_nome or usuario, mensagem[:100])
           
             self._execute_with_retry(
                 """
                 INSERT INTO mensagens
-                (usuario, mensagem, resposta, numero, contexto_id, tipo_contexto,
-                 is_reply, mensagem_original, reply_to_bot, humor, modo_resposta,
-                 emocao_detectada, confianca_emocao)
-                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                (usuario, usuario_nome, mensagem, resposta, numero, contexto_id, tipo_contexto,
+                 tipo_mensagem, is_reply, mensagem_original, reply_to_bot, reply_info_json,
+                 humor, modo_resposta, emocao_detectada, confianca_emocao, grupo_id, grupo_nome)
+                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                 """,
                 (
-                    usuario[:50], mensagem[:1000], resposta[:1000], numero_final,
-                    contexto_id, tipo_contexto, int(is_reply), mensagem_original,
-                    int(reply_to_bot), humor, modo_resposta,
-                    emocao_detectada, confianca_emocao
+                    usuario[:50], usuario_nome[:100] or usuario[:100], 
+                    mensagem[:1000], resposta[:1000], numero_final,
+                    contexto_id, tipo_contexto, tipo_mensagem, int(is_reply), 
+                    mensagem_original, int(reply_to_bot), reply_info_json,
+                    humor, modo_resposta, emocao_detectada, confianca_emocao,
+                    grupo_id[:50], grupo_nome[:100]
                 ),
                 commit=True,
                 fetch=False
             )
           
-            logger.debug(f"✅ Mensagem salva: {numero_final} ({tipo_contexto})")
+            logger.debug(f"✅ Mensagem salva: {numero_final} ({tipo_contexto}) | Tipo: {tipo_mensagem}")
             return True
           
         except Exception as e:
             logger.error(f"Erro ao salvar mensagem: {e}")
             return False
+
+    def _atualizar_contexto_grupo(self, grupo_id: str, grupo_nome: str, usuario: str, mensagem: str):
+        """Atualiza contexto de grupo"""
+        try:
+            contexto_id = self._gerar_contexto_id(grupo_id, 'grupo')
+            
+            # Verifica se já existe
+            result = self._execute_with_retry(
+                "SELECT 1 FROM grupo_contexto WHERE grupo_id = ?",
+                (grupo_id,),
+                fetch=True
+            )
+            
+            if result:
+                # Atualiza
+                self._execute_with_retry(
+                    """
+                    UPDATE grupo_contexto
+                    SET ultima_atividade = CURRENT_TIMESTAMP,
+                        data_atualizacao = CURRENT_TIMESTAMP
+                    WHERE grupo_id = ?
+                    """,
+                    (grupo_id,),
+                    commit=True,
+                    fetch=False
+                )
+            else:
+                # Insere novo
+                self._execute_with_retry(
+                    """
+                    INSERT INTO grupo_contexto
+                    (grupo_id, contexto_id, grupo_nome, ultima_atividade)
+                    VALUES (?, ?, ?, CURRENT_TIMESTAMP)
+                    """,
+                    (grupo_id, contexto_id, grupo_nome[:100]),
+                    commit=True,
+                    fetch=False
+                )
+            
+        except Exception as e:
+            logger.error(f"Erro ao atualizar contexto de grupo: {e}")
   
     def salvar_contexto(self, numero: str, **kwargs):
         try:
             numero_final = str(numero).strip()
             contexto_id = self._gerar_contexto_id(numero_final, 'auto')
             tipo_contexto = "grupo" if ("@g.us" in numero_final or "120363" in numero_final) else "pv"
+            
+            # Adiciona nome_usuario se fornecido
+            if 'nome_usuario' in kwargs:
+                kwargs['ultimo_contato'] = "CURRENT_TIMESTAMP"
           
             campos = {
                 "numero": numero_final,
@@ -941,6 +1084,63 @@ class Database:
         except Exception as e:
             logger.error(f"Erro ao salvar transição: {e}")
 
+    # ================================================================
+    # NOVOS MÉTODOS PARA TRANSCRIÇÕES DE ÁUDIO
+    # ================================================================
+    def salvar_transcricao_audio(self, numero_usuario: str, audio_hash: str, 
+                                texto_transcrito: str, fonte: str = "deepgram",
+                                sucesso: bool = True, confianca: float = 0.5,
+                                duracao_segundos: float = 0.0) -> bool:
+        """Salva transcrição de áudio"""
+        try:
+            self._execute_with_retry(
+                """
+                INSERT INTO audio_transcricoes
+                (numero_usuario, audio_hash, texto_transcrito, fonte, sucesso, confianca, duracao_segundos)
+                VALUES (?, ?, ?, ?, ?, ?, ?)
+                """,
+                (
+                    str(numero_usuario).strip(), audio_hash[:64], texto_transcrito[:5000],
+                    fonte[:50], int(sucesso), confianca, duracao_segundos
+                ),
+                commit=True,
+                fetch=False
+            )
+            logger.info(f"✅ Transcrição de áudio salva: {numero_usuario} | Sucesso: {sucesso}")
+            return True
+        except Exception as e:
+            logger.error(f"❌ Erro ao salvar transcrição: {e}")
+            return False
+    
+    def obter_historico_audio(self, numero_usuario: str, limite: int = 5) -> List[Dict]:
+        """Recupera histórico de transcrições de áudio"""
+        try:
+            results = self._execute_with_retry(
+                """
+                SELECT texto_transcrito, fonte, sucesso, confianca, timestamp
+                FROM audio_transcricoes
+                WHERE numero_usuario = ?
+                ORDER BY timestamp DESC
+                LIMIT ?
+                """,
+                (str(numero_usuario).strip(), limite),
+                fetch=True
+            )
+            
+            return [
+                {
+                    "texto": r[0],
+                    "fonte": r[1],
+                    "sucesso": bool(r[2]),
+                    "confianca": r[3],
+                    "timestamp": r[4]
+                }
+                for r in results
+            ]
+        except Exception as e:
+            logger.error(f"Erro ao recuperar histórico de áudio: {e}")
+            return []
+
     # ================================================================
     # MÉTODOS FALTANTES ADICIONADOS (RESOLVEM OS ERROS)
     # ================================================================
@@ -975,12 +1175,12 @@ class Database:
             self._execute_with_retry(
                 """
                 INSERT INTO mensagens
-                (usuario, mensagem, resposta, numero, contexto_id, tipo_contexto,
+                (usuario, usuario_nome, mensagem, resposta, numero, contexto_id, tipo_contexto,
                  humor, modo_resposta, emocao_detectada, confianca_emocao)
-                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                 """,
                 (
-                    "SISTEMA_TOM",
+                    "SISTEMA_TOM", "SISTEMA",
                     f"[TOM_DETECTADO: {tom}] {mensagem_contexto or ''}",
                     "",
                     numero_final,
@@ -997,7 +1197,8 @@ class Database:
             self.salvar_contexto(
                 numero=numero_final,
                 tom=tom,
-                emocao_tendencia=tom
+                emocao_tendencia=tom,
+                nome_usuario="SISTEMA"
             )
             logger.info(f"✅ Tom registrado: {tom} (confiança: {confianca:.2f}) para {numero_final}")
             return True
@@ -1041,6 +1242,81 @@ class Database:
             logger.error(f"❌ Erro ao salvar aprendizado detalhado: {e}")
             return False
 
+    # ================================================================
+    # MÉTODOS ADICIONAIS PARA O INDEX.JS
+    # ================================================================
+    def obter_estatisticas_usuario(self, numero: str) -> Dict:
+        """Retorna estatísticas do usuário"""
+        try:
+            # Conta mensagens
+            result = self._execute_with_retry(
+                "SELECT COUNT(*) FROM mensagens WHERE numero = ? AND deletado = 0",
+                (str(numero).strip(),),
+                fetch=True
+            )
+            total_mensagens = result[0][0] if result else 0
+            
+            # Última atividade
+            result = self._execute_with_retry(
+                "SELECT MAX(timestamp) FROM mensagens WHERE numero = ?",
+                (str(numero).strip(),),
+                fetch=True
+            )
+            ultima_atividade = result[0][0] if result else None
+            
+            # Transições de humor
+            result = self._execute_with_retry(
+                "SELECT COUNT(*) FROM transicoes_humor WHERE numero = ?",
+                (str(numero).strip(),),
+                fetch=True
+            )
+            transicoes_humor = result[0][0] if result else 0
+            
+            # Gírias aprendidas
+            result = self._execute_with_retry(
+                "SELECT COUNT(*) FROM girias_aprendidas WHERE numero = ?",
+                (str(numero).strip(),),
+                fetch=True
+            )
+            girias_aprendidas = result[0][0] if result else 0
+            
+            return {
+                "total_mensagens": total_mensagens,
+                "ultima_atividade": ultima_atividade,
+                "transicoes_humor": transicoes_humor,
+                "girias_aprendidas": girias_aprendidas,
+                "privilegiado": self.is_usuario_privilegiado(numero)
+            }
+            
+        except Exception as e:
+            logger.error(f"Erro ao obter estatísticas: {e}")
+            return {
+                "total_mensagens": 0,
+                "ultima_atividade": None,
+                "transicoes_humor": 0,
+                "girias_aprendidas": 0,
+                "privilegiado": False
+            }
+    
+    def limpar_mensagens_antigas(self, dias: int = 30) -> int:
+        """Limpa mensagens antigas"""
+        try:
+            result = self._execute_with_retry(
+                """
+                DELETE FROM mensagens 
+                WHERE timestamp < datetime('now', ?)
+                AND deletado = 0
+                """,
+                (f'-{dias} days',),
+                commit=True,
+                fetch=False
+            )
+            logger.info(f"🧹 Mensagens antigas ({dias} dias) limpas: {result} registros")
+            return result
+        except Exception as e:
+            logger.error(f"Erro ao limpar mensagens antigas: {e}")
+            return 0
+
     # ================================================================
     # FECHAMENTO SEGURO DA CONEXÃO
     # ================================================================