Update modules/api.py

modules/api.py

@@ -1,6 +1,5 @@
 # modules/api.py — AKIRA V19 ULTIMATE (Dezembro 2025)
-"""
-API Flask com 6 provedores de IA em fallback cascata
+"""API Flask com 6 provedores de IA em fallback cascata
 - Mistral → Gemini → Groq → Cohere → Together → HuggingFace
 - Respostas em <5s (média)
 - Contexto de reply estruturado
@@ -52,37 +51,33 @@ class SimpleTTLCache:
 
 class MultiAPIManager:
     """Gerencia chamadas para 6 APIs com fallback automático"""
-    
     def __init__(self):
         self.timeout = config.API_TIMEOUT
         self.apis_disponiveis = self._verificar_apis()
         logger.info(f"APIs disponíveis: {', '.join(self.apis_disponiveis)}")
-    
+
     def _verificar_apis(self):
-        """Verifica quais APIs estão configuradas"""
+        """Verifica quais APIs estão configuradas (ignora vazias)"""
         apis = []
-        if config.MISTRAL_API_KEY:
+        if config.MISTRAL_API_KEY and config.MISTRAL_API_KEY.strip():
             apis.append("mistral")
-        if config.GEMINI_API_KEY:
+        if config.GEMINI_API_KEY and config.GEMINI_API_KEY.strip():
             apis.append("gemini")
-        if config.GROQ_API_KEY:
+        if config.GROQ_API_KEY and config.GROQ_API_KEY.strip():
             apis.append("groq")
-        if config.COHERE_API_KEY:
+        if config.COHERE_API_KEY and config.COHERE_API_KEY.strip():
             apis.append("cohere")
-        if config.TOGETHER_API_KEY:
+        if config.TOGETHER_API_KEY and config.TOGETHER_API_KEY.strip():
             apis.append("together")
-        if config.HF_API_KEY:
+        if config.HF_API_KEY and config.HF_API_KEY.strip():
             apis.append("huggingface")
         return apis
-    
-    def _construir_prompt(self, mensagem: str, historico: list, mensagem_citada: str, 
-                         humor: str, tom_usuario: str) -> str:
+
+    def _construir_prompt(self, mensagem: str, historico: list, mensagem_citada: str, humor: str, tom_usuario: str) -> str:
         """Constrói prompt otimizado com contexto"""
-        
         # === INFORMAÇÕES DA EMPRESA (SOFTEDGE) ===
         empresa_info = EmpresaInfo()
         info_context = ""
-        
         # Detecta se usuário pergunta sobre criador/empresa
         msg_lower = mensagem.lower()
         if any(palavra in msg_lower for palavra in ["criou", "criador", "quem fez", "desenvolveu", "softedge", "isaac"]):
@@ -91,26 +86,22 @@ class MultiAPIManager:
                 info_context = f"\n[INFO IMPORTANTE]: {empresa_info.get_resposta_sobre_empresa(mensagem, tom_formal)}\n"
             elif "softedge" in msg_lower or "empresa" in msg_lower:
                 info_context = f"\n[INFO IMPORTANTE]: Softedge é empresa angolana de IA fundada por Isaac Quarenta em 2024. WhatsApp: {empresa_info.get_canal_whatsapp()}\n"
-        
         # === DATA E HORA ATUAL ===
         from datetime import datetime
         agora = datetime.now()
         data_hora_atual = agora.strftime("%d de %B de %Y, %H:%M")
         # Traduz mês para português
         meses = {
-            "January": "janeiro", "February": "fevereiro", "March": "março",
-            "April": "abril", "May": "maio", "June": "junho",
-            "July": "julho", "August": "agosto", "September": "setembro",
-            "October": "outubro", "November": "novembro", "December": "dezembro"
+            "January": "janeiro", "February": "fevereiro", "March": "março", "April": "abril",
+            "May": "maio", "June": "junho", "July": "julho", "August": "agosto",
+            "September": "setembro", "October": "outubro", "November": "novembro", "December": "dezembro"
         }
         for en, pt in meses.items():
             data_hora_atual = data_hora_atual.replace(en, pt)
-        
         # Contexto de reply (se existir)
         reply_context = ""
         if mensagem_citada:
             reply_context = f"\n[USUÁRIO ESTÁ RESPONDENDO A]: \"{mensagem_citada[:100]}...\"\n"
-        
         # Histórico formatado
         historico_texto = ""
         if historico:
@@ -119,25 +110,10 @@ class MultiAPIManager:
                 role = msg.get("role", "user")
                 content = msg.get("content", "")
                 historico_texto += f"{role.upper()}: {content}\n"
-        
         # Prompt final
