Update modules/database.py

modules/database.py

@@ -6,13 +6,16 @@ Banco de dados SQLite para Akira IA.
 - Suporte a pronomes por tom (formal, rude, casual)
 - WAL + retry para concorrência
 - MÉTODOS: recuperar_mensagens, recuperar_aprendizado_detalhado, analisar_emocoes_mensagem
+- ADICIONADO: carregar_contexto, salvar_contexto, listar_girias_aprendidas
 """
 import sqlite3
 import time
 import os
+import json
 from typing import Optional, List, Dict, Any, Tuple
 from loguru import logger
 
+
 class Database:
     def __init__(self, db_path: str):
         self.db_path = db_path
@@ -202,6 +205,15 @@ class Database:
                         tom TEXT PRIMARY KEY,
                         pronomes TEXT NOT NULL
                     );
+                    -- CONTEXTO POR USUÁRIO (NOVO!)
+                    CREATE TABLE IF NOT EXISTS contexto (
+                        user_key TEXT PRIMARY KEY,
+                        historico TEXT,
+                        emocao_atual TEXT,
+                        termos TEXT,
+                        girias TEXT,
+                        tom TEXT
+                    );
                 ''')
                 # Insere pronomes padrão
                 c.executescript('''
@@ -236,6 +248,13 @@ class Database:
                     'tom_usuario': [
                         ("intensidade", "REAL DEFAULT 0.5"),
                         ("contexto", "TEXT")
+                    ],
+                    'contexto': [
+                        ("historico", "TEXT"),
+                        ("emocao_atual", "TEXT"),
+                        ("termos", "TEXT"),
+                        ("girias", "TEXT"),
+                        ("tom", "TEXT")
                     ]
                 }
                 for table, cols in migrations.items():
@@ -256,7 +275,8 @@ class Database:
                     "CREATE INDEX IF NOT EXISTS idx_tom_usuario ON tom_usuario(numero_usuario);",
                     "CREATE INDEX IF NOT EXISTS idx_aprendizados_usuario ON aprendizados(numero_usuario);",
                     "CREATE INDEX IF NOT EXISTS idx_embeddings_texto ON embeddings(texto);",
-                    "CREATE INDEX IF NOT EXISTS idx_pronomes_tom ON pronomes_por_tom(tom);"
+                    "CREATE INDEX IF NOT EXISTS idx_pronomes_tom ON pronomes_por_tom(tom);",
+                    "CREATE INDEX IF NOT EXISTS idx_contexto_user ON contexto(user_key);"
                 ]
                 for idx in indexes:
                     try:
@@ -268,7 +288,55 @@ class Database:
             logger.error(f"Erro na migração/índices: {e}")
 
     # ================================================================
-    # MÉTODOS PRINCIPAIS
+    # MÉTODOS OBRIGATÓRIOS (ADICIONADOS!)
+    # ================================================================
+
+    def carregar_contexto(self, user_key: str) -> Dict[str, Any]:
+        """Carrega contexto do usuário (para Contexto class)"""
+        try:
+            result = self._execute_with_retry(
+                "SELECT historico, emocao_atual, termos, girias, tom FROM contexto WHERE user_key = ?",
+                (user_key,), fetch=True
+            )
+            if result:
+                row = result[0]
+                return {
+                    "historico": row[0] or "[]",
+                    "emocao_atual": row[1] or "neutra",
+                    "termos": row[2] or "[]",
+                    "girias": row[3] or "[]",
+                    "tom": row[4] or "neutro"
+                }
+            return {}
+        except Exception as e:
+            logger.warning(f"Erro ao carregar contexto: {e}")
+            return {}
+
+    def salvar_contexto(self, user_key: str, historico: str, emocao_atual: str,
+                        termos: str, girias: str, tom: str):
+        """Salva contexto do usuário"""
+        try:
+            self._execute_with_retry("""
+                INSERT OR REPLACE INTO contexto (user_key, historico, emocao_atual, termos, girias, tom)
+                VALUES (?, ?, ?, ?, ?, ?)
+            """, (user_key, historico, emocao_atual, termos, girias, tom), commit=True)
+        except Exception as e:
+            logger.error(f"Erro ao salvar contexto: {e}")
+
+    def listar_girias_aprendidas(self, user_key: str) -> List[Tuple[str, str]]:
+        """Retorna lista de (giria, significado)"""
+        try:
+            result = self._execute_with_retry(
+                "SELECT giria, significado FROM girias_aprendidas WHERE numero_usuario = ? ORDER BY frequencia DESC, updated_at DESC LIMIT 10",
+                (user_key,), fetch=True
+            )
+            return result or []
+        except Exception as e:
+            logger.warning(f"Erro ao listar gírias: {e}")
+            return []
+
+    # ================================================================
+    # MÉTODOS PRINCIPAIS (JÁ EXISTENTES)
     # ================================================================
     def salvar_mensagem(self, usuario, mensagem, resposta, numero=None, is_reply=False, mensagem_original=None):
         try: