Update modules/database.py

modules/database.py

@@ -3,7 +3,6 @@ Banco de dados SQLite para Akira IA.
 Gerencia contexto, mensagens, embeddings, gírias, tom e aprendizados detalhados.
 Versão 11/2025 - compatível com treinamento.py (SentenceTransformers)
 """
-
 import sqlite3
 import time
 import os
@@ -118,19 +117,17 @@ class Database:
                     girias TEXT,
                     tom TEXT
                 );
-                
-                -- ADICIONADO: Tabela para resolver o erro "no such table"
                 CREATE TABLE IF NOT EXISTS pronomes_por_tom (
                     tom TEXT PRIMARY KEY,
                     pronomes TEXT
                 );
                 """);
-                
-                # ADICIONADO: Inserção de dados padrão para a nova tabela
+
+                # Dados padrão
                 c.execute("INSERT OR IGNORE INTO pronomes_por_tom (tom, pronomes) VALUES (?, ?)", ('neutro', 'tu/você'))
                 c.execute("INSERT OR IGNORE INTO pronomes_por_tom (tom, pronomes) VALUES (?, ?)", ('formal', 'o senhor/a senhora'))
                 c.execute("INSERT OR IGNORE INTO pronomes_por_tom (tom, pronomes) VALUES (?, ?)", ('informal', 'puto/kota'))
-                
+
                 conn.commit()
                 logger.info(f"Banco inicializado: {self.db_path}")
         except Exception as e:
@@ -234,7 +231,7 @@ class Database:
     # ================================================================
     def salvar_contexto(self, user_key, historico, emocao_atual, termos, girias, tom):
         self._execute_with_retry(
-            """INSERT OR REPLACE INTO contexto 
+            """INSERT OR REPLACE INTO contexto
                (user_key, historico, emocao_atual, termos, girias, tom)
                VALUES (?, ?, ?, ?, ?, ?)""",
             (user_key, historico, emocao_atual, termos, girias, tom),
@@ -285,4 +282,34 @@ class Database:
                 "INSERT INTO girias_aprendidas (numero_usuario, giria, significado, contexto) VALUES (?, ?, ?, ?)",
                 (numero_usuario, giria, significado, contexto),
                 commit=True
-            )
+            )
+
+    # ================================================================
+    # MÉTODO ADICIONADO: recuperar_girias_usuario
+    # ================================================================
+    def recuperar_girias_usuario(self, numero_usuario: str) -> List[Dict[str, Any]]:
+        """
+        Recupera todas as gírias aprendidas para um usuário.
+        Retorna: [{'giria': 'baza', 'significado': 'ir embora', 'frequencia': 5}, ...]
+        """
+        try:
+            rows = self._execute_with_retry(
+                """
+                SELECT giria, significado, frequencia, contexto
+                FROM girias_aprendidas
+                WHERE numero_usuario = ?
+                ORDER BY frequencia DESC, updated_at DESC
+                """,
+                (numero_usuario,)
+            )
+            return [
+                {
+                    'giria': row[0],
+                    'significado': row[1],
+                    'frequencia': row[2],
+                    'contexto': row[3]
+                } for row in rows
+            ] if rows else []
+        except Exception as e:
+            logger.warning(f"Falha ao carregar gírias do usuário {numero_usuario}: {e}")
+            return []