Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 import os
 import google.generativeai as genai
 import json
+from google.generativeai.tool import GoogleSearch
 
 # --- 1. CARREGAMENTO DO CONTEXTO FILOSÓFICO IMUTÁVEL ---
 def carregar_contexto_filosofico(caminho_arquivo="epct0.md"):
@@ -20,51 +21,50 @@ def carregar_contexto_filosofico(caminho_arquivo="epct0.md"):
 CONTEXTO_EPICTETO = carregar_contexto_filosofico()
 
 
-# --- 2. CONFIGURAÇÃO DA API GEMINI ---
+# --- 2. CONFIGURAÇÃO DAS APIS GEMINI ---
 api_key = os.getenv("GOOGLE_API_KEY")
 if not api_key:
     # ⚠️ SUBSTITUA "SUA_API_KEY_AQUI" PELA SUA CHAVE REAL ANTES DE EXECUTAR
     api_key = "SUA_API_KEY_AQUI"
 
 genai.configure(api_key=api_key)
-model = genai.GenerativeModel('gemini-flash-latest')
 
+# Inicializa os dois modelos separadamente
+# O Conselheiro, para raciocínio profundo
+counselor_model = genai.GenerativeModel('gemini-flash-latest')
+# O Supervisor, rápido e com ferramenta de busca para fact-checking
+supervisor_model = genai.GenerativeModel('gemini-flash-latest', tools=[GoogleSearch()])
 
-# --- 3. PROMPT DE SISTEMA METODOLÓGICO (O PROTOCOLO PADRÃO) ---
+
+# --- 3. DEFINIÇÃO DOS PROMPTS ---
 PROMPT_SISTEMA_METODOLOGICO = """
 Você é um especialista em análise de intenções que opera sob um protocolo de 6 passos.
 Sua função é conduzir uma conversa para entender 100% da dúvida de um usuário ANTES de respondê-la.
 Sua SAÍDA DEVE SER SEMPRE E SOMENTE um objeto JSON válido.
 
 **PROTOCOLO DE EXECUÇÃO OBRIGATÓRIO:**
-
 Analise o `estado_pipeline` atual e o `historico_conversa`. Decida qual passo executar. Sua resposta em JSON deve conter o estado COMPLETAMENTE ATUALIZADO da pipeline.
 
 ---
 **Passo 1: Análise Primária (Clareza)**
 - **Meta:** A dúvida é compreensível?
 - **Decisão:** Se a clareza for 'baixa', pergunte por mais contexto. Senão, defina `proximo_passo` como `"passo_2_proposito"`.
-
 ---
 **Passo 2: Análise de Propósito (Para Que)**
 - **Meta:** Para qual fim a resposta servirá?
 - **Decisão:** Se a confiança no propósito for 'baixa', pergunte sobre o objetivo. Senão, defina `proximo_passo` como `"passo_3_motivacao"`.
-
 ---
 **Passo 3: Análise de Motivação (Porquê) e Interesses**
 - **Meta:** Por que o usuário precisa da resposta agora?
 - **Decisão:** Se a confiança na motivação for 'baixa', pergunte sobre o que despertou o interesse. Senão, defina `proximo_passo` como `"passo_4_coerencia"`.
-
 ---
 **Passo 4: Análise de Coerência**
 - **Meta:** As informações coletadas fazem sentido juntas?
 - **Decisão:** Se a coerência for 'baixa', pergunte para resolver a contradição. Senão, defina `proximo_passo` como `"passo_5_ambiguidade"`.
-
 ---
 **Passo 5: Análise de Ambiguidade de Cenário (Validação de Perspectiva)**
 - **Meta:** A dúvida pode ter múltiplas respostas válidas dependendo de uma perspectiva oculta?
 - **Decisão:** Se existem vários cenários de resposta válidos (confiança 'baixa'), formule uma pergunta que force o usuário a escolher uma perspectiva. Defina `proximo_passo` como `"aguardando_usuario"`. Se o cenário é único (confiança 'alta'), defina `proximo_passo` como `"passo_6_resposta_final"`.
-
 ---
 **Passo 6: Geração da Resposta Final**
 - **Meta:** Tenho 100% de clareza para dar uma resposta definitiva e sem subjetividade.
@@ -83,7 +83,15 @@ Analise o `estado_pipeline` atual e o `historico_conversa`. Decida qual passo ex
 }
 """
 
