Update database.py

database.py

@@ -1,11 +1,13 @@
-# database.py — V23 — TODAS AS TABELAS DO V18 + NOVAS (POO)
+# database.py — V24 — COMPLETO COM salvar_embedding() + NUMPY + MIGRAÇÕES
 import sqlite3
 import time
 import os
 import json
+import numpy as np
 from typing import List, Dict, Any, Tuple
 from loguru import logger
 
+
 class Database:
     def __init__(self, db_path: str = "/app/akira.db"):
         self.db_path = db_path
@@ -189,7 +191,6 @@ class Database:
                     girias TEXT,
                     tom TEXT
                 );
-
                 -- TABELAS NOVAS (V22/V23)
                 CREATE TABLE IF NOT EXISTS abrev_usuarios (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -198,7 +199,6 @@ class Database:
                     palavra_completa TEXT,
                     frequencia INTEGER DEFAULT 1
                 );
-
                 -- DADOS INICIAIS
                 INSERT OR IGNORE INTO pronomes_por_tom (tom, pronomes) VALUES
                 ('formal', 'Sr., ilustre, boss, maior, homem'),
@@ -207,7 +207,7 @@ class Database:
                 ('neutro', 'amigo, parceiro, camarada');
             ''')
             conn.commit()
-        logger.info(f"Banco V23 inicializado: {self.db_path}")
+        logger.info(f"Banco V24 inicializado: {self.db_path}")
 
     # ================================================================
     # MIGRAÇÃO AUTOMÁTICA — GARANTE COLUNAS + ÍNDICES
@@ -224,7 +224,6 @@ class Database:
                 c.execute("ALTER TABLE embeddings ADD COLUMN source_type TEXT DEFAULT 'conversa'")
             if 'texto' not in cols:
                 c.execute("ALTER TABLE embeddings ADD COLUMN texto TEXT DEFAULT ''")
-
             # Índices
             c.executescript('''
                 CREATE INDEX IF NOT EXISTS idx_mensagens_numero ON mensagens(numero);
@@ -291,4 +290,34 @@ class Database:
         casual = any(w in mensagem.lower() for w in ["epá", "kandando", "bué", "kota"])
         if rude: return "rude"
         if casual: return "casual"
-        return "neutro"
+        return "neutro"
+
+    # ================================================================
+    # NOVO MÉTODO: salvar_embedding() — COM NUMPY
+    # ================================================================
+    def salvar_embedding(self, numero_usuario: str, mensagem: str, resposta: str, embedding: list, texto: str = ""):
+        """
+        Salva embedding no banco (tabela 'embeddings').
+        """
+        try:
+            embedding_blob = np.array(embedding, dtype=np.float32).tobytes()
+            
+            self._execute(
+                """INSERT INTO embeddings 
+                (numero_usuario, source_type, texto, embedding) 
+                VALUES (?, ?, ?, ?)""",
+                (numero_usuario, 'conversa', texto or f"{mensagem} {resposta}", embedding_blob),
+                commit=True
+            )
+            logger.debug(f"Embedding salvo para {numero_usuario}")
+        except Exception as e:
+            logger.error(f"Erro ao salvar embedding: {e}")
+
+
+# ================================================================
+# TESTE RÁPIDO (OPCIONAL)
+# ================================================================
+if __name__ == "__main__":
+    db = Database()
+    db.salvar_mensagem("Zé", "Oi", "Epá!", "244999123456")
+    print("Banco V24 pronto!")