-        prompt = f"""{config.PERSONA.format(humor=humor, tom_usuario=tom_usuario)}
-
-{config.SYSTEM_PROMPT.format(mensagem_citada=mensagem_citada or "nenhuma", humor=humor)}
-
-DATA E HORA ATUAL: {data_hora_atual}
-{info_context}
-
-CONTEXTO DA CONVERSA:
-{historico_texto}
-{reply_context}
-
-USUÁRIO: {mensagem}
-
-AKIRA (responda EM 1-2 LINHAS curtas, sem markdown):"""
-        
+        prompt = f"""{config.PERSONA.format(humor=humor, tom_usuario=tom_usuario)}{config.SYSTEM_PROMPT.format(mensagem_citada=mensagem_citada or "nenhuma", humor=humor)}DATA E HORA ATUAL: {data_hora_atual}{info_context}CONTEXTO DA CONVERSA:{historico_texto}{reply_context}USUÁRIO: {mensagem}AKIRA (responda EM 1-2 LINHAS curtas, sem markdown):"""
         return prompt
-    
+
     # === API 1: MISTRAL ===
     def _chamar_mistral(self, prompt: str) -> str:
         """Chama Mistral AI"""
@@ -155,12 +131,16 @@ AKIRA (responda EM 1-2 LINHAS curtas, sem markdown):"""
                 headers=headers,
                 timeout=self.timeout
             )
+            logger.debug(f"Mistral response: {resp.status_code} - {resp.text[:100]}...")
             if resp.status_code == 200:
                 return resp.json()["choices"][0]["message"]["content"].strip()
+            else:
+                logger.warning(f"Mistral erro: {resp.status_code} - {resp.text}")
+                return None
         except Exception as e:
             logger.warning(f"Mistral falhou: {e}")
-        return None
-    
+            return None
+
     # === API 2: GEMINI ===
     def _chamar_gemini(self, prompt: str) -> str:
         """Chama Google Gemini"""
@@ -174,12 +154,16 @@ AKIRA (responda EM 1-2 LINHAS curtas, sem markdown):"""
                 }
             }
             resp = requests.post(url, json=payload, timeout=self.timeout)
+            logger.debug(f"Gemini response: {resp.status_code} - {resp.text[:100]}...")
             if resp.status_code == 200:
                 return resp.json()["candidates"][0]["content"]["parts"][0]["text"].strip()
+            else:
+                logger.warning(f"Gemini erro: {resp.status_code} - {resp.text}")
+                return None
         except Exception as e:
             logger.warning(f"Gemini falhou: {e}")
-        return None
-    
+            return None
+
     # === API 3: GROQ ===
     def _chamar_groq(self, prompt: str) -> str:
         """Chama Groq (ultra-rápido)"""
@@ -197,12 +181,16 @@ AKIRA (responda EM 1-2 LINHAS curtas, sem markdown):"""
                 headers=headers,
                 timeout=self.timeout
             )
+            logger.debug(f"Groq response: {resp.status_code} - {resp.text[:100]}...")
             if resp.status_code == 200:
                 return resp.json()["choices"][0]["message"]["content"].strip()
+            else:
+                logger.warning(f"Groq erro: {resp.status_code} - {resp.text}")
+                return None
         except Exception as e:
             logger.warning(f"Groq falhou: {e}")
-        return None
-    
+            return None
+
     # === API 4: COHERE ===
     def _chamar_cohere(self, prompt: str) -> str:
         """Chama Cohere"""
@@ -220,12 +208,16 @@ AKIRA (responda EM 1-2 LINHAS curtas, sem markdown):"""
                 headers=headers,
                 timeout=self.timeout
             )
+            logger.debug(f"Cohere response: {resp.status_code} - {resp.text[:100]}...")
             if resp.status_code == 200:
                 return resp.json()["text"].strip()
+            else:
+                logger.warning(f"Cohere erro: {resp.status_code} - {resp.text}")
+                return None
         except Exception as e:
             logger.warning(f"Cohere falhou: {e}")
-        return None
-    
+            return None
+
     # === API 5: TOGETHER AI ===
     def _chamar_together(self, prompt: str) -> str:
         """Chama Together AI"""
@@ -243,12 +235,16 @@ AKIRA (responda EM 1-2 LINHAS curtas, sem markdown):"""
                 headers=headers,
                 timeout=self.timeout
             )
+            logger.debug(f"Together response: {resp.status_code} - {resp.text[:100]}...")
             if resp.status_code == 200:
                 return resp.json()["choices"][0]["message"]["content"].strip()
+            else:
+                logger.warning(f"Together erro: {resp.status_code} - {resp.text}")
+                return None
         except Exception as e:
             logger.warning(f"Together falhou: {e}")
-        return None
-    
+            return None
+
     # === API 6: HUGGING FACE ===
     def _chamar_huggingface(self, prompt: str) -> str:
         """Chama HuggingFace Inference API"""
@@ -261,67 +257,68 @@ AKIRA (responda EM 1-2 LINHAS curtas, sem markdown):"""
                 headers=headers,
                 timeout=self.timeout
             )
