Update modules/database.py

modules/database.py

@@ -1,7 +1,8 @@
-# modules/database.py — AKIRA V21 FINAL CORRIGIDO (Dezembro 2025) - CORREÇÃO: message_id
+# modules/database.py — AKIRA V21 FINAL CORRIGIDO (Dezembro 2025) - CORREÇÃO: Com suporte a nivel_transicao
 """
 ✅ TOTALMENTE ADAPTADO ao index.js atualizado
 ✅ CORREÇÃO: Problema com message_id UNIQUE resolvido
+✅ CORREÇÃO: Suporte para nivel_transicao adicionado
 ✅ Métodos corretos para api.py, contexto.py, treinamento.py
 ✅ Estrutura completa com reply_metadata
 ✅ Todos os métodos necessários implementados
@@ -101,6 +102,10 @@ class Database:
                         emocao_detectada TEXT,
                         confianca_emocao REAL DEFAULT 0.5,
                         
+                        -- Transição
+                        nivel_transicao INTEGER DEFAULT 0,
+                        info_transicao_json TEXT,
+                        
                         -- Grupo
                         grupo_id TEXT DEFAULT '',
                         grupo_nome TEXT DEFAULT '',
@@ -114,7 +119,7 @@ class Database:
                         comando_executado TEXT,
                         has_media BOOLEAN DEFAULT 0,
                         media_type TEXT DEFAULT '',
-                        message_id TEXT,  -- CORREÇÃO: Removido UNIQUE
+                        message_id TEXT,
                         bot_response_time_ms INTEGER DEFAULT 0,
                         is_mention BOOLEAN DEFAULT 0,
                         timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
@@ -169,6 +174,7 @@ class Database:
                         humor_atual TEXT DEFAULT 'normal_ironico',
                         modo_resposta TEXT DEFAULT 'normal_ironico',
                         nivel_transicao INTEGER DEFAULT 0,
+                        info_transicao_json TEXT,
                         humor_alvo TEXT DEFAULT 'normal_ironico',
                         termos TEXT,
                         girias TEXT,
@@ -196,6 +202,7 @@ class Database:
                         output_text TEXT NOT NULL,
                         humor TEXT DEFAULT 'normal_ironico',
                         modo_resposta TEXT DEFAULT 'normal_ironico',
+                        nivel_transicao INTEGER DEFAULT 0,
                         emocao_contexto TEXT,
                         contexto_super_claro TEXT,
                         tipo_interacao TEXT DEFAULT 'normal',
@@ -216,9 +223,10 @@ class Database:
                         contexto_id TEXT NOT NULL,
                         humor_anterior TEXT NOT NULL,
                         humor_novo TEXT NOT NULL,
+                        nivel_transicao_anterior INTEGER DEFAULT 0,
+                        nivel_transicao_novo INTEGER DEFAULT 0,
                         emocao_trigger TEXT,
                         confianca_emocao REAL,
-                        nivel_transicao INTEGER,
                         razao TEXT,
                         intensidade REAL DEFAULT 0.5,
                         contexto_mensagem TEXT,
@@ -270,8 +278,26 @@ class Database:
                     )
                 ''')
                 
+                # Interações (para treinamento)
+                c.execute('''
+                    CREATE TABLE IF NOT EXISTS interacoes (
+                        id INTEGER PRIMARY KEY AUTOINCREMENT,
+                        numero TEXT NOT NULL,
+                        mensagem TEXT NOT NULL,
+                        resposta TEXT NOT NULL,
+                        humor TEXT DEFAULT 'normal_ironico',
+                        modo_resposta TEXT DEFAULT 'normal_ironico',
+                        nivel_transicao INTEGER DEFAULT 0,
+                        emocao_detectada TEXT,
+                        tipo_conversa TEXT DEFAULT 'pv',
+                        reply_info_json TEXT,
+                        qualidade_score REAL DEFAULT 1.0,
+                        data_criacao DATETIME DEFAULT CURRENT_TIMESTAMP
+                    )
+                ''')
+                
                 conn.commit()
-                logger.info("✅ Tabelas criadas/verificadas com message_id corrigido")
+                logger.info("✅ Tabelas criadas/verificadas com suporte a nivel_transicao")
                 
         except Exception as e:
             logger.error(f"❌ Erro ao criar tabelas: {e}")
@@ -298,8 +324,10 @@ class Database:
                     ("is_mention", "BOOLEAN DEFAULT 0"),
                     ("has_media", "BOOLEAN DEFAULT 0"),
                     ("media_type", "TEXT DEFAULT ''"),
-                    ("message_id", "TEXT"),  # CORREÇÃO: Sem UNIQUE aqui também
-                    ("bot_response_time_ms", "INTEGER DEFAULT 0")
+                    ("message_id", "TEXT"),
+                    ("bot_response_time_ms", "INTEGER DEFAULT 0"),
+                    ("nivel_transicao", "INTEGER DEFAULT 0"),
+                    ("info_transicao_json", "TEXT")
                 ]
                 
                 for col_name, col_def in novas_colunas:
@@ -308,6 +336,41 @@ class Database:
                     except sqlite3.OperationalError:
                         pass
                 
+                # Colunas para contexto
+                contexto_colunas = [
+                    ("info_transicao_json", "TEXT"),
+                    ("nivel_transicao", "INTEGER DEFAULT 0")
+                ]
+                
+                for col_name, col_def in contexto_colunas:
+                    try:
+                        c.execute(f"ALTER TABLE contexto ADD COLUMN {col_name} {col_def}")
+                    except sqlite3.OperationalError:
+                        pass
+                
+                # Colunas para transições
+                transicoes_colunas = [
+                    ("nivel_transicao_anterior", "INTEGER DEFAULT 0"),
+                    ("nivel_transicao_novo", "INTEGER DEFAULT 0")
+                ]
+                
+                for col_name, col_def in transicoes_colunas:
+                    try:
+                        c.execute(f"ALTER TABLE transicoes_humor ADD COLUMN {col_name} {col_def}")
+                    except sqlite3.OperationalError:
+                        pass
+                
+                # Colunas para training_examples
+                training_colunas = [
+                    ("nivel_transicao", "INTEGER DEFAULT 0")
+                ]
+                
+                for col_name, col_def in training_colunas:
+                    try:
+                        c.execute(f"ALTER TABLE training_examples ADD COLUMN {col_name} {col_def}")
+                    except sqlite3.OperationalError:
+                        pass
+                
                 conn.commit()
                 
         except Exception as e:
@@ -326,6 +389,8 @@ class Database:
                        modo_resposta: str = 'normal_ironico',
                        emocao_detectada: str = None,
                        confianca_emocao: float = 0.5,
+                       nivel_transicao: int = 0,
+                       info_transicao: dict = None,
                        tipo_mensagem: str = 'texto',
                        reply_info_json: str = None,
                        usuario_nome: str = '',
@@ -341,7 +406,7 @@ class Database:
                        message_id: str = None,
                        bot_response_time_ms: int = 0,
                        is_mention: bool = False) -> bool:
-        """Salva mensagem no banco - CORREÇÃO: message_id sem UNIQUE constraint"""
+        """Salva mensagem no banco - COM SUPORTE A nivel_transicao"""
         try:
             numero_final = str(numero or usuario).strip()
             contexto_id = self._gerar_contexto_id(numero_final, tipo_conversa)
@@ -350,14 +415,18 @@ class Database:
             if isinstance(reply_info_json, dict):
                 reply_info_json = json.dumps(reply_info_json, ensure_ascii=False)
             
-            # CORREÇÃO: Gera message_id único se não fornecido
+            # Converte info_transicao para JSON se for dict
+            info_transicao_json = None
+            if info_transicao:
+                info_transicao_json = json.dumps(info_transicao, ensure_ascii=False)
+            
+            # Gera message_id único se não fornecido
             if not message_id:
                 timestamp = int(time.time() * 1000)
                 random_suffix = random.randint(1000, 9999)
                 message_id = f"{numero_final}_{timestamp}_{random_suffix}"
             
-            # CORREÇÃO: Adiciona um sufixo aleatório extra para garantir unicidade
-            # mesmo que o timestamp seja o mesmo
+            # Adiciona um sufixo aleatório extra para garantir unicidade
             unique_suffix = random.randint(100, 999)
             message_id = f"{message_id}_{unique_suffix}"
             
@@ -368,10 +437,10 @@ class Database:
                     (usuario, usuario_nome, mensagem, resposta, numero, contexto_id, tipo_contexto,
                      tipo_conversa, tipo_mensagem, is_reply, mensagem_original, mensagem_citada_limpa,
                      reply_to_bot, reply_info_json, humor, modo_resposta, emocao_detectada, 
-                     confianca_emocao, grupo_id, grupo_nome, audio_transcricao, fonte_stt, 
-                     confianca_stt, comando_executado, has_media, media_type, message_id,
-                     bot_response_time_ms, is_mention)
-                    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                     confianca_emocao, nivel_transicao, info_transicao_json, grupo_id, grupo_nome, 
+                     audio_transcricao, fonte_stt, confianca_stt, comando_executado, has_media, 
+                     media_type, message_id, bot_response_time_ms, is_mention)
+                    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                     """,
                     (
                         usuario[:50], usuario_nome[:100] or usuario[:100], 
@@ -379,7 +448,8 @@ class Database:
                         contexto_id, tipo_conversa, tipo_conversa, tipo_mensagem,
                         int(is_reply), mensagem_original, mensagem_citada_limpa,
                         int(reply_to_bot), reply_info_json, humor, modo_resposta,
-                        emocao_detectada, confianca_emocao, grupo_id[:50], grupo_nome[:100],
+                        emocao_detectada, confianca_emocao, nivel_transicao,
+                        info_transicao_json, grupo_id[:50], grupo_nome[:100],
                         audio_transcricao[:2000] if audio_transcricao else None,
                         fonte_stt[:50], confianca_stt, comando_executado[:100] if comando_executado else None,
                         int(has_media), media_type[:50], message_id[:200],
@@ -389,7 +459,7 @@ class Database:
                     fetch=False
                 )
                 
-                logger.debug(f"✅ Mensagem salva: {numero_final} | Message ID: {message_id}")
+                logger.debug(f"✅ Mensagem salva: {numero_final} | Nível: {nivel_transicao} | Message ID: {message_id}")
                 return True
                 
             except sqlite3.IntegrityError as e:
@@ -405,10 +475,10 @@ class Database:
                         (usuario, usuario_nome, mensagem, resposta, numero, contexto_id, tipo_contexto,
                          tipo_conversa, tipo_mensagem, is_reply, mensagem_original, mensagem_citada_limpa,
                          reply_to_bot, reply_info_json, humor, modo_resposta, emocao_detectada, 
-                         confianca_emocao, grupo_id, grupo_nome, audio_transcricao, fonte_stt, 
-                         confianca_stt, comando_executado, has_media, media_type, message_id,
-                         bot_response_time_ms, is_mention)
-                        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                         confianca_emocao, nivel_transicao, info_transicao_json, grupo_id, grupo_nome, 
+                         audio_transcricao, fonte_stt, confianca_stt, comando_executado, has_media, 
+                         media_type, message_id, bot_response_time_ms, is_mention)
+                        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                         """,
                         (
                             usuario[:50], usuario_nome[:100] or usuario[:100], 
@@ -416,7 +486,8 @@ class Database:
                             contexto_id, tipo_conversa, tipo_conversa, tipo_mensagem,
                             int(is_reply), mensagem_original, mensagem_citada_limpa,
                             int(reply_to_bot), reply_info_json, humor, modo_resposta,
-                            emocao_detectada, confianca_emocao, grupo_id[:50], grupo_nome[:100],
+                            emocao_detectada, confianca_emocao, nivel_transicao,
+                            info_transicao_json, grupo_id[:50], grupo_nome[:100],
                             audio_transcricao[:2000] if audio_transcricao else None,
                             fonte_stt[:50], confianca_stt, comando_executado[:100] if comando_executado else None,
                             int(has_media), media_type[:50], new_message_id[:200],
@@ -437,13 +508,14 @@ class Database:
     def salvar_training_example(self, input_text: str, output_text: str,
                                humor: str = "normal_ironico",
                                modo_resposta: str = "normal_ironico",
+                               nivel_transicao: int = 0,
                                emocao_contexto: str = None,
                                qualidade_score: float = 1.0,
                                contexto_super_claro: Dict = None,
                                tipo_interacao: str = "normal",
                                score_relevancia: float = 1.0,
                                tags: List[str] = None) -> bool:
-        """Salva exemplo de treinamento"""
+        """Salva exemplo de treinamento - COM nivel_transicao"""
         try:
             contexto_json = json.dumps(contexto_super_claro, ensure_ascii=False) if contexto_super_claro else None
             tags_str = ",".join(tags) if tags else ""
@@ -451,50 +523,54 @@ class Database:
             self._execute_with_retry(
                 """
                 INSERT INTO training_examples
