Update app.py

app.py

@@ -1,6 +1,6 @@
 # ╔════════════════════════════════════════════════════════════════════════════╗
-# ║             PIPELINE v25: ORQUESTRAÇÃO COM UPLOAD DE EVIDÊNCIAS              ║
-# ║     Novidade: Upload de arquivos de texto (.txt, .md, .csv) no input         ║
+# ║             PIPELINE v26: INTERFACE CHAT MODERNIZADA                       ║
+# ║        Layout: Chatbot -> Inputs -> DNA (JSON) -> Logs Brutos              ║
 # ╚════════════════════════════════════════════════════════════════════════════╝
 
 import os
@@ -13,10 +13,12 @@ import google.generativeai as genai
 
 # ==================== 1. CONFIGURAÇÃO ====================
 api_key = os.getenv("GOOGLE_API_KEY", "SUA_API_KEY_AQUI")
-genai.configure(api_key=api_key)
 
-model_flash = genai.GenerativeModel("gemini-flash-latest")
-model_pro   = genai.GenerativeModel("gemini-pro-latest")
+if api_key:
+    genai.configure(api_key=api_key)
+
+model_flash = genai.GenerativeModel("gemini-1.5-flash-latest")
+model_pro   = genai.GenerativeModel("gemini-1.5-pro-latest")
 
 ARQUIVO_CONFIG = "protocolo.json"
 
@@ -29,27 +31,17 @@ def carregar_protocolo_padrao():
 
 def salvar_protocolo_disco(conteudo_json):
     try:
-        json.loads(conteudo_json) # Valida JSON
+        json.loads(conteudo_json)
         with open(ARQUIVO_CONFIG, "w", encoding="utf-8") as f: f.write(conteudo_json)
-        return "✅ Salvo!"
+        return "✅ Protocolo Salvo!"
     except Exception as e: return f"❌ Erro: {e}"
 
 def ler_arquivo_input(arquivo):
-    """
-    Lê o conteúdo de um arquivo de texto enviado pelo usuário.
-    """
-    if arquivo is None: 
-        return ""
-    
+    if arquivo is None: return ""
     try:
-        # Tenta ler como UTF-8
         with open(arquivo.name, "r", encoding="utf-8") as f:
-            conteudo = f.read()
-            return f"\n\n=== CONTEÚDO DO ARQUIVO ANEXADO ({os.path.basename(arquivo.name)}) ===\n{conteudo}\n=======================================\n"
-    except UnicodeDecodeError:
-        return "\n\n[ERRO: O arquivo anexado não é um texto válido ou tem codificação desconhecida.]"
-    except Exception as e:
-        return f"\n\n[ERRO AO LER ARQUIVO: {str(e)}]"
+            return f"\n\n=== ANEXO: {os.path.basename(arquivo.name)} ===\n{f.read()}\n=================\n"
+    except: return "\n[Erro ao ler arquivo]"
 
 # ==================== 3. ENGINE DE EXECUÇÃO ====================
 
@@ -63,16 +55,9 @@ def executar_no(timeline: list, config: dict):
     modelo_ativo = model_pro if config.get("modelo") == "pro" else model_flash
     
     contexto_str = json.dumps(timeline, ensure_ascii=False, indent=2)
-    prompt_bruto = f"""
---- TIMELINE DO CASO ---
-{contexto_str}
-------------------------
-AGENTE: {config['nome']}
-MISSÃO: {config['missao']}
-    """
+    prompt_bruto = f"--- TIMELINE ---\n{contexto_str}\n----------------\nAGENTE: {config['nome']}\nMISSÃO: {config['missao']}"
     
-    log_entry = f"\n{'='*40}\n🔶 EXECUTANDO: {config['nome']}\n{'='*40}\n"
-    log_entry += f"📤 PROMPT:\n{prompt_bruto[:1000]}... [truncado para visualização]\n"
+    log_entry = f"\n🔹 {config['nome']}:\n"
     
     try:
         inicio = time.time()
@@ -80,127 +65,157 @@ MISSÃO: {config['missao']}
         output_bruto = response.text
         tempo = time.time() - inicio
         
-        log_entry += f"📥 OUTPUT ({tempo:.2f}s):\n{output_bruto}\n"
+        log_entry += f"⏱️ {tempo:.2f}s | INPUT_LEN: {len(prompt_bruto)} | OUTPUT_LEN: {len(output_bruto)}\n"
         
         if config['tipo_saida'] == 'json': conteudo = limpar_json(output_bruto)
         else: conteudo = output_bruto
             
-        return {"role": "assistant", "agent": config['nome'], "content": conteudo}, log_entry
+        return {"role": "assistant", "agent": config['nome'], "content": conteudo}, log_entry, output_bruto
         
     except Exception as e:
