Update modules/database.py

modules/database.py

@@ -291,19 +291,30 @@ class Database:
     # MÉTODOS OBRIGATÓRIOS (SEM fetch!)
     # ================================================================
     def carregar_contexto(self, user_key: str) -> Dict[str, Any]:
-        """Carrega contexto do usuário"""
+        """Carrega contexto do usuário com correção JSON segura"""
         try:
             result = self._execute_with_retry(
                 "SELECT historico, emocao_atual, termos, girias, tom FROM contexto WHERE user_key = ?",
                 (user_key,)
             )
+
+            def safe_json_load(v, default):
+                if not v:
+                    return default
+                if isinstance(v, (list, dict)):
+                    return v
+                try:
+                    return json.loads(v)
+                except Exception:
+                    return default
+
             if result:
                 row = result[0]
                 return {
-                    "historico": row[0] or "[]",
+                    "historico": safe_json_load(row[0], []),
                     "emocao_atual": row[1] or "neutra",
-                    "termos": row[2] or "[]",
-                    "girias": row[3] or "[]",
+                    "termos": safe_json_load(row[2], []),
+                    "girias": safe_json_load(row[3], []),
                     "tom": row[4] or "neutro"
                 }
             return {}
@@ -341,31 +352,55 @@ class Database:
         try:
             cols = ['usuario', 'mensagem', 'resposta']
             vals = [usuario, mensagem, resposta]
-            if numero: cols.append('numero'); vals.append(numero)
-            if is_reply is not None: cols.append('is_reply'); vals.append(int(is_reply))
-            if mensagem_original: cols.append('mensagem_original'); vals.append(mensagem_original)
+            if numero:
+                cols.append('numero')
+                vals.append(numero)
+            if is_reply is not None:
+                cols.append('is_reply')
+                vals.append(int(is_reply))
+            if mensagem_original:
+                cols.append('mensagem_original')
+                vals.append(mensagem_original)
             placeholders = ', '.join(['?' for _ in cols])
             query = f"INSERT INTO mensagens ({', '.join(cols)}) VALUES ({placeholders})"
             self._execute_with_retry(query, tuple(vals), commit=True)
         except Exception as e:
             logger.warning(f"Fallback salvar_mensagem: {e}")
-            self._execute_with_retry("INSERT INTO mensagens (usuario, mensagem, resposta) VALUES (?, ?, ?)", (usuario, mensagem, resposta), commit=True)
+            self._execute_with_retry(
+                "INSERT INTO mensagens (usuario, mensagem, resposta) VALUES (?, ?, ?)",
+                (usuario, mensagem, resposta),
+                commit=True
+            )
 
-    def recuperar_mensagens(self, usuario: str, limite: int = 5) -> List[Tuple]:
+    def recuperar_mensagens(self, usuario: str, limite: int = 5) -> List[Tuple]]:
         return self._execute_with_retry(
             "SELECT mensagem, resposta FROM mensagens WHERE usuario=? OR numero=? ORDER BY id DESC LIMIT ?",
             (usuario, usuario, limite)
         ) or []
 
     def salvar_giria_aprendida(self, numero_usuario: str, giria: str, significado: str, contexto: Optional[str] = None):
-        existing = self._execute_with_retry("SELECT id, frequencia FROM girias_aprendidas WHERE numero_usuario=? AND giria=?", (numero_usuario, giria))
+        existing = self._execute_with_retry(
+            "SELECT id, frequencia FROM girias_aprendidas WHERE numero_usuario=? AND giria=?",
+            (numero_usuario, giria)
+        )
         if existing:
-            self._execute_with_retry("UPDATE girias_aprendidas SET frequencia=frequencia+1, updated_at=CURRENT_TIMESTAMP WHERE id=?", (existing[0][0],), commit=True)
+            self._execute_with_retry(
+                "UPDATE girias_aprendidas SET frequencia=frequencia+1, updated_at=CURRENT_TIMESTAMP WHERE id=?",
+                (existing[0][0],),
+                commit=True
+            )
         else:
-            self._execute_with_retry("INSERT INTO girias_aprendidas (numero_usuario, giria, significado, contexto) VALUES (?, ?, ?, ?)", (numero_usuario, giria, significado, contexto), commit=True)
+            self._execute_with_retry(
+                "INSERT INTO girias_aprendidas (numero_usuario, giria, significado, contexto) VALUES (?, ?, ?, ?)",
+                (numero_usuario, giria, significado, contexto),
+                commit=True
+            )
 
     def recuperar_girias_usuario(self, numero_usuario: str) -> List[Dict[str, Any]]:
-        result = self._execute_with_retry("SELECT giria, significado, contexto, frequencia FROM girias_aprendidas WHERE numero_usuario=? ORDER BY frequencia DESC, updated_at DESC", (numero_usuario,))
+        result = self._execute_with_retry(
+            "SELECT giria, significado, contexto, frequencia FROM girias_aprendidas WHERE numero_usuario=? ORDER BY frequencia DESC, updated_at DESC",
+            (numero_usuario,)
+        )
         return [{"giria": r[0], "significado": r[1], "contexto": r[2], "frequencia": r[3]} for r in result] if result else []
 
     def obter_tom_predominante(self, numero_usuario: str, limite: int = 10) -> Optional[str]:
@@ -377,7 +412,11 @@ class Database:
         return result[0][0] if result else None
 
     def registrar_tom_usuario(self, numero_usuario: str, tom_detectado: str, intensidade: float = 0.5, contexto: Optional[str] = None):
-        self._execute_with_retry("INSERT INTO tom_usuario (numero_usuario, tom_detectado, intensidade, contexto) VALUES (?, ?, ?, ?)", (numero_usuario, tom_detectado, intensidade, contexto), commit=True)
+        self._execute_with_retry(
+            "INSERT INTO tom_usuario (numero_usuario, tom_detectado, intensidade, contexto) VALUES (?, ?, ?, ?)",
+            (numero_usuario, tom_detectado, intensidade, contexto),
+            commit=True
+        )
 
     def obter_pronomes_por_tom(self, tom: str) -> str:
         result = self._execute_with_retry("SELECT pronomes FROM pronomes_por_tom WHERE tom=?", (tom.lower(),))
@@ -426,4 +465,4 @@ class Database:
             emocoes["tristeza"] = 0.8; emocoes["neutro"] = 0.2
         elif any(p in msg for p in ["medo", "assustado", "tenso", "nervoso"]):
             emocoes["medo"] = 0.8; emocoes["neutro"] = 0.2
-        return emocoes
+        return emocoes