Update modules/database.py

modules/database.py

@@ -5,13 +5,13 @@ Banco de dados SQLite para Akira IA.
 - Índices para performance
 - Suporte a pronomes por tom (formal, rude, casual)
 - WAL + retry para concorrência
+- MÉTODO recuperar_mensagens() ADICIONADO
 """
 import sqlite3
-import json
 import time
 import os
 from typing import Optional, List, Dict, Any, Tuple
-from loguru import logger  # <-- LOGURU (FUNCIONA NO HF)
+from loguru import logger
 
 class Database:
     def __init__(self, db_path: str):
@@ -289,11 +289,9 @@ class Database:
             logger.error(f"Erro na migração/índices: {e}")
 
     # ================================================================
-    # MÉTODOS PRINCIPAIS (mantidos, mas com DATETIME)
+    # MÉTODOS PRINCIPAIS
     # ================================================================
-    # ... [TODOS OS MÉTODOS DO ORIGINAL, apenas com DATETIME]
 
-    # Exemplo de método corrigido:
     def salvar_mensagem(self, usuario, mensagem, resposta, numero=None, is_reply=False, mensagem_original=None):
         try:
             cols = ['usuario', 'mensagem', 'resposta']
@@ -308,20 +306,37 @@ class Database:
             logger.warning(f"Fallback salvar_mensagem: {e}")
             self._execute_with_retry("INSERT INTO mensagens (usuario, mensagem, resposta) VALUES (?, ?, ?)", (usuario, mensagem, resposta), commit=True)
 
-    # ... [demais métodos mantidos iguais, apenas com DATETIME onde necessário]
+    def recuperar_mensagens(self, usuario: str, limite: int = 5) -> List[Tuple]:
+        """MÉTODO ADICIONADO — usado pelo Contexto"""
+        return self._execute_with_retry(
+            "SELECT mensagem, resposta FROM mensagens WHERE usuario=? OR numero=? ORDER BY id DESC LIMIT ?",
+            (usuario, usuario, limite)
+        ) or []
+
+    def salvar_giria_aprendida(self, numero_usuario: str, giria: str, significado: str, contexto: Optional[str] = None):
+        existing = self._execute_with_retry("SELECT id, frequencia FROM girias_aprendidas WHERE numero_usuario=? AND giria=?", (numero_usuario, giria))
+        if existing:
+            self._execute_with_retry("UPDATE girias_aprendidas SET frequencia=frequencia+1, updated_at=CURRENT_TIMESTAMP WHERE id=?", (existing[0][0],), commit=True)
+        else:
+            self._execute_with_retry("INSERT INTO girias_aprendidas (numero_usuario, giria, significado, contexto) VALUES (?, ?, ?, ?)", (numero_usuario, giria, significado, contexto), commit=True)
+
+    def recuperar_girias_usuario(self, numero_usuario: str) -> List[Dict[str, Any]]:
+        result = self._execute_with_retry("SELECT giria, significado, contexto, frequencia FROM girias_aprendidas WHERE numero_usuario=? ORDER BY frequencia DESC, updated_at DESC", (numero_usuario,))
+        return [{"giria": r[0], "significado": r[1], "contexto": r[2], "frequencia": r[3]} for r in result] if result else []
+
+    def obter_tom_predominante(self, numero_usuario: str, limite: int = 10) -> Optional[str]:
+        result = self._execute_with_retry("""
+            SELECT tom_detectado, COUNT(*) FROM tom_usuario
+            WHERE numero_usuario=? GROUP BY tom_detectado
+            ORDER BY COUNT(*) DESC LIMIT 1
+        """, (numero_usuario,))
+        return result[0][0] if result else None
+
+    def registrar_tom_usuario(self, numero_usuario: str, tom_detectado: str, intensidade: float = 0.5, contexto: Optional[str] = None):
+        self._execute_with_retry("INSERT INTO tom_usuario (numero_usuario, tom_detectado, intensidade, contexto) VALUES (?, ?, ?, ?)", (numero_usuario, tom_detectado, intensidade, contexto), commit=True)
 
     def obter_pronomes_por_tom(self, tom: str) -> str:
         result = self._execute_with_retry("SELECT pronomes FROM pronomes_por_tom WHERE tom=?", (tom.lower(),))
         return result[0][0] if result else ""
 
-    def obter_contexto_aprendizado(self, numero_usuario: str) -> Dict[str, Any]:
-        return {
-            "girias": self.recuperar_girias_usuario(numero_usuario),
-            "tom_predominante": self.obter_tom_predominante(numero_usuario),
-            "estilo": self.recuperar_estilo(numero_usuario),
-            "preferencia_tom": self.recuperar_preferencia_tom(numero_usuario),
-            "afinidade": self.recuperar_afinidade(numero_usuario),
-            "adaptacoes_recentes": self.obter_adaptacoes_usuario(numero_usuario)[-5:],
-            "aprendizados_detalhados": dict(self.recuperar_aprendizado_detalhado(numero_usuario) or []),
-            "pronomes_tom": self.obter_pronomes_por_tom(self.obter_tom_predominante(numero_usuario) or "neutro")
-        }
+    # ... [demais métodos: salvar_info_geral, etc.]