Update modules/api.py

modules/api.py

@@ -1,11 +1,12 @@
-# modules/api.py — AKIRA V21 FINAL INTEGRADO (Dezembro 2025)
+# modules/api.py — AKIRA V21 FINAL INTEGRADO (Dezembro 2025) - CORRIGIDO
 """
-✅ TOTALMENTE INTEGRADO com config.py, database.py, contexto.py e treinamento.py atualizados
-✅ CORREÇÃO: Usa métodos corretos do Database
+✅ TOTALMENTE INTEGRADO com database.py corrigido (message_id sem UNIQUE)
+✅ CORREÇÃO: Usa métodos corretos do Database atualizado
 ✅ COMPATÍVEL com index.js e reply_metadata
 ✅ Sistema multi-API com fallback
 ✅ Cache de contexto otimizado
 ✅ Treinamento automático integrado
+✅ Resolve problema de UNIQUE constraint failed
 """
 
 import time
@@ -93,6 +94,7 @@ class MultiAPIManager:
         """
         try:
             logger.debug(f"📝 Construindo prompt para: {usuario}")
+            logger.debug(f"📝 Mensagem citada: {mensagem_citada[:80] if mensagem_citada else '[Nenhuma]'}...")
             
             # Usa a função do config.py
             prompt = config.construir_prompt_api(
@@ -139,6 +141,7 @@ RESPONDA COMO AKIRA (IRÔNICA, SEM FILTRO, 1-2 LINHAS):"""
     ) -> str:
         """Gera resposta usando multi-API fallback"""
         logger.info(f"🎯 Gerando resposta para {usuario}")
