Update modules/database.py

modules/database.py

@@ -1,11 +1,6 @@
 """
 Banco de dados SQLite para Akira IA.
-- Tabelas: mensagens, girias, tom, pronomes, embeddings, etc.
-- Migração automática de colunas
-- Índices para performance
-- Suporte a pronomes por tom (formal, rude, casual)
-- WAL + retry para concorrência
-- MÉTODOS: carregar_contexto, salvar_contexto, listar_girias_aprendidas (SEM fetch!)
+[...]
 """
 import sqlite3
 import time
@@ -28,8 +23,10 @@ class Database:
     # CONEXÃO COM RETRY + WAL
     # ================================================================
     def _get_connection(self) -> sqlite3.Connection:
+        # ... (Mantido o código original)
         for attempt in range(self.max_retries):
             try:
+                # check_same_thread=False é necessário para WAL em threads
                 conn = sqlite3.connect(self.db_path, timeout=30.0, check_same_thread=False)
                 conn.execute('PRAGMA journal_mode=WAL')
                 conn.execute('PRAGMA synchronous=NORMAL')
@@ -47,7 +44,7 @@ class Database:
         raise sqlite3.OperationalError("Falha ao conectar após retries")
 
     def _execute_with_retry(self, query: str, params: Optional[tuple] = None, commit: bool = False) -> Optional[List[Tuple]]:
-        """Executa query com retry. Retorna fetchall() se SELECT."""
+        # ... (Mantido o código original)
         for attempt in range(self.max_retries):
             try:
                 with self._get_connection() as conn:
@@ -72,6 +69,7 @@ class Database:
     # INICIALIZAÇÃO + MIGRAÇÃO AUTOMÁTICA
     # ================================================================
     def _init_db(self):
+        # ... (Mantido o código original)
         try:
             with self._get_connection() as conn:
                 c = conn.cursor()
@@ -95,7 +93,7 @@ class Database:
                         chave TEXT PRIMARY KEY,
                         valor TEXT NOT NULL
                     );
-                    -- estilos
+                    -- estilos (OBS: esta tabela parece não ser usada pelo contexto/treinamento)
                     CREATE TABLE IF NOT EXISTS estilos (
                         numero_usuario TEXT PRIMARY KEY,
                         estilo TEXT NOT NULL
@@ -230,6 +228,7 @@ class Database:
             raise
 
     def _ensure_all_columns_and_indexes(self):
+        # ... (Mantido o código original)
         try:
             with self._get_connection() as conn:
                 c = conn.cursor()
@@ -293,11 +292,13 @@ class Database:
     def carregar_contexto(self, user_key: str) -> Dict[str, Any]:
         """Carrega contexto do usuário com correção JSON segura"""
         try:
+            # Seleciona as colunas na ordem esperada: historico, emocao_atual, termos, girias, tom
             result = self._execute_with_retry(
                 "SELECT historico, emocao_atual, termos, girias, tom FROM contexto WHERE user_key = ?",
                 (user_key,)
             )
 
+            # Define a função de carregamento seguro de JSON (como no original)
             def safe_json_load(v, default):
                 if not v:
                     return default
@@ -356,6 +357,7 @@ class Database:
     # MÉTODOS PRINCIPAIS
     # ================================================================
     def salvar_mensagem(self, usuario, mensagem, resposta, numero=None, is_reply=False, mensagem_original=None):
+        # ... (Mantido o código original)
         try:
             cols = ['usuario', 'mensagem', 'resposta']
             vals = [usuario, mensagem, resposta]
@@ -441,17 +443,27 @@ class Database:
     def obter_info_geral(self, chave: str) -> Optional[str]:
         result = self._execute_with_retry("SELECT valor FROM info_geral WHERE chave=?", (chave,))
         return result[0][0] if result else None
-
-    def analisar_emocoes_mensagem(self, mensagem: str) -> Dict[str, float]:
-        emocoes = {
-            "alegria": 0.5, "tristeza": 0.0, "raiva": 0.0,
-            "medo": 0.0, "surpresa": 0.0, "neutro": 0.5
-        }
-        msg = (mensagem or "").lower()
-        if any(p in msg for p in ["feliz", "bué fixe", "gosto", "amo", "kkk", "rsrs", "fixe"]):
-            emocoes["alegria"] = 0.9; emocoes["neutro"] = 0.1
-        elif any(p in msg for p in ["triste", "chateado", "puto", "raiva", "aff", "merda"]):
-            emocoes["tristeza"] = 0.8; emocoes["neutro"] = 0.2
-        elif any(p in msg for p in ["medo", "assustado", "tenso", "nervoso"]):
-            emocoes["medo"] = 0.8; emocoes["neutro"] = 0.2
-        return emocoes
+    
+    # MÉTODO ANALISAR_EMOCOES_MENSAGEM FOI REMOVIDO DA CLASSE DATABASE
+    # E PERMANECE NO CONTEXTO/MÓDULO DE NEGÓCIO.
+    def salvar_embedding(self, texto: str, embedding: bytes):
+        """Salva o embedding no banco (método adicionado por necessidade do treinamento.py)"""
+        try:
+            self._execute_with_retry(
+                "INSERT INTO embeddings (texto, embedding) VALUES (?, ?)",
+                (texto, embedding),
+                commit=True
+            )
+        except Exception as e:
+            logger.warning(f"Erro ao salvar embedding: {e}")
+            
+    def salvar_preferencia_tom(self, numero_usuario: str, tom: str):
+        """Salva a preferência de tom no banco (método adicionado por necessidade do treinamento.py)"""
+        try:
+            self._execute_with_retry(
+                "INSERT OR REPLACE INTO preferencias_tom (numero_usuario, tom) VALUES (?, ?)",
+                (numero_usuario, tom),
+                commit=True
+            )
+        except Exception as e:
+            logger.warning(f"Erro ao salvar preferencia de tom: {e}")