Update modules/database.py

modules/database.py

@@ -1,6 +1,7 @@
 """
 Banco de dados SQLite para Akira IA.
 Gerencia contexto, mensagens, embeddings, gírias, tom e aprendizados detalhados.
+Versão completa 11/2025.
 """
 
 import sqlite3
@@ -295,11 +296,8 @@ class Database:
             (usuario, usuario, limite)
         ) or []
 
-    def salvar_embedding(self, numero_usuario: str, mensagem: str, resposta: str, embedding: bytes):
-        """Salva embedding no banco junto com mensagem e resposta."""
+    def salvar_embedding(self, texto: str, embedding: bytes):
         try:
-            texto = f"{numero_usuario}: {mensagem} | {resposta}"
-            # converte numpy array para bytes, se necessário
             if hasattr(embedding, "tobytes"):
                 embedding = embedding.tobytes()
             self._execute_with_retry(
@@ -310,11 +308,58 @@ class Database:
         except Exception as e:
             logger.warning(f"Erro ao salvar embedding: {e}")
 
-    # Outros métodos (girias, contexto, tom, aprendizados) permanecem iguais ao original
-    # Exemplo:
+    # ================================================================
+    # MÉTODOS ADICIONAIS PARA CONTEXTO / APRENDIZADOS / GIRIAS
+    # ================================================================
+    def recuperar_girias_usuario(self, numero_usuario: str) -> List[Dict[str, Any]]:
+        rows = self._execute_with_retry(
+            "SELECT giria, significado, contexto, frequencia FROM girias_aprendidas WHERE numero_usuario=?",
+            (numero_usuario,)
+        ) or []
+        return [{'giria': r[0], 'significado': r[1], 'contexto': r[2], 'frequencia': r[3]} for r in rows]
+
+    def recuperar_aprendizado_detalhado(self, numero_usuario: str) -> Dict[str, str]:
+        rows = self._execute_with_retry(
+            "SELECT chave, valor FROM aprendizados WHERE numero_usuario=?",
+            (numero_usuario,)
+        ) or []
+        return {r[0]: r[1] for r in rows}
+
+    def obter_tom_predominante(self, numero_usuario: str) -> Optional[str]:
+        rows = self._execute_with_retry(
+            "SELECT tom_detectado FROM tom_usuario WHERE numero_usuario=? ORDER BY created_at DESC LIMIT 1",
+            (numero_usuario,)
+        ) or []
+        return rows[0][0] if rows 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 salvar_aprendizado_detalhado(self, numero_usuario: str, chave: str, valor: str):
+        self._execute_with_retry(
+            "INSERT INTO aprendizados (numero_usuario, chave, valor) VALUES (?, ?, ?)",
+            (numero_usuario, chave, valor),
+            commit=True
+        )
+
+    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
+            )