-        return {"role": "system", "error": str(e)}, log_entry + f"\n❌ ERRO: {str(e)}\n"
+        return {"role": "system", "error": str(e)}, f"\n❌ ERRO: {str(e)}\n", str(e)
 
-# ==================== 4. ORQUESTRADOR ====================
+# ==================== 4. ORQUESTRADOR (ADAPTADO PARA CHAT) ====================
 
-def orquestrador(texto_caso, arquivo_anexo, json_protocolo_str):
+def orquestrador_chat(msg_texto, msg_arquivo, history, json_protocolo):
     """
-    Recebe texto + arquivo, combina ambos e inicia a pipeline.
+    Gerador que atualiza o Chatbot, o DNA e os Logs passo a passo.
     """
-    # 1. Processa o arquivo anexado
-    conteudo_anexo = ler_arquivo_input(arquivo_anexo)
-    
-    # 2. Combina Inputs
-    input_completo = f"{texto_caso}\n{conteudo_anexo}".strip()
+    # 1. Preparação
+    anexo_str = ler_arquivo_input(msg_arquivo)
+    input_total = f"{msg_texto}\n{anexo_str}".strip()
     
-    if not input_completo:
-        yield "⚠️ Erro: Insira texto ou anexe um arquivo.", "", {}
+    if not input_total:
+        yield history, {}, "Por favor, insira texto ou arquivo."
         return
 
+    # Adiciona mensagem do usuário no Chat
+    history = history + [[msg_texto + (" [Arquivo Anexado]" if msg_arquivo else ""), None]]
+    
     try:
-        protocolo = json.loads(json_protocolo_str)
-    except Exception as e:
-        yield f"⚠️ Erro no JSON de Configuração: {e}", "", {}
+        protocolo = json.loads(json_protocolo)
+    except:
+        history[-1][1] = "❌ Erro no JSON de configuração."
+        yield history, {}, "Erro JSON"
         return
 
-    # 3. Inicializa
-    info_anexo = "Sim" if conteudo_anexo else "Não"
-    logs_totais = f"🚀 INICIANDO (Anexo: {info_anexo})...\n"
-    timeline = [{"role": "user", "content": input_completo}]
-    relatorio_final = "⏳ Aguardando..."
+    # 2. Inicialização
+    timeline = [{"role": "user", "content": input_total}]
+    logs_brutos = f"🚀 INÍCIO: {datetime.now().strftime('%H:%M:%S')}\n"
     
-    yield logs_totais, relatorio_final, timeline
+    # Mostra status inicial no chat
+    history[-1][1] = "⏳ Iniciando orquestração..."
+    yield history, timeline, logs_brutos
 
-    # 4. Loop de Agentes
-    for config_agente in protocolo:
-        resultado, log_no = executar_no(timeline, config_agente)
+    # 3. Loop de Agentes
+    relatorio_acumulado = ""
+    
+    for config in protocolo:
+        # Atualiza status no chat
+        history[-1][1] = f"⚙️ {config['nome']} está pensando..."
+        yield history, timeline, logs_brutos
+        
+        # Executa
+        resultado, log_meta, raw_out = executar_no(timeline, config)
+        
         timeline.append(resultado)
-        logs_totais += log_no
+        logs_totais = logs_brutos + log_meta + f"OUTPUT RAW:\n{raw_out[:200]}...\n" # Log resumido
         
-        if config_agente['tipo_saida'] == 'texto':
-            relatorio_final = resultado['content']
+        # Se for o passo final (Texto), exibe no chat. Se não, mantém loading.
+        if config['tipo_saida'] == 'texto':
+            relatorio_acumulado = resultado['content']
+            history[-1][1] = relatorio_acumulado
         else:
-            relatorio_final = f"⏳ {config_agente['nome']} trabalhando..."
+            # Mostra progresso no DNA
+            pass
             
-        yield logs_totais, relatorio_final, timeline
+        yield history, timeline, logs_totais
 
-    logs_totais += "\n🏁 FINALIZADO."
-    yield logs_totais, relatorio_final, timeline
+    # Finalização
+    logs_totais += "\n✅ FIM."
+    yield history, timeline, logs_totais
 
-# ==================== 5. INTERFACE ====================
+# ==================== 5. INTERFACE UI (V26) ====================
 
-def construir_interface():
+def ui_v26():
     config_inicial = carregar_protocolo_padrao()
     
-    with gr.Blocks(title="Pipeline v25 - Com Upload", theme=gr.themes.Base()) as app:
-        gr.Markdown("# 🕵️‍♂️ Pipeline Forense v25: Input Multimodal (Texto + Arquivo)")
+    with gr.Blocks(title="Pipeline v26 - Chat", theme=gr.themes.Soft()) as app:
         
         with gr.Tabs():