+        logger.info(f"📤 Mensagem: {mensagem[:80]}...")
         
         # Construir prompt
         prompt = self._construir_prompt(
@@ -154,23 +157,28 @@ RESPONDA COMO AKIRA (IRÔNICA, SEM FILTRO, 1-2 LINHAS):"""
         # Obter parâmetros
         parametros = self._obter_parametros_api(analise)
         
+        logger.debug(f"🔧 Parâmetros: {parametros}")
+        
         # Tentar APIs em ordem
         for api_name in config.API_FALLBACK_ORDER:
             if api_name not in self.apis_disponiveis:
                 continue
                 
             try:
+                logger.debug(f"🔄 Tentando API: {api_name}")
                 resposta = self._chamar_api(api_name, prompt, parametros)
                 if resposta:
                     resposta_limpa = self._limpar_resposta(resposta)
                     logger.info(f"✅ API {api_name} respondeu: {resposta_limpa[:80]}...")
                     return resposta_limpa
             except Exception as e:
-                logger.warning(f"❌ API {api_name} falhou: {e}")
+                logger.warning(f"❌ API {api_name} falhou: {str(e)[:100]}")
                 continue
         
         # Fallback contextual
-        return self._gerar_fallback_contextual(mensagem, mensagem_citada, reply_info)
+        fallback = self._gerar_fallback_contextual(mensagem, mensagem_citada, reply_info)
+        logger.info(f"🔄 Usando fallback: {fallback}")
+        return fallback
 
     def _chamar_api(self, api_name: str, prompt: str, params: Dict[str, Any]) -> str:
         """Chama API específica"""
@@ -340,7 +348,7 @@ class AkiraAPI:
         self.api = Blueprint("akira_api", __name__)
         self.contexto_cache = SimpleTTLCache(ttl_seconds=300)
         
-        # Inicializa Database com caminho CORRETO
+        # Inicializa Database CORRETAMENTE
         self.db = Database(config.DB_PATH)
         self.llm_manager = MultiAPIManager()
         
@@ -358,7 +366,7 @@ class AkiraAPI:
                 self.treinador = None
         
         self._setup_routes()
-        logger.success("🚀 AKIRA V21 FINAL inicializada")
+        logger.success("🚀 AKIRA V21 FINAL inicializada (Database corrigido)")
 
     def _get_user_context(self, numero: str, tipo_conversa: str, grupo_id: str = "") -> Contexto:
         """Obtém contexto isolado"""
@@ -401,7 +409,7 @@ class AkiraAPI:
         }
 
     def _extrair_payload_indexjs(self, data: Dict[str, Any]) -> Dict[str, Any]:
-        """Extrai dados do payload do index.js"""
+        """Extrai dados do payload do index.js - ATUALIZADO"""
         payload = {
             'numero': str(data.get('numero', '')).strip(),
             'usuario': data.get('usuario', 'Anônimo').strip(),
@@ -445,7 +453,7 @@ class AkiraAPI:
 
         @self.api.route('/akira', methods=['POST'])
         def akira_endpoint():
-            """Endpoint principal"""
+            """Endpoint principal - CORRIGIDO para usar database atualizado"""
             try:
                 # Recebe payload
                 data = request.get_json() or {}
@@ -532,7 +540,15 @@ class AkiraAPI:
                 if payload['reply_metadata']:
                     reply_to_bot = payload['reply_metadata'].get('reply_to_bot', False)
                 
-                # Atualiza contexto
+                # Mede tempo de resposta
+                tempo_resposta_ms = int((time.time() - request.start_time) * 1000) if hasattr(request, 'start_time') else 0
+                
+                # CORREÇÃO: Prepara reply_info_json para o Database
+                reply_info_json = None
+                if payload['reply_metadata']:
+                    reply_info_json = json.dumps(payload['reply_metadata'], ensure_ascii=False)
+                
+                # Atualiza contexto usando o Database CORRIGIDO
                 contexto.atualizar_contexto(
                     mensagem=payload['mensagem'],
                     resposta=resposta,
@@ -542,6 +558,38 @@ class AkiraAPI:
                     reply_to_bot=reply_to_bot
                 )
                 
+                # Salva mensagem diretamente no banco (backup)
+                try:
+                    # Gera message_id único com timestamp e random
+                    timestamp = int(time.time() * 1000)
+                    random_suffix = random.randint(1000, 9999)
+                    message_id = f"{payload['numero']}_{timestamp}_{random_suffix}"
+                    
+                    self.db.salvar_mensagem(
+                        usuario=payload['usuario'],
+                        mensagem=payload['mensagem'],
+                        resposta=resposta,
+                        numero=payload['numero'],
+                        is_reply=is_reply,
+                        mensagem_original=payload['mensagem_citada'],
+                        reply_to_bot=reply_to_bot,
+                        humor=analise.get('humor_atualizado', 'normal_ironico'),
+                        modo_resposta=analise.get('modo_resposta', 'normal_ironico'),
+                        emocao_detectada=analise.get('emocao_primaria', 'neutral'),
+                        confianca_emocao=analise.get('confianca_emocao', 0.5),
+                        tipo_mensagem=payload['tipo_mensagem'],
+                        reply_info_json=reply_info_json,
+                        usuario_nome=payload['usuario'],
+                        grupo_id=payload['grupo_id'],
+                        grupo_nome=payload['grupo_nome'],
+                        tipo_conversa=payload['tipo_conversa'],
+                        message_id=message_id,
+                        bot_response_time_ms=tempo_resposta_ms
+                    )
+                    logger.debug(f"✅ Mensagem salva no banco com message_id: {message_id}")
+                except Exception as db_error:
+                    logger.warning(f"⚠️ Erro ao salvar mensagem no banco: {db_error}")
+                
                 # Registra interação para treinamento
                 if hasattr(self, 'treinador') and self.treinador:
                     try:
@@ -558,6 +606,7 @@ class AkiraAPI:
                             reply_to_bot=reply_to_bot,
                             reply_metadata=payload['reply_metadata']
                         )
+                        logger.debug("✅ Interação registrada para treinamento")
                     except Exception as e:
                         logger.warning(f"⚠️ Erro ao registrar interação: {e}")
                 
@@ -582,8 +631,10 @@ class AkiraAPI:
                 "status": "✅ AKIRA V21 ONLINE",
                 "hora_luanda": agora.strftime("%H:%M"),
                 "versao": config.VERSAO,
+                "database": "Corrigido (message_id sem UNIQUE)",
                 "apis_disponiveis": self.llm_manager.apis_disponiveis,
-                "cache_size": len(self.contexto_cache._store)
+                "cache_size": len(self.contexto_cache._store),
+                "treinamento_ativo": hasattr(self, 'treinador') and self.treinador is not None
             })
 
         @self.api.route('/reset', methods=['POST'])
@@ -616,11 +667,13 @@ class AkiraAPI:
                 "desenvolvedor": "Isaac Quarenta",
                 "empresa": "Softedge",
                 "versao": config.VERSAO,
+                "database_status": "Corrigido - message_id sem UNIQUE constraint",
                 "endpoints": ["/akira", "/health", "/reset", "/info"],
                 "configuracoes": {
                     "temperatura_padrao": config.TEMPERATURE,
                     "max_tokens": config.MAX_TOKENS,
-                    "timezone_offset": config.TIMEZONE_OFFSET_HOURS
+                    "timezone_offset": config.TIMEZONE_OFFSET_HOURS,
+                    "treinamento_auto": config.START_PERIODIC_TRAINER
                 }
             })
 
@@ -692,7 +745,7 @@ def gerar_resposta_direta(
 # ============================================================================
 if __name__ == "__main__":
     print("=" * 80)
-    print("TESTANDO API.PY - FINAL INTEGRADO")
+    print("TESTANDO API.PY - FINAL INTEGRADO E CORRIGIDO")
     print("=" * 80)
     
     # Teste simples
@@ -723,11 +776,40 @@ if __name__ == "__main__":
         print(f"📝 Mensagem: {test_payload['mensagem']}")
         print(f"💬 Resposta: {resposta}")
         
+        # Teste com mensagem citada
+        test_payload2 = {
+            "numero": "244978787009",
+            "usuario": "Isaac Quarenta",
+            "mensagem": "Você disse que gosta de futebol?",
+            "tipo_conversa": "pv",
+            "tipo_mensagem": "texto",
+            "mensagem_citada": "Sim, gosto do Petro de Luanda.",
+            "reply_metadata": {
+                "is_reply": True,
+                "reply_to_bot": True,
+                "quoted_author_name": "Akira"
+            }
+        }
+        
+        resposta2 = gerar_resposta_direta(
+            mensagem=test_payload2["mensagem"],
+            usuario=test_payload2["usuario"],
+            numero=test_payload2["numero"],
+            tipo_conversa=test_payload2["tipo_conversa"],
+            mensagem_citada=test_payload2["mensagem_citada"],
+            reply_metadata=test_payload2["reply_metadata"]
+        )
+        
+        print(f"\n✅ Teste com reply OK")
+        print(f"📝 Mensagem: {test_payload2['mensagem']}")
+        print(f"📝 Mensagem citada: {test_payload2['mensagem_citada']}")
+        print(f"💬 Resposta: {resposta2}")
+        
     except Exception as e:
         print(f"❌ Erro: {e}")
         import traceback
         traceback.print_exc()
     
     print("\n" + "=" * 80)
-    print("API.PY - INTEGRADO E PRONTO PARA USO")
+    print("API.PY - INTEGRADO, CORRIGIDO E PRONTO PARA USO")
     print("=" * 80)