+            logger.debug(f"HF response: {resp.status_code} - {resp.text[:100]}...")
             if resp.status_code == 200:
                 return resp.json()[0]["generated_text"].split("AKIRA:")[-1].strip()
+            else:
+                logger.warning(f"HF erro: {resp.status_code} - {resp.text}")
+                return None
         except Exception as e:
             logger.warning(f"HuggingFace falhou: {e}")
-        return None
-    
+            return None
+
     # === MÉTODO PRINCIPAL DE GERAÇÃO ===
-    def gerar_resposta(self, mensagem: str, historico: list, mensagem_citada: str,
-                      humor: str, tom_usuario: str) -> str:
-        """
-        Tenta gerar resposta usando todas as APIs na ordem configurada
+    def gerar_resposta(self, mensagem: str, historico: list, mensagem_citada: str, humor: str, tom_usuario: str) -> str:
+        """ 
+        Tenta gerar resposta usando todas as APIs na ordem configurada.
+        - 2 retries por API
+        - Até 2 loops full se todas falharem
+        - Logs detalhados
         """
         prompt = self._construir_prompt(mensagem, historico, mensagem_citada, humor, tom_usuario)
-        
-        # Tenta cada API na ordem de fallback
-        for api_name in config.API_FALLBACK_ORDER:
-            if api_name not in self.apis_disponiveis:
-                continue
-            
-            logger.info(f"Tentando {api_name.upper()}...")
-            
-            try:
-                if api_name == "mistral":
-                    resposta = self._chamar_mistral(prompt)
-                elif api_name == "gemini":
-                    resposta = self._chamar_gemini(prompt)
-                elif api_name == "groq":
-                    resposta = self._chamar_groq(prompt)
-                elif api_name == "cohere":
-                    resposta = self._chamar_cohere(prompt)
-                elif api_name == "together":
-                    resposta = self._chamar_together(prompt)
-                elif api_name == "huggingface":
-                    resposta = self._chamar_huggingface(prompt)
-                
-                if resposta:
-                    logger.success(f"✓ Resposta gerada via {api_name.upper()}")
-                    return self._limpar_resposta(resposta)
-            
-            except Exception as e:
-                logger.error(f"{api_name} erro crítico: {e}")
-        
-        # Se todas falharem
-        return "Barra no bardeado."
-    
+        max_loops = 2
+        for loop in range(max_loops):
+            logger.info(f"Fallback loop {loop+1}/{max_loops}")
+            for api_name in config.API_FALLBACK_ORDER:
+                if api_name not in self.apis_disponiveis:
+                    continue
+                for retry in range(2):  # 2 tentativas por API
+                    logger.info(f"Tentando {api_name.upper()} (retry {retry+1}/2)...")
+                    try:
+                        if api_name == "mistral":
+                            resposta = self._chamar_mistral(prompt)
+                        elif api_name == "gemini":
+                            resposta = self._chamar_gemini(prompt)
+                        elif api_name == "groq":
+                            resposta = self._chamar_groq(prompt)
+                        elif api_name == "cohere":
+                            resposta = self._chamar_cohere(prompt)
+                        elif api_name == "together":
+                            resposta = self._chamar_together(prompt)
+                        elif api_name == "huggingface":
+                            resposta = self._chamar_huggingface(prompt)
+                        if resposta:
+                            logger.success(f"✓ Resposta gerada via {api_name.upper()}")
+                            return self._limpar_resposta(resposta)
+                        time.sleep(1)  # Sleep entre retries
+                    except Exception as e:
+                        logger.error(f"{api_name} erro crítico (retry {retry+1}): {e}")
+            time.sleep(2)  # Sleep entre loops full
+        return "Barra no bardeado"
+
     def _limpar_resposta(self, resposta: str) -> str:
         """Remove markdown e limita tamanho"""
         # Remove markdown
         resposta = resposta.replace("**", "").replace("*", "")
         resposta = resposta.replace("```", "").replace("`", "")
-        
         # Remove prefixos comuns de IA
         prefixos = ["AKIRA:", "Akira:", "RESPOSTA:", "Resposta:"]
         for p in prefixos:
             if resposta.startswith(p):
                 resposta = resposta[len(p):].strip()
-        
         # Limita tamanho (máximo 300 caracteres)
         if len(resposta) > 300:
             resposta = resposta[:297] + "..."
-        
         return resposta.strip()
 
 # ============================================================================
@@ -338,7 +335,7 @@ class AkiraAPI:
         self.web_search = get_web_search()  # Instância de WebSearch
         self._setup_routes()
         self._setup_trainer()
-    
+
     def _setup_trainer(self):
         """Inicializa treinamento (desativado por padrão)"""
         if getattr(self.config, 'START_PERIODIC_TRAINER', False):
@@ -347,10 +344,9 @@ class AkiraAPI:
                 logger.info("Treinamento periódico INICIADO")
             except Exception as e:
                 logger.error(f"Treinador falhou: {e}")
-    
+
     def _setup_routes(self):
         """Configura rotas Flask"""
-        
         @self.api.before_request
         def handle_options():
             if request.method == 'OPTIONS':
@@ -359,12 +355,12 @@ class AkiraAPI:
                 resp.headers['Access-Control-Allow-Headers'] = 'Content-Type'
                 resp.headers['Access-Control-Allow-Methods'] = 'POST, GET'
                 return resp
-        
+
         @self.api.after_request
         def add_cors(response):
             response.headers['Access-Control-Allow-Origin'] = '*'
             return response
-        
+
         @self.api.route('/akira', methods=['POST'])
         def akira_endpoint():
             try:
@@ -373,21 +369,16 @@ class AkiraAPI:
                 numero = data.get('numero', '').strip()
                 mensagem = data.get('mensagem', '').strip()
                 mensagem_citada = data.get('mensagem_citada', '').strip()
-                
                 if not mensagem and not mensagem_citada:
                     return jsonify({'error': 'mensagem obrigatória'}), 400
-                
                 logger.info(f"[{usuario}] ({numero}): {mensagem[:60]}")
-                
                 # === RESPOSTA RÁPIDA PARA HORA ===
                 if any(k in mensagem.lower() for k in ["hora", "horas", "que horas"]):
                     agora = datetime.datetime.now()
                     return jsonify({'resposta': f"São {agora.strftime('%H:%M')} em Luanda, puto."})
-                
                 # === DETECTAR INTENÇÃO DE BUSCA WEB ===
                 intencao_busca = WebSearch.detectar_intencao_busca(mensagem)
                 contexto_web = ""
-                
                 if intencao_busca == "noticias":
                     logger.info("Buscando notícias de Angola...")
                     contexto_web = self.web_search.pesquisar_noticias_angola()
@@ -404,21 +395,17 @@ class AkiraAPI:
                 elif intencao_busca == "busca_geral":
                     logger.info("Buscando informações gerais...")
                     contexto_web = self.web_search.buscar_geral(mensagem)
-                
                 # === CONTEXTO DO USUÁRIO ===
                 contexto = self._get_user_context(numero)
                 historico = contexto.obter_historico_para_llm()
-                
                 # === ANÁLISE DE TOM E HUMOR ===
                 analise = contexto.analisar_intencao_e_normalizar(mensagem, historico)
                 tom_usuario = analise.get("estilo", "casual")
                 humor_atual = contexto.obter_emocao_atual()
-                
                 # === VERIFICAR SE É USUÁRIO PRIVILEGIADO ===
                 if numero in config.USUARIOS_PRIVILEGIADOS:
                     tom_usuario = "formal"
                     logger.info(f"Usuário privilegiado detectado: {config.USUARIOS_PRIVILEGIADOS[numero]}")
-                
                 # === GERAR RESPOSTA VIA MULTI-API ===
                 # Adiciona contexto web ao histórico se disponível
                 historico_com_web = historico.copy()
@@ -427,7 +414,6 @@ class AkiraAPI:
                         "role": "system",
                         "content": f"CONTEXTO ADICIONAL (busca web):\n{contexto_web}"
                     })
-                
                 resposta = self.llm_manager.gerar_resposta(
                     mensagem=mensagem,
                     historico=historico_com_web,
@@ -435,10 +421,8 @@ class AkiraAPI:
                     humor=humor_atual,
                     tom_usuario=tom_usuario
                 )
-                
                 # === SALVAR NO BANCO + CONTEXTO ===
                 contexto.atualizar_contexto(mensagem, resposta)
-                
                 try:
                     trainer = Treinamento(self.db)
                     trainer.registrar_interacao(
@@ -451,17 +435,15 @@ class AkiraAPI:
                     )
                 except Exception as e:
                     logger.warning(f"Erro ao treinar: {e}")
-                
                 return jsonify({'resposta': resposta})
-            
             except Exception as e:
                 logger.exception("Erro em /akira")
                 return jsonify({'resposta': 'Erro interno, já volto!'}), 500
-        
+
         @self.api.route('/health', methods=['GET'])
         def health_check():
             return 'OK', 200
-    
+
     def _get_user_context(self, numero: str) -> Contexto:
         """Retorna contexto do usuário (com cache)"""
         if not numero: