Update modules/contexto.py

modules/contexto.py

@@ -9,7 +9,7 @@ Contexto de conversa por usuário
 - SÓ STRING NO BANCO → json.loads seguro
 """
 import json
-from typing import List, Tuple, Dict
+from typing import List, Tuple, Dict, Any
 from loguru import logger
 import modules.config as config
 from .database import Database
@@ -19,6 +19,7 @@ class Contexto:
     def __init__(self, db: Database, user_key: str):
         self.db = db
         self.user_key = user_key
+        # Inicializa com tipos garantidos
         self.historico: List[Tuple[str, str]] = []
         self.emocao_atual: str = config.HUMOR_INICIAL
         self.termo_contexto: List[str] = []
@@ -27,8 +28,15 @@ class Contexto:
         self._carregar_do_banco()
 
     def _carregar_do_banco(self):
+        # Variáveis temporárias para carregar com segurança
+        temp_historico: List[Tuple[str, str]] = []
+        temp_emocao: str = config.HUMOR_INICIAL
+        temp_termos: List[str] = []
+        temp_girias: List[Dict[str, str]] = []
+        temp_tom: str = "neutro"
+        
         try:
-            dados = self.db.carregar_contexto(self.user_key)
+            dados: Dict[str, Any] = self.db.carregar_contexto(self.user_key)
             if not dados:
                 return
 
@@ -36,55 +44,65 @@ class Contexto:
             hist_raw = dados.get("historico", "[]")
             if isinstance(hist_raw, str):
                 try:
-                    self.historico = json.loads(hist_raw)
+                    loaded_hist = json.loads(hist_raw)
+                    # Melhoria: Garante que é uma lista (e assume o formato correto)
+                    if isinstance(loaded_hist, list):
+                        temp_historico = loaded_hist
                 except:
-                    self.historico = []
-            else:
-                self.historico = []
+                    pass
 
             # --- EMOÇÃO: sempre str ---
-            self.emocao_atual = str(dados.get("emocao_atual", config.HUMOR_INICIAL))
+            temp_emocao = str(dados.get("emocao_atual", config.HUMOR_INICIAL))
 
             # --- TERMOS: sempre str → json.loads ---
             termos_raw = dados.get("termos", "[]")
             if isinstance(termos_raw, str):
                 try:
-                    self.termo_contexto = json.loads(termos_raw)
+                    loaded_termos = json.loads(termos_raw)
+                    # Melhoria: Garante que é uma lista de strings
+                    if isinstance(loaded_termos, list) and all(isinstance(t, str) for t in loaded_termos):
+                        temp_termos = loaded_termos
                 except:
-                    self.termo_contexto = []
-            else:
-                self.termo_contexto = []
+                    pass
 
             # --- GÍRIAS: sempre str → json.loads + filtro ---
             girias_raw = dados.get("girias", "[]")
             if isinstance(girias_raw, str):
                 try:
                     girias = json.loads(girias_raw)
-                    self.girias_aprendidas = [
-                        {"giria": g.get("giria", ""), "significado": g.get("significado", "")}
-                        for g in girias
-                        if isinstance(g, dict) and "giria" in g
-                    ]
+                    # Melhoria: Garante que é uma lista antes de iterar
+                    if isinstance(girias, list):
+                        temp_girias = [
+                            {"giria": str(g.get("giria", "")), "significado": str(g.get("significado", ""))}
+                            for g in girias
+                            if isinstance(g, dict) and "giria" in g
+                        ]
                 except Exception as e:
-                    logger.warning(f"Erro ao parsear gírias: {e}")
-                    self.girias_aprendidas = []
-            else:
-                self.girias_aprendidas = []
+                    logger.warning(f"Erro ao parsear gírias (JSON): {e}")
+                    # temp_girias permanece []
 
             # --- TOM: sempre str ---
-            self.ton_predominante = str(dados.get("tom", "neutro"))
+            temp_tom = str(dados.get("tom", "neutro"))
 
         except Exception as e:
-            logger.warning(f"Erro ao carregar contexto: {e}")
-            self.historico = []
-            self.girias_aprendidas = []
-            self.termo_contexto = []
+            # Em caso de erro total (ex: problema na db.carregar_contexto)
+            logger.warning(f"Erro ao carregar contexto principal: {e}")
+            
+        finally:
+            # Atribui apenas os valores seguros ou padrões
+            self.historico = temp_historico
+            self.emocao_atual = temp_emocao
+            self.termo_contexto = temp_termos
+            self.girias_aprendidas = temp_girias
+            self.ton_predominante = temp_tom
+
 
     def salvar_no_banco(self):
         try:
             self.db.salvar_contexto(
                 self.user_key,
-                json.dumps(self.historico[-config.MEMORIA_MAX:]),
+                # Garante que 'historico' é uma lista e limita o tamanho
+                json.dumps([tuple(item) for item in self.historico[-config.MEMORIA_MAX:]]), 
                 self.emocao_atual,
                 json.dumps(self.termo_contexto),
                 json.dumps(self.girias_aprendidas),
@@ -96,8 +114,10 @@ class Contexto:
     def atualizar_aprendizados_do_banco(self):
         try:
             girias = self.db.listar_girias_aprendidas(self.user_key)
+            # Garante que o resultado é uma lista de dicts com chaves str
             self.girias_aprendidas = [
-                {"giria": g[0], "significado": g[1]} for g in girias
+                {"giria": str(g[0]), "significado": str(g[1])} 
+                for g in girias
             ]
         except Exception as e:
             logger.warning(f"Erro ao atualizar gírias: {e}")
@@ -105,6 +125,7 @@ class Contexto:
 
     def analisar_emocoes_mensagem(self, mensagem: str) -> str:
         msg = mensagem.lower()
+        # Lembre-se que este método retorna uma STRING
         if any(p in msg for p in ["feliz", "fixe", "bué bom", "adoro", "rsrs", "kkk"]):
             return "feliz"
         if any(p in msg for p in ["triste", "chateado", "merda", "puto", "caralho"]):
@@ -116,6 +137,7 @@ class Contexto:
         return "neutra"
 
     def obter_historico(self) -> List[Tuple[str, str]]:
+        # Garante que sempre retorna uma lista de tuples
         return self.historico[-3:]
 
     def atualizar_contexto(self, mensagem: str, resposta: str):