-            
-            # === ABA EXECUÇÃO ===
-            with gr.TabItem("🚀 Execução"):
+            # === ABA 1: CHATBOT (PRINCIPAL) ===
+            with gr.TabItem("💬 Investigador Chat"):
+                
+                # 1. O Chatbot (Topo)
+                chatbot = gr.Chatbot(
+                    label="Diálogo Forense",
+                    height=550,
+                    show_copy_button=True,
+                    avatar_images=(None, "⚖️") # Opcional: ícones
+                )
+                
+                # 2. Área de Input (Meio)
                 with gr.Row():
-                    with gr.Column(scale=1):
-                        inp_caso = gr.Textbox(
-                            label="Instruções / Resumo do Caso", 
-                            lines=5, 
-                            placeholder="Digite instruções adicionais aqui..."
-                        )
-                        # NOVO COMPONENTE DE UPLOAD
-                        inp_arquivo = gr.File(
-                            label="Anexar Evidências (Texto/Logs/Transcrições)",
-                            file_types=[".txt", ".md", ".csv", ".json"],
-                            file_count="single"
+                    with gr.Column(scale=8):
+                        txt_input = gr.Textbox(
+                            show_label=False, 
+                            placeholder="Digite o caso aqui ou anexe um arquivo...",
+                            lines=2,
+                            container=False
                         )
-                        btn_executar = gr.Button("⚡ EXECUTAR ANÁLISE COMPLETA", variant="primary", size="lg")
-                    
-                    with gr.Column(scale=1):
-                        out_logs = gr.Textbox(
-                            label="📟 Console de Logs (Raw I/O)", 
-                            lines=15, 
-                            interactive=False,
-                            elem_id="logs"
+                    with gr.Column(scale=1, min_width=50):
+                        file_btn = gr.File(
+                            label="📎", 
+                            file_count="single", 
+                            type="filepath",
+                            scale=1,
+                            min_width=50
                         )
+                    with gr.Column(scale=1, min_width=80):
+                        send_btn = gr.Button("Enviar", variant="primary", scale=1, min_width=80)
+
+                # 3. O DNA (Abaixo)
+                gr.Markdown("### 🧬 DNA da Análise (Timeline JSON)")
+                out_dna = gr.JSON(label="Estrutura de Raciocínio")
                 
-                gr.Markdown("---")
-                
-                with gr.Row():
-                    out_relatorio = gr.Markdown(label="📄 Parecer Final")
-                    out_timeline = gr.JSON(label="🧠 Memória JSON")
+                # 4. Logs (Acordeão para não poluir)
+                with gr.Accordion("🛠️ Logs Brutos do Sistema (Debug)", open=False):
+                    out_logs = gr.Textbox(label="Terminal Output", lines=10)
 
-            # === ABA CONFIGURAÇÃO ===
-            with gr.TabItem("⚙️ Configuração (JSON)"):
+            # === ABA 2: CONFIGURAÇÃO (JSON) ===
+            with gr.TabItem("⚙️ Configurar Protocolo"):
                 with gr.Row():
-                    btn_save_config = gr.Button("💾 Salvar Protocolo")
-                    lbl_config = gr.Label()
+                    save_btn = gr.Button("💾 Salvar Alterações")
+                    status_lbl = gr.Label(show_label=False)
                 
-                editor_json = gr.Code(
-                    label="protocolo.json", 
-                    language="json", 
+                json_editor = gr.Code(
                     value=config_inicial, 
-                    lines=20
+                    language="json", 
+                    label="protocolo.json",
+                    lines=25
                 )
                 
-                btn_save_config.click(salvar_protocolo_disco, editor_json, lbl_config)
+                save_btn.click(salvar_protocolo_disco, json_editor, status_lbl)
 
-        # EVENTO DE EXECUÇÃO
-        btn_executar.click(
-            orquestrador,
-            inputs=[inp_caso, inp_arquivo, editor_json], # Passa Texto + Arquivo + Config
-            outputs=[out_logs, out_relatorio, out_timeline]
+        # === CONEXÕES ===
+        
+        # Enviar via Botão
+        send_btn.click(
+            orquestrador_chat,
+            inputs=[txt_input, file_btn, chatbot, json_editor],
+            outputs=[chatbot, out_dna, out_logs]
         )
         
+        # Enviar via Enter
+        txt_input.submit(
+            orquestrador_chat,
+            inputs=[txt_input, file_btn, chatbot, json_editor],
+            outputs=[chatbot, out_dna, out_logs]
+        )
+        
+        # Limpar input após envio (opcional, requer função extra, mantendo simples por agora)
+
     return app
 
 if __name__ == "__main__":
-    print("Iniciando v25...")
-    construir_interface().launch()
+    ui_v26().launch()