Update modules/api.py

modules/api.py

@@ -1,4 +1,3 @@
-# modules/api.py — VERSÃO FINAL OFICIAL: Contexto Fixo, Web Search Ativo, Akira Viva!
 """
 API wrapper Akira IA.
 Prioridade: Mistral API (Phi-3 Mini) → Gemini → Fallback
@@ -19,7 +18,7 @@ import google.generativeai as genai
 from mistralai import Mistral
 
 # LOCAL MODULES
-# from .local_llm import HermesLLM  # ← REMOVIDO: Era o modelo que causava a carga de 101% CPU
+# from .local_llm import HermesLLM # ← REMOVIDO: Era o modelo que causava a carga de 101% CPU
 from .contexto import Contexto
 from .database import Database
 from .treinamento import Treinamento
@@ -64,11 +63,6 @@ class LLMManager:
         self.providers = []
 
         # PRIORIDADE MÁXIMA AGORA É O MISTRAL (PHI-3 MINI)
-        # if HermesLLM.is_available(): # REMOVIDO PELA CARGA DE CPU
-        #     self.hermes_available = True
-        #     self.providers.append('hermes')
-        #     logger.info("HERMES 7B LOCAL (GGUF + LoRA ANGOLANO) ATIVO → PRIORIDADE MÁXIMA → 8-12s RESPOSTA!")
-        
         if self.mistral_client:
             self.providers.append('mistral') # Mistral (usando Phi-3) é o novo principal
         if self.gemini_model:
@@ -127,11 +121,16 @@ class LLMManager:
             messages.append({"role": role, "content": turn["content"]})
         
         # Extrai a mensagem limpa do prompt (necessário para APIs)
-        user_message_clean_match = re.search(r'### Mensagem Atual ###\n(.*?)\n\nAkira:', user_prompt, re.DOTALL)
+        # O prompt completo é formatado em _build_prompt, mas as APIs usam o formato de messages.
+        # Precisamos extrair apenas a última mensagem do usuário do prompt longo para garantir que
+        # o histórico (que já está em context_history) não seja duplicado.
+        user_message_clean_match = re.search(r'(### Mensagem Atual ###|### USUÁRIO RESPONDEU A ESSA MENSAGEM: ###)\n(.*?)\n\n(Akira:|$)', user_prompt, re.DOTALL)
         if user_message_clean_match:
-            user_message_clean = user_message_clean_match.group(1).strip()
+            # Captura o grupo 2 (o conteúdo da mensagem)
+            user_message_clean = user_message_clean_match.group(2).strip() 
         else:
-            user_message_clean = user_prompt # Fallback
+            # Fallback (caso o formato do prompt mude)
+            user_message_clean = user_prompt 
             
         messages.append({"role": "user", "content": user_message_clean})
         
@@ -140,10 +139,6 @@ class LLMManager:
 
 
         for provider in self.providers:
-            # 1. HERMES LOCAL → PULADO (REMOVIDO DO __init__)
-            # if provider == 'hermes' and self.hermes_available:
-            #     ...
-
             # 1. MISTRAL API (AGORA PRIORIDADE MÁXIMA)
             if provider == 'mistral' and self.mistral_client:
                 try:
@@ -171,6 +166,7 @@ class LLMManager:
                     if getattr(self.config, 'GEMINI_API_KEY', '').startswith('AIza'):
                          genai.configure(api_key=self.config.GEMINI_API_KEY)
                          
+                    # Cria o histórico no formato esperado pelo Gemini (list of Content)
                     gemini_hist = []
                     for msg in messages[1:]:
                         role = "user" if msg["role"] == "user" else "model"
@@ -245,15 +241,38 @@ class AkiraAPI:
                 
                 # CORREÇÃO: Verifica se o método existe antes de chamar
                 if hasattr(trainer, 'start_periodic_training'):
-                     trainer.start_periodic_training()
-                     self.logger.info("Treinamento periódico (start_periodic_training) iniciado com sucesso.")
+                    trainer.start_periodic_training()
+                    self.logger.info("Treinamento periódico (start_periodic_training) iniciado com sucesso.")
                 else:
-                     self.logger.info("Treinamento periódico (via __init__) iniciado.")
-                     
+                    self.logger.info("Treinamento periódico (via __init__) iniciado.")
+                    
             except Exception as e:
                 self.logger.exception(f"Treinador periódico falhou ao iniciar: {e}")
 
     def _setup_routes(self):
+        """
+        Configura as rotas da API, incluindo o tratamento de CORS.
+        """
+        # --- CORREÇÃO: Adiciona suporte manual a CORS ---
+        # 1. CORS Preflight Handler (Responde a requests OPTIONS)
+        @self.api.before_request
+        def handle_options():
+            if request.method == 'OPTIONS':
+                response = self.app.make_response('')
+                response.headers.add('Access-Control-Allow-Origin', '*')
+                response.headers.add('Access-Control-Allow-Headers', 'Content-Type, Authorization')
+                response.headers.add('Access-Control-Allow-Methods', 'POST, GET, OPTIONS')
+                return response
+
+        # 2. CORS Post-Request Header Addition (Adiciona headers em toda resposta)
+        @self.api.after_request
+        def add_cors_headers(response):
+            response.headers.add('Access-Control-Allow-Origin', '*')
+            response.headers.add('Access-Control-Allow-Headers', 'Content-Type, Authorization')
+            response.headers.add('Access-Control-Allow-Methods', 'POST, GET, OPTIONS')
+            return response
+        # ------------------------------------------------
+
         @self.api.route('/akira', methods=['POST'])
         def akira_endpoint():
             try:
@@ -272,14 +291,17 @@ class AkiraAPI:
 
                 # --- CORREÇÃO: Resposta rápida para "Que dia é hoje?" ---
                 prompt_lower = mensagem.lower().strip()
-                if any(keyword in prompt_lower for keyword in ["que dia é hoje", "qual é a data", "dia da semana"]):
+                if any(keyword in prompt_lower for keyword in ["que dia é hoje", "qual é a data", "dia da semana", "que horas"]):
                     hoje = datetime.datetime.now()
                     dia_semana = hoje.strftime("%A")
                     dia_mes = hoje.day
                     mes = hoje.strftime("%B")
                     ano = hoje.year
+                    hora_minuto = hoje.strftime("%H:%M")
                     
-                    if any(k in prompt_lower for k in ["que dia", "hoje é que dia", "dia da semana"]) and not any(k in prompt_lower for k in ["mês", "ano", "data", "completa"]):
+                    if "que horas" in prompt_lower:
+                        resposta = f"São {hora_minuto} agora, meu."
+                    elif any(k in prompt_lower for k in ["que dia", "hoje é que dia", "dia da semana"]) and not any(k in prompt_lower for k in ["mês", "ano", "data", "completa"]):
                         resposta = f"Hoje é {dia_semana.capitalize()}, {dia_mes}, meu."
                     else:
                         resposta = f"Hoje é {dia_semana.capitalize()}, {dia_mes} de {mes.capitalize()} de {ano}, meu."
@@ -373,7 +395,7 @@ class AkiraAPI:
         web_search_context = ""
         
         # Palavras-chave que sugerem necessidade de informação em tempo real ou muito específica
-        trigger_keywords = ['hoje', 'agora', 'recente', 'notícias', 'busca na web', 'pesquisa', 'investiga']
+        trigger_keywords = ['hoje', 'agora', 'recente', 'notícias', 'busca na web', 'pesquisa', 'investiga', 'último']
         
         search_query = f"{mensagem} {mensagem_citada}".strip().lower()
 
@@ -443,9 +465,10 @@ class AkiraAPI:
             temperature = getattr(self.config, 'TOP_P', 0.8) 
             
             # Extrai a mensagem limpa do prompt (necessário para APIs)
-            user_prompt_clean_match = re.search(r'### Mensagem Atual ###\n(.*?)\n\nAkira:', prompt, re.DOTALL)
+            # Usa o mesmo regex do LLMManager.generate para manter a consistência
+            user_prompt_clean_match = re.search(r'(### Mensagem Atual ###|### USUÁRIO RESPONDEU A ESSA MENSAGEM: ###)\n(.*?)\n\n(Akira:|$)', prompt, re.DOTALL)
             if user_prompt_clean_match:
-                user_prompt_clean = user_prompt_clean_match.group(1).strip()
+                user_prompt_clean = user_prompt_clean_match.group(2).strip()
             else:
                 user_prompt_clean = prompt # Fallback