-# --- 4. GERENCIAMENTO DE ESTADO ---
+PROMPT_SUPERVISOR = """
+Você é um Supervisor de IA, um fact-checker rigoroso e objetivo. Sua única tarefa é analisar o texto fornecido abaixo em busca de incorreções factuais.
+- **Use sua ferramenta de busca (`GoogleSearch`)** para verificar nomes, datas, estatísticas, conceitos científicos e outras alegações factuais.
+- Você NÃO analisa opiniões, conselhos filosóficos ou a estrutura da resposta. Apenas fatos.
+- Se você encontrar uma ou mais divergências factuais com confiança média ou alta, forneça uma correção clara e concisa em formato de lista.
+- **Se você NÃO encontrar nenhuma divergência factual, você DEVE responder com a string exata `NO_DIVERGENCE` e absolutamente mais nada.**
+"""
+
+# --- 4. GERENCIAMENTO DE ESTADO E FUNÇÕES AUXILIARES ---
 def resetar_estado():
     """Inicializa ou reseta o dicionário de estado da pipeline."""
     print("Resetando o estado da pipeline para uma nova conversa.")
@@ -97,13 +105,25 @@ def resetar_estado():
 estado_pipeline_global = resetar_estado()
 historico_conversa_llm = []
 
+def run_supervisor(text_to_check):
+    """Aciona o modelo Supervisor para verificar os fatos de um texto."""
+    print("--- Acionando o Supervisor para verificação de fatos ---")
+    try:
+        full_prompt = f"{PROMPT_SUPERVISOR}\n\n--- TEXTO PARA ANÁLISE ---\n{text_to_check}"
+        response = supervisor_model.generate_content(full_prompt)
+        
+        if "NO_DIVERGENCE" in response.text:
+            print("Supervisor não encontrou divergências factuais.")
+            return None
+        else:
+            print(f"Supervisor encontrou uma divergência: {response.text}")
+            return response.text
+    except Exception as e:
+        print(f"Erro durante a execução do Supervisor: {e}")
+        return None
 
 # --- 5. FUNÇÃO PRINCIPAL DO CHATBOT ---
-def handle_chat(mensagem, historico_chat_gradio, modo_sabio):
-    """
-    Esta função é chamada a cada mensagem do usuário, gerenciando o estado,
-    construindo o prompt, chamando a API e retornando a resposta.
-    """
+def handle_chat(mensagem, historico_chat_gradio, modo_sabio, modo_supervisor):
     global estado_pipeline_global, historico_conversa_llm
 
     if not historico_chat_gradio:
@@ -119,19 +139,17 @@ def handle_chat(mensagem, historico_chat_gradio, modo_sabio):
     if modo_sabio and CONTEXTO_EPICTETO:
         prompt_base = f"""
 **AVISO: MODO SÁBIO ATIVADO. DIRETRIZES DE CONTEXTO IMUTÁVEIS SEGUEM ABAIXO.**
-
-Sua tarefa é agir como um conselheiro filosófico. Antes de qualquer raciocínio, você DEVE consultar, refletir e basear suas respostas nos princípios contidos no texto de referência abaixo. Este texto é sua principal fonte de verdade e guia ético para contornar vieses e fornecer sabedoria prática. Ele não é para ser resumido para o usuário; é para ser a base da sua reflexão interna.
+Sua tarefa é agir como um conselheiro filosófico. Antes de qualquer raciocínio, você DEVE consultar e refletir sobre os princípios no texto de referência abaixo. Este texto é seu guia ético para contornar vieses. Ele é para sua reflexão interna, não para resumir ao usuário.
 
 --- INÍCIO DO TEXTO DE REFERÊNCIA (epct0.md) ---
 {CONTEXTO_EPICTETO}
 --- FIM DO TEXTO DE REFERÊNCIA ---
 
 Agora, tendo refletido sobre estes princípios, aplique-os ao executar sua metodologia de análise.
-
 {PROMPT_SISTEMA_METODOLOGICO}
 """
     elif modo_sabio and not CONTEXTO_EPICTETO:
-        return "O 'Modo Sábio' não pôde ser ativado porque o arquivo de contexto `epct0.md` não foi encontrado. Operando no modo padrão."
+        return "O 'Modo Sábio' não pôde ser ativado porque o arquivo `epct0.md` não foi encontrado. Operando no modo padrão."
 
     prompt_completo = f"""
     {prompt_base}
@@ -146,25 +164,32 @@ Agora, tendo refletido sobre estes princípios, aplique-os ao executar sua metod
     """
 
     try:
