Update modules/database.py

modules/database.py

@@ -1,3 +1,13 @@
+# modules/database.py
+"""
+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
+"""
+
 import sqlite3
 import json
 import time
@@ -74,7 +84,6 @@ class Database:
                     dado TEXT,
                     valor TEXT
                 );
-
                 -- exemplos
                 CREATE TABLE IF NOT EXISTS exemplos (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -82,31 +91,26 @@ class Database:
                     entrada TEXT NOT NULL,
                     resposta TEXT NOT NULL
                 );
-
                 -- info_geral
                 CREATE TABLE IF NOT EXISTS info_geral (
                     chave TEXT PRIMARY KEY,
                     valor TEXT NOT NULL
                 );
-
                 -- estilos
                 CREATE TABLE IF NOT EXISTS estilos (
                     numero_usuario TEXT PRIMARY KEY,
                     estilo TEXT NOT NULL
                 );
-
                 -- preferencias_tom
                 CREATE TABLE IF NOT EXISTS preferencias_tom (
                     numero_usuario TEXT PRIMARY KEY,
                     tom TEXT NOT NULL
                 );
-
                 -- afinidades
                 CREATE TABLE IF NOT EXISTS afinidades (
                     numero_usuario TEXT PRIMARY KEY,
                     afinidade REAL NOT NULL
                 );
-
                 -- termos
                 CREATE TABLE IF NOT EXISTS termos (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -114,7 +118,6 @@ class Database:
                     termo TEXT NOT NULL,
                     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
                 );
-
                 -- aprendizados (detalhado)
                 CREATE TABLE IF NOT EXISTS aprendizados (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -123,7 +126,6 @@ class Database:
                     valor TEXT NOT NULL,
                     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
                 );
-
                 -- vocabulário patenteado
                 CREATE TABLE IF NOT EXISTS vocabulario_patenteado (
                     termo TEXT PRIMARY KEY,
@@ -131,13 +133,11 @@ class Database:
                     uso TEXT NOT NULL,
                     exemplo TEXT NOT NULL
                 );
-
                 -- usuarios privilegiados
                 CREATE TABLE IF NOT EXISTS usuarios_privilegiados (
                     numero_usuario TEXT PRIMARY KEY,
                     nome TEXT NOT NULL
                 );
-
                 -- whatsapp_ids
                 CREATE TABLE IF NOT EXISTS whatsapp_ids (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -145,14 +145,12 @@ class Database:
                     sender_number TEXT NOT NULL,
                     UNIQUE (whatsapp_id, sender_number)
                 );
-
                 -- embeddings
                 CREATE TABLE IF NOT EXISTS embeddings (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
                     texto TEXT NOT NULL,
                     embedding BLOB NOT NULL
                 );
-
                 -- mensagens
                 CREATE TABLE IF NOT EXISTS mensagens (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -165,7 +163,6 @@ class Database:
                     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                     data TIMESTAMP DEFAULT CURRENT_TIMESTAMP
                 );
-
                 -- emocao_exemplos
                 CREATE TABLE IF NOT EXISTS emocao_exemplos (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -175,7 +172,6 @@ class Database:
                     tom TEXT NOT NULL,
                     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
                 );
-
                 -- girias_aprendidas
                 CREATE TABLE IF NOT EXISTS girias_aprendidas (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -187,7 +183,6 @@ class Database:
                     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                     updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
                 );
-
                 -- tom_usuario
                 CREATE TABLE IF NOT EXISTS tom_usuario (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -197,7 +192,6 @@ class Database:
                     contexto TEXT,
                     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
                 );
-
                 -- adaptacao_dinamica
                 CREATE TABLE IF NOT EXISTS adaptacao_dinamica (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -206,14 +200,28 @@ class Database:
                     valor_anterior TEXT,
                     valor_novo TEXT,
                     razao TEXT,
-                    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+                    created_at TIMESTAMP DEFAULT CURRENT TIMESTAMP
+                );
+                -- NOVA TABELA: PRONOMES POR TOM
+                CREATE TABLE IF NOT EXISTS pronomes_por_tom (
+                    tom TEXT PRIMARY KEY,
+                    pronomes TEXT NOT NULL
                 );
             ''')
 
+            # Insere pronomes padrão (só na primeira vez)
+            c.executescript('''
+                INSERT OR IGNORE INTO pronomes_por_tom (tom, pronomes) VALUES
+                ('formal', 'Sr., ilustre, boss, maior, homem'),
+                ('rude', 'parvo, estúpido, burro, analfabeto, desperdício de esperma'),
+                ('casual', 'mano, puto, cota, mwangolé, kota'),
+                ('neutro', 'amigo, parceiro, camarada');
+            ''')
+
             conn.commit()
 
     def _ensure_all_columns_and_indexes(self):
-        """Garante que TODAS as colunas existam + cria índices para performance"""
+        """Garante que TODAS as colunas existam + cria índices"""
         with self._get_connection() as conn:
             c = conn.cursor()
 
@@ -256,7 +264,8 @@ class Database:
                 "CREATE INDEX IF NOT EXISTS idx_girias_giria ON girias_aprendidas(giria);",
                 "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_embeddings_texto ON embeddings(texto);",
+                "CREATE INDEX IF NOT EXISTS idx_pronomes_tom ON pronomes_por_tom(tom);"
             ]
 
             for idx in indexes:
@@ -268,7 +277,7 @@ class Database:
             conn.commit()
 
     # ================================================================
-    # MÉTODOS PRINCIPAIS (mantidos e otimizados)
+    # MÉTODOS PRINCIPAIS
     # ================================================================
 
     # --- EXEMPLOS ---
@@ -461,6 +470,12 @@ class Database:
             "palavras_chave_encontradas": max_count
         }
 
+    # --- PRONOMES POR TOM (NOVA FUNÇÃO) ---
+    def obter_pronomes_por_tom(self, tom: str) -> str:
+        """Retorna string com pronomes para o tom (ex: 'Sr., boss')"""
+        result = self._execute_with_retry("SELECT pronomes FROM pronomes_por_tom WHERE tom=?", (tom.lower(),))
+        return result[0][0] if result else ""
+
     # --- CONTEXTO COMPLETO ---
     def obter_contexto_aprendizado(self, numero_usuario: str) -> Dict[str, Any]:
         return {
@@ -470,5 +485,6 @@ class Database:
             "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 [])
+            "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")
         }