Update modules/database.py

modules/database.py

@@ -1,7 +1,8 @@
-# modules/database.py — AKIRA V21 FINAL CORRIGIDO (Dezembro 2025)
+# modules/database.py — AKIRA V21 FINAL CORRIGIDO (Dezembro 2025) - CORREÇÃO: message_id
 """
 ✅ TOTALMENTE ADAPTADO ao index.js atualizado
-✅ CORREÇÃO: Métodos corretos para api.py, contexto.py, treinamento.py
+✅ CORREÇÃO: Problema com message_id UNIQUE resolvido
+✅ Métodos corretos para api.py, contexto.py, treinamento.py
 ✅ Estrutura completa com reply_metadata
 ✅ Todos os métodos necessários implementados
 """
@@ -11,6 +12,7 @@ import time
 import os
 import json
 import hashlib
+import random
 from datetime import datetime
 from typing import Optional, List, Dict, Any, Tuple
 from loguru import logger
@@ -67,12 +69,12 @@ class Database:
                 raise
 
     def _init_db(self):
-        """Cria todas as tabelas necessárias"""
+        """Cria todas as tabelas necessárias - CORREÇÃO: message_id sem UNIQUE"""
         try:
             with self._get_connection() as conn:
                 c = conn.cursor()
                 
-                # Tabela principal de mensagens
+                # Tabela principal de mensagens - CORREÇÃO: message_id sem UNIQUE
                 c.execute('''
                     CREATE TABLE IF NOT EXISTS mensagens (
                         id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -112,7 +114,7 @@ class Database:
                         comando_executado TEXT,
                         has_media BOOLEAN DEFAULT 0,
                         media_type TEXT DEFAULT '',
-                        message_id TEXT UNIQUE,
+                        message_id TEXT,  -- CORREÇÃO: Removido UNIQUE
                         bot_response_time_ms INTEGER DEFAULT 0,
                         is_mention BOOLEAN DEFAULT 0,
                         timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
@@ -120,6 +122,18 @@ class Database:
                     )
                 ''')
                 
+                # Índice para message_id para performance (sem UNIQUE)
+                c.execute('''
+                    CREATE INDEX IF NOT EXISTS idx_mensagens_message_id 
+                    ON mensagens(message_id)
+                ''')
+                
+                # Índice para busca por número
+                c.execute('''
+                    CREATE INDEX IF NOT EXISTS idx_mensagens_numero 
+                    ON mensagens(numero)
+                ''')
+                
                 # Usuários privilegiados
                 c.execute('''
                     CREATE TABLE IF NOT EXISTS usuarios_privilegiados (
@@ -257,7 +271,7 @@ class Database:
                 ''')
                 
                 conn.commit()
-                logger.info("✅ Tabelas criadas/verificadas")
+                logger.info("✅ Tabelas criadas/verificadas com message_id corrigido")
                 
         except Exception as e:
             logger.error(f"❌ Erro ao criar tabelas: {e}")
@@ -284,7 +298,7 @@ class Database:
                     ("is_mention", "BOOLEAN DEFAULT 0"),
                     ("has_media", "BOOLEAN DEFAULT 0"),
                     ("media_type", "TEXT DEFAULT ''"),
-                    ("message_id", "TEXT UNIQUE"),
+                    ("message_id", "TEXT"),  # CORREÇÃO: Sem UNIQUE aqui também
                     ("bot_response_time_ms", "INTEGER DEFAULT 0")
                 ]
                 
@@ -297,10 +311,10 @@ class Database:
                 conn.commit()
                 
         except Exception as e:
-            logger.warning(f"Erro ao verificar colunas: {e}")
+            logger.warning(f"⚠️ Erro ao verificar colunas: {e}")
 
     # ========================================================================
-    # MÉTODOS DE SALVAMENTO (ADAPTADOS AO INDEX.JS)
+    # MÉTODOS DE SALVAMENTO (ADAPTADOS AO INDEX.JS) - CORRIGIDOS
     # ========================================================================
     
     def salvar_mensagem(self, usuario: str, mensagem: str, resposta: str,
@@ -327,7 +341,7 @@ class Database:
                        message_id: str = None,
                        bot_response_time_ms: int = 0,
                        is_mention: bool = False) -> bool:
-        """Salva mensagem no banco - COMPATÍVEL COM INDEX.JS"""
+        """Salva mensagem no banco - CORREÇÃO: message_id sem UNIQUE constraint"""
         try:
             numero_final = str(numero or usuario).strip()
             contexto_id = self._gerar_contexto_id(numero_final, tipo_conversa)
@@ -336,39 +350,85 @@ class Database:
             if isinstance(reply_info_json, dict):
                 reply_info_json = json.dumps(reply_info_json, ensure_ascii=False)
             
-            # Gera message_id se não fornecido
+            # CORREÇÃO: Gera message_id único se não fornecido
             if not message_id:
-                message_id = f"{numero_final}_{int(time.time() * 1000)}"
+                timestamp = int(time.time() * 1000)
+                random_suffix = random.randint(1000, 9999)
+                message_id = f"{numero_final}_{timestamp}_{random_suffix}"
             
-            self._execute_with_retry(
-                """
-                INSERT INTO mensagens
-                (usuario, usuario_nome, mensagem, resposta, numero, contexto_id, tipo_contexto,
-                 tipo_conversa, tipo_mensagem, is_reply, mensagem_original, mensagem_citada_limpa,
-                 reply_to_bot, reply_info_json, humor, modo_resposta, emocao_detectada, 
-                 confianca_emocao, grupo_id, grupo_nome, audio_transcricao, fonte_stt, 
-                 confianca_stt, comando_executado, has_media, media_type, message_id,
-                 bot_response_time_ms, is_mention)
-                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-                """,
-                (
-                    usuario[:50], usuario_nome[:100] or usuario[:100], 
-                    mensagem[:4000], resposta[:4000], numero_final,
-                    contexto_id, tipo_conversa, tipo_conversa, tipo_mensagem,
-                    int(is_reply), mensagem_original, mensagem_citada_limpa,
-                    int(reply_to_bot), reply_info_json, humor, modo_resposta,
-                    emocao_detectada, confianca_emocao, grupo_id[:50], grupo_nome[:100],
-                    audio_transcricao[:2000] if audio_transcricao else None,
-                    fonte_stt[:50], confianca_stt, comando_executado[:100] if comando_executado else None,
-                    int(has_media), media_type[:50], message_id[:100],
-                    bot_response_time_ms, int(is_mention)
-                ),
-                commit=True,
-                fetch=False
-            )
+            # CORREÇÃO: Adiciona um sufixo aleatório extra para garantir unicidade
+            # mesmo que o timestamp seja o mesmo
+            unique_suffix = random.randint(100, 999)
+            message_id = f"{message_id}_{unique_suffix}"
             
-            logger.debug(f"✅ Mensagem salva: {numero_final} | {tipo_mensagem}")
-            return True
+            try:
+                self._execute_with_retry(
+                    """
+                    INSERT INTO mensagens
+                    (usuario, usuario_nome, mensagem, resposta, numero, contexto_id, tipo_contexto,
+                     tipo_conversa, tipo_mensagem, is_reply, mensagem_original, mensagem_citada_limpa,
+                     reply_to_bot, reply_info_json, humor, modo_resposta, emocao_detectada, 
+                     confianca_emocao, grupo_id, grupo_nome, audio_transcricao, fonte_stt, 
+                     confianca_stt, comando_executado, has_media, media_type, message_id,
+                     bot_response_time_ms, is_mention)
+                    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                    """,
+                    (
+                        usuario[:50], usuario_nome[:100] or usuario[:100], 
+                        mensagem[:4000], resposta[:4000], numero_final,
+                        contexto_id, tipo_conversa, tipo_conversa, tipo_mensagem,
+                        int(is_reply), mensagem_original, mensagem_citada_limpa,
+                        int(reply_to_bot), reply_info_json, humor, modo_resposta,
+                        emocao_detectada, confianca_emocao, grupo_id[:50], grupo_nome[:100],
+                        audio_transcricao[:2000] if audio_transcricao else None,
+                        fonte_stt[:50], confianca_stt, comando_executado[:100] if comando_executado else None,
+                        int(has_media), media_type[:50], message_id[:200],
+                        bot_response_time_ms, int(is_mention)
+                    ),
+                    commit=True,
+                    fetch=False
+                )
+                
+                logger.debug(f"✅ Mensagem salva: {numero_final} | Message ID: {message_id}")
+                return True
+                
+            except sqlite3.IntegrityError as e:
+                # CORREÇÃO: Se ainda houver erro de UNIQUE (pode ser de outra coluna)
+                if "UNIQUE constraint failed" in str(e):
+                    logger.warning(f"🔄 Erro de UNIQUE, gerando novo message_id")
+                    # Gera um novo message_id completamente diferente
+                    new_message_id = f"{numero_final}_{int(time.time() * 1000)}_{random.randint(10000, 99999)}"
+                    
+                    self._execute_with_retry(
+                        """
+                        INSERT INTO mensagens
+                        (usuario, usuario_nome, mensagem, resposta, numero, contexto_id, tipo_contexto,
+                         tipo_conversa, tipo_mensagem, is_reply, mensagem_original, mensagem_citada_limpa,
+                         reply_to_bot, reply_info_json, humor, modo_resposta, emocao_detectada, 
+                         confianca_emocao, grupo_id, grupo_nome, audio_transcricao, fonte_stt, 
+                         confianca_stt, comando_executado, has_media, media_type, message_id,
+                         bot_response_time_ms, is_mention)
+                        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                        """,
+                        (
+                            usuario[:50], usuario_nome[:100] or usuario[:100], 
+                            mensagem[:4000], resposta[:4000], numero_final,
+                            contexto_id, tipo_conversa, tipo_conversa, tipo_mensagem,
+                            int(is_reply), mensagem_original, mensagem_citada_limpa,
+                            int(reply_to_bot), reply_info_json, humor, modo_resposta,
+                            emocao_detectada, confianca_emocao, grupo_id[:50], grupo_nome[:100],
+                            audio_transcricao[:2000] if audio_transcricao else None,
+                            fonte_stt[:50], confianca_stt, comando_executado[:100] if comando_executado else None,
+                            int(has_media), media_type[:50], new_message_id[:200],
+                            bot_response_time_ms, int(is_mention)
+                        ),
+                        commit=True,
+                        fetch=False
+                    )
+                    logger.debug(f"✅ Mensagem salva com novo message_id: {new_message_id}")
+                    return True
+                else:
+                    raise
             
         except Exception as e:
             logger.error(f"❌ Erro ao salvar mensagem: {e}")
@@ -616,7 +676,8 @@ class Database:
                 (numero, comando, parametros, sucesso, resposta, tipo_conversa, grupo_id)
                 VALUES (?, ?, ?, ?, ?, ?, ?)
                 """,
-                (str(numero).strip(), comando, parametros, int(sucesso), resposta, tipo_conversa, grupo_id),
+                (
+                    str(numero).strip(), comando, parametros, int(sucesso), resposta, tipo_conversa, grupo_id),
                 commit=True,
                 fetch=False
             )