-        response = model.generate_content(prompt_completo)
-        resposta_texto = response.text.strip().replace("```json", "").replace("```", "")
+        response_counselor = counselor_model.generate_content(prompt_completo)
+        resposta_texto = response_counselor.text.strip().replace("```json", "").replace("```", "")
         decisao_json = json.loads(resposta_texto)
         estado_pipeline_global = decisao_json.get("estado_pipeline_atualizado", estado_pipeline_global)
         
+        resposta_para_usuario = None
         if decisao_json.get("resposta_final"):
             resposta_para_usuario = decisao_json["resposta_final"]
-            historico_conversa_llm.append({"role": "model", "parts": [resposta_para_usuario]})
-            return resposta_para_usuario
         elif decisao_json.get("pergunta_para_usuario"):
             resposta_para_usuario = decisao_json["pergunta_para_usuario"]
-            historico_conversa_llm.append({"role": "model", "parts": [resposta_para_usuario]})
-            return resposta_para_usuario
         else:
-            return "Ocorreu um erro no meu raciocínio. Poderia tentar reformular?"
+            resposta_para_usuario = "Ocorreu um erro no meu raciocínio. Poderia tentar reformular?"
+        
+        historico_conversa_llm.append({"role": "model", "parts": [resposta_para_usuario]})
+
+        if modo_supervisor and decisao_json.get("resposta_final"):
+            feedback_supervisor = run_supervisor(resposta_para_usuario)
+            if feedback_supervisor:
+                resposta_para_usuario += f"\n\n---\n\n**🔍 Análise do Supervisor:**\n{feedback_supervisor}"
+        
+        return resposta_para_usuario
+
     except (json.JSONDecodeError, AttributeError, Exception) as e:
         print(f"Erro ao processar resposta da API: {e}")
         try:
-            print(f"Resposta recebida (pode ter causado o erro): {response.text}")
+            print(f"Resposta recebida (pode ter causado o erro): {response_counselor.text}")
         except:
             print("Não foi possível extrair o texto da resposta do erro.")
         return "Desculpe, tive um problema técnico. Por favor, clique no botão 'Limpar' e tente novamente."
@@ -173,18 +198,19 @@ Agora, tendo refletido sobre estes princípios, aplique-os ao executar sua metod
 # --- 6. INTERFACE GRÁFICA COM GRADIO ---
 iface = gr.ChatInterface(
     fn=handle_chat,
-    title="🤖 Conselheiro de Raciocínio v5 (Guiado por Contexto)",
-    description="Faça sua pergunta. Para uma orientação guiada por um framework filosófico, ative o 'Modo Sábio'.",
+    title="🤖 Conselheiro de Raciocínio v6 (com Supervisor)",
+    description="Faça sua pergunta. Ative o 'Modo Sábio' para orientação filosófica e o 'Modo Supervisor' para verificação de fatos.",
     chatbot=gr.Chatbot(height=600, label="Diálogo"),
     textbox=gr.Textbox(placeholder="Qual é a sua inquietação?", container=False, scale=7),
     additional_inputs=[
-        gr.Checkbox(label="🧘 Ativar Modo Sábio (Guiado por Contexto de Epicteto)", value=False)
+        gr.Checkbox(label="🧘 Ativar Modo Sábio (Guiado por Contexto)", value=False),
+        gr.Checkbox(label="🔍 Ativar Modo Supervisor (Verificação de Fatos)", value=False)
     ],
     theme="soft",
     examples=[
-        ["Meu chefe me deu um feedback injusto."],
-        ["Estou com medo de falhar em um novo projeto."],
-        ["Como posso parar de me preocupar com a opinião dos outros?"]
+        ["Qual a distância da Terra até a Lua e como isso afeta minhas finanças?"],
+        ["O Monte Everest é o ponto mais distante do centro da Terra?"],
+        ["Meu amigo me tratou mal. Como devo reagir segundo a filosofia estóica?"]
     ],
     cache_examples=False
 )
@@ -193,4 +219,4 @@ iface = gr.ChatInterface(
 if __name__ == "__main__":
     iface.launch()
 
-# --- FIM DO CÓDIGO ---
+# --- FIM DO CÓDIGO ---```