-                (input_text, output_text, humor, modo_resposta, emocao_contexto,
-                 qualidade_score, contexto_super_claro, tipo_interacao, score_relevancia, tags)
-                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                (input_text, output_text, humor, modo_resposta, nivel_transicao,
+                 emocao_contexto, qualidade_score, contexto_super_claro, 
+                 tipo_interacao, score_relevancia, tags)
+                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                 """,
                 (
                     input_text[:2000], output_text[:2000], humor, modo_resposta,
-                    emocao_contexto, qualidade_score, contexto_json,
+                    nivel_transicao, emocao_contexto, qualidade_score, contexto_json,
                     tipo_interacao, score_relevancia, tags_str[:200]
                 ),
                 commit=True,
                 fetch=False
             )
-            logger.debug(f"✅ Training example salvo | Score: {qualidade_score:.2f}")
+            logger.debug(f"✅ Training example salvo | Nível: {nivel_transicao} | Score: {qualidade_score:.2f}")
             return True
         except Exception as e:
             logger.error(f"❌ Erro ao salvar training: {e}")
             return False
 
     def salvar_transicao_humor(self, numero: str, humor_anterior: str,
-                              humor_novo: str, emocao_trigger: str = None,
+                              humor_novo: str, nivel_transicao_anterior: int = 0,
+                              nivel_transicao_novo: int = 0,
+                              emocao_trigger: str = None,
                               confianca_emocao: float = 0.5,
-                              nivel_transicao: int = 0,
                               razao: str = "", intensidade: float = 0.5,
                               contexto_mensagem: str = None):
-        """Salva transição de humor"""
+        """Salva transição de humor - COM nivel_transicao"""
         try:
             contexto_id = self._gerar_contexto_id(numero, 'auto')
             
             self._execute_with_retry(
                 """
                 INSERT INTO transicoes_humor
-                (numero, contexto_id, humor_anterior, humor_novo, emocao_trigger,
-                 confianca_emocao, nivel_transicao, razao, intensidade, contexto_mensagem)
-                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                (numero, contexto_id, humor_anterior, humor_novo, 
+                 nivel_transicao_anterior, nivel_transicao_novo, emocao_trigger,
+                 confianca_emocao, razao, intensidade, contexto_mensagem)
+                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                 """,
                 (
                     str(numero).strip(), contexto_id, humor_anterior, humor_novo,
-                    emocao_trigger, confianca_emocao, nivel_transicao,
-                    razao[:200], intensidade, contexto_mensagem[:500] if contexto_mensagem else None
+                    nivel_transicao_anterior, nivel_transicao_novo, emocao_trigger,
+                    confianca_emocao, razao[:200], intensidade, 
+                    contexto_mensagem[:500] if contexto_mensagem else None
                 ),
                 commit=True,
                 fetch=False
             )
-            logger.debug(f"🎭 Transição salva: {humor_anterior} → {humor_novo}")
+            logger.debug(f"🎭 Transição salva: {humor_anterior}→{humor_novo} | Nível: {nivel_transicao_anterior}→{nivel_transicao_novo}")
         except Exception as e:
             logger.error(f"❌ Erro ao salvar transição: {e}")
 
@@ -538,6 +614,136 @@ class Database:
             logger.error(f"❌ Erro ao salvar gíria: {e}")
             return False
 
+    # ========================================================================
+    # MÉTODOS DE CONTEXTO - COM nivel_transicao
+    # ========================================================================
+    
+    def atualizar_contexto(self, numero: str, humor_atual: str = None,
+                          modo_resposta: str = None, nivel_transicao: int = None,
+                          info_transicao: dict = None, tom: str = None,
+                          emocao_tendencia: str = None) -> bool:
+        """Atualiza contexto do usuário - COM nivel_transicao"""
+        try:
+            numero_final = str(numero).strip()
+            contexto_id = self._gerar_contexto_id(numero_final, 'auto')
+            
+            # Verifica se contexto existe
+            result = self._execute_with_retry(
+                "SELECT 1 FROM contexto WHERE numero = ?",
+                (numero_final,),
+                fetch=True
+            )
+            
+            info_transicao_json = None
+            if info_transicao:
+                info_transicao_json = json.dumps(info_transicao, ensure_ascii=False)
+            
+            if result:
+                # Atualiza existente
+                updates = []
+                params = []
+                
+                if humor_atual:
+                    updates.append("humor_atual = ?")
+                    params.append(humor_atual)
+                if modo_resposta:
+                    updates.append("modo_resposta = ?")
+                    params.append(modo_resposta)
+                if nivel_transicao is not None:
+                    updates.append("nivel_transicao = ?")
+                    params.append(nivel_transicao)
+                if info_transicao_json:
+                    updates.append("info_transicao_json = ?")
+                    params.append(info_transicao_json)
+                if tom:
+                    updates.append("tom = ?")
+                    params.append(tom)
+                if emocao_tendencia:
+                    updates.append("emocao_tendencia = ?")
+                    params.append(emocao_tendencia)
+                
+                updates.append("ultimo_contato = CURRENT_TIMESTAMP")
+                updates.append("data_atualizacao = CURRENT_TIMESTAMP")
+                
+                if updates:
+                    query = f"UPDATE contexto SET {', '.join(updates)} WHERE numero = ?"
+                    params.append(numero_final)
+                    self._execute_with_retry(query, tuple(params), commit=True, fetch=False)
+            else:
+                # Cria novo contexto
+                self._execute_with_retry(
+                    """
+                    INSERT INTO contexto
+                    (numero, contexto_id, humor_atual, modo_resposta, nivel_transicao,
+                     info_transicao_json, tom, emocao_tendencia, ultimo_contato)
+                    VALUES (?, ?, ?, ?, ?, ?, ?, ?, CURRENT_TIMESTAMP)
+                    """,
+                    (
+                        numero_final, contexto_id,
+                        humor_atual or 'normal_ironico',
+                        modo_resposta or 'normal_ironico',
+                        nivel_transicao or 0,
+                        info_transicao_json,
+                        tom or 'normal',
+                        emocao_tendencia or 'neutral'
+                    ),
+                    commit=True,
+                    fetch=False
+                )
+            
+            logger.debug(f"✅ Contexto atualizado: {numero_final} | Nível: {nivel_transicao}")
+            return True
+            
+        except Exception as e:
+            logger.error(f"❌ Erro ao atualizar contexto: {e}")
+            return False
+
+    def recuperar_contexto(self, numero: str) -> Dict[str, Any]:
+        """Recupera contexto completo do usuário"""
+        try:
+            result = self._execute_with_retry(
+                """
+                SELECT humor_atual, modo_resposta, nivel_transicao, info_transicao_json,
+                       tom, emocao_tendencia, ultimo_contato
+                FROM contexto WHERE numero = ?
+                """,
+                (str(numero).strip(),),
+                fetch=True
+            )
+            
+            if result:
+                row = result[0]
+                info_transicao = {}
+                if row[3]:
+                    try:
+                        info_transicao = json.loads(row[3])
+                    except:
+                        pass
+                
+                return {
+                    "humor_atual": row[0] or "normal_ironico",
+                    "modo_resposta": row[1] or "normal_ironico",
+                    "nivel_transicao": row[2] or 0,
+                    "info_transicao": info_transicao,
+                    "tom": row[4] or "normal",
+                    "emocao_tendencia": row[5] or "neutral",
+                    "ultimo_contato": row[6]
+                }
+            
+            return {
+                "humor_atual": "normal_ironico",
+                "modo_resposta": "normal_ironico",
+                "nivel_transicao": 0,
+                "info_transicao": {},
+                "tom": "normal",
+                "emocao_tendencia": "neutral",
+                "ultimo_contato": None
+            }
+            
+        except Exception as e:
+            logger.error(f"❌ Erro ao recuperar contexto: {e}")
+            return {}
+
     # ========================================================================
     # MÉTODOS DE RECUPERAÇÃO
     # ========================================================================
@@ -548,7 +754,7 @@ class Database:
             results = self._execute_with_retry(
                 """
                 SELECT mensagem, resposta, is_reply, mensagem_original, 
-                       reply_to_bot, humor, modo_resposta, timestamp
+                       reply_to_bot, humor, modo_resposta, nivel_transicao, timestamp
                 FROM mensagens
                 WHERE numero = ? AND deletado = 0
                 ORDER BY timestamp DESC
@@ -589,13 +795,25 @@ class Database:
         except Exception:
             return "normal_ironico"
 
+    def recuperar_nivel_transicao(self, numero: str) -> int:
+        """Recupera nível de transição"""
+        try:
+            result = self._execute_with_retry(
+                "SELECT nivel_transicao FROM contexto WHERE numero = ?",
+                (str(numero).strip(),),
+                fetch=True
+            )
+            return result[0][0] if result else 0
+        except Exception:
+            return 0
+
     def recuperar_training_examples(self, limite: int = 100, usado: bool = False) -> List[Dict]:
         """Recupera exemplos de treinamento"""
         try:
             where_clause = "WHERE usado = 0" if not usado else ""
             results = self._execute_with_retry(
                 f"""
-                SELECT input_text, output_text, humor, modo_resposta,
+                SELECT input_text, output_text, humor, modo_resposta, nivel_transicao,
                        qualidade_score, tipo_interacao
                 FROM training_examples
                 {where_clause}
@@ -612,8 +830,9 @@ class Database:
                     "output": r[1],
                     "humor": r[2],
                     "modo": r[3],
-                    "score": r[4],
-                    "tipo": r[5]
+                    "nivel_transicao": r[4],
+                    "score": r[5],
+                    "tipo": r[6]
                 }
                 for r in results
             ]
@@ -637,6 +856,44 @@ class Database:
         except Exception as e:
             logger.error(f"❌ Erro ao marcar exemplos: {e}")
 
+    # ========================================================================
+    # MÉTODO PARA REGISTRAR INTERAÇÃO (PARA TREINAMENTO)
+    # ========================================================================
+    
+    def registrar_interacao(self, numero: str, mensagem: str, resposta: str,
+                           humor: str = 'normal_ironico',
+                           modo_resposta: str = 'normal_ironico',
+                           nivel_transicao: int = 0,
+                           emocao_detectada: str = None,
+                           tipo_conversa: str = 'pv',
+                           reply_info_json: str = None,
+                           qualidade_score: float = 1.0) -> bool:
+        """Registra interação para treinamento - COM nivel_transicao"""
+        try:
+            if isinstance(reply_info_json, dict):
+                reply_info_json = json.dumps(reply_info_json, ensure_ascii=False)
+            
+            self._execute_with_retry(
+                """
+                INSERT INTO interacoes
+                (numero, mensagem, resposta, humor, modo_resposta, nivel_transicao,
+                 emocao_detectada, tipo_conversa, reply_info_json, qualidade_score)
+                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                """,
+                (
+                    str(numero).strip(), mensagem[:2000], resposta[:2000], humor, modo_resposta,
+                    nivel_transicao, emocao_detectada, tipo_conversa, reply_info_json,
+                    qualidade_score
+                ),
+                commit=True,
+                fetch=False
+            )
+            logger.debug(f"✅ Interação registrada: {numero} | Nível: {nivel_transicao}")
+            return True
+        except Exception as e:
+            logger.error(f"❌ Erro ao registrar interação: {e}")
+            return False
+
     # ========================================================================
     # PRIVILÉGIOS
     # ========================================================================
@@ -757,10 +1014,14 @@ class Database:
                 'timestamp': time.time()
             }
             
+            nivel_transicao = contexto.get('nivel_transicao', 0)
+            
             return self.salvar_training_example(
                 input_text=input_text,
                 output_text=output_text,
                 humor=contexto.get("humor_atualizado", "normal_ironico"),
+                modo_resposta=contexto.get("modo_resposta", "normal_ironico"),
+                nivel_transicao=nivel_transicao,
                 qualidade_score=qualidade_score,
                 contexto_super_claro=contexto_super_claro,
                 tipo_interacao=tipo_aprendizado