Upload app (17).py

app (17).py

@@ -0,0 +1,184 @@
+# ╔════════════════════════════════════════════════════════════════════════════╗
+# ║             PIPELINE v27: UI LIMPA & ESTRUTURA DE ABAS                     ║
+# ║        Layout: Chat (Aba 1) | Debug (Aba 2) | Config (Aba 3)               ║
+# ╚════════════════════════════════════════════════════════════════════════════╝
+
+import os
+import json
+import re
+import time
+from datetime import datetime
+import gradio as gr
+import google.generativeai as genai
+
+# ==================== 1. CONFIGURAÇÃO ====================
+api_key = os.getenv("GOOGLE_API_KEY", "SUA_API_KEY_AQUI")
+if api_key: genai.configure(api_key=api_key)
+
+model_flash = genai.GenerativeModel("gemini-2.0-flash")
+model_pro   = genai.GenerativeModel("gemini-2.0-pro")
+
+ARQUIVO_CONFIG = "protocolo.json"
+
+# ==================== 2. UTILIDADES ====================
+
+def carregar_protocolo():
+    try:
+        with open(ARQUIVO_CONFIG, "r", encoding="utf-8") as f: return f.read()
+    except: return "[]"
+
+def salvar_protocolo(conteudo):
+    try:
+        json.loads(conteudo)
+        with open(ARQUIVO_CONFIG, "w", encoding="utf-8") as f: f.write(conteudo)
+        return "✅ Salvo"
+    except: return "❌ Erro JSON"
+
+def ler_anexo(arquivo):
+    if arquivo is None: return ""
+    try:
+        with open(arquivo.name, "r", encoding="utf-8") as f:
+            return f"\n\n[ANEXO SISTEMA: {os.path.basename(arquivo.name)}]\n{f.read()}\n[FIM ANEXO]\n"
+    except: return ""
+
+# ==================== 3. ENGINE DE EXECUÇÃO ====================
+
+def executar_no(timeline, config):
+    modelo = model_pro if config.get("modelo") == "pro" else model_flash
+    contexto = json.dumps(timeline, ensure_ascii=False, indent=2)
+    prompt = f"--- TIMELINE ---\n{contexto}\n----------------\nAGENTE: {config['nome']}\nMISSÃO: {config['missao']}"
+    
+    log = f"\n🔸 {config['nome']}..."
+    try:
+        inicio = time.time()
+        resp = modelo.generate_content(prompt)
+        out = resp.text
+        tempo = time.time() - inicio
+        
+        content = json.loads(out.strip().replace('```json','').replace('```','')) if config['tipo_saida']=='json' else out
+        log += f" (OK - {tempo:.2f}s)"
+        return {"role": "assistant", "agent": config['nome'], "content": content}, log, out
+    except Exception as e:
+        return {"role": "system", "error": str(e)}, f" (ERRO: {e})", str(e)
+
+# ==================== 4. ORQUESTRADOR ====================
+
+def orquestrador(texto, arquivo, history, json_config):
+    # 1. Input Check
+    anexo = ler_anexo(arquivo)
+    full_input = f"{texto}\n{anexo}".strip()
+    
+    if not full_input:
+        yield history, {}, "Sem input."
+        return
+
+    # 2. Setup
+    history = history + [[texto + (" 📎" if arquivo else ""), None]]
+    try: protocolo = json.loads(json_config)
+    except: 
+        history[-1][1] = "❌ Erro no JSON de Configuração."
+        yield history, {}, "Erro JSON"
+        return
+
+    timeline = [{"role": "user", "content": full_input}]
+    logs = f"🚀 START: {datetime.now().strftime('%H:%M:%S')}\n"
+    history[-1][1] = "⏳ Iniciando análise..."
+    yield history, timeline, logs
+
+    # 3. Loop
+    final_response = ""
+    for cfg in protocolo:
+        history[-1][1] = f"⚙️ {cfg['nome']} trabalhando..."
+        yield history, timeline, logs
+        
+        res, log_add, raw = executar_no(timeline, cfg)
+        timeline.append(res)
+        logs += log_add + "\n"
+        
+        if cfg['tipo_saida'] == 'texto':
+            final_response = res['content']
+            history[-1][1] = final_response # Mostra texto progressivo se fosse stream, aqui mostra final
+        
+        yield history, timeline, logs
+
+    logs += "✅ FIM."
+    yield history, timeline, logs
+
+# ==================== 5. UI LIMPA (v27) ====================
+
+def ui_clean():
+    css = """
+    footer {display: none !important;} 
+    .contain {border: none !important;}
+    """
+    
+    config_init = carregar_protocolo()
+
+    with gr.Blocks(title="AI Forensics", css=css, theme=gr.themes.Soft()) as app:
+        
+        with gr.Tabs():
+            
+            # === ABA 1: CHAT (LIMPO) ===
+            with gr.Tab("💬 Investigador"):
+                chatbot = gr.Chatbot(
+                    label="", 
+                    show_label=False, 
+                    height=600, 
+                    show_copy_button=True,
+                    render_markdown=True
+                )
+                
+                with gr.Row():
+                    with gr.Column(scale=10):
+                        txt_in = gr.Textbox(
+                            show_label=False, 
+                            placeholder="Descreva o caso ou instrução...", 
+                            lines=1,
+                            max_lines=5,
+                            container=False
+                        )
+                    with gr.Column(scale=1, min_width=50):
+                        file_in = gr.UploadButton(
+                            "📎", 
+                            file_types=[".txt", ".md", ".csv", ".json"],
+                            size="sm"
+                        )
+                    with gr.Column(scale=1, min_width=80):
+                        btn_send = gr.Button("Enviar", variant="primary", size="sm")
+                
+                # Feedback visual sutil do arquivo
+                file_status = gr.Markdown("", visible=True)
+                file_in.upload(lambda x: f"📎 Anexo: {os.path.basename(x.name)}", file_in, file_status)
+
+            # === ABA 2: DEPURAÇÃO (ESCONDIDO) ===
+            with gr.Tab("🕵️ Depuração"):
+                with gr.Row():
+                    out_dna = gr.JSON(label="DNA (Timeline)")
+                    out_logs = gr.Textbox(label="Logs do Sistema", lines=20)
+
+            # === ABA 3: CONFIG (TÉCNICO) ===
+            with gr.Tab("⚙️ Config"):
+                with gr.Row():
+                    btn_save = gr.Button("Salvar Config")
+                    lbl_save = gr.Label(show_label=False)
+                code_json = gr.Code(value=config_init, language="json", label="protocolo.json")
+                btn_save.click(salvar_protocolo, code_json, lbl_save)
+
+        # === TRIGGERS ===
+        # Enter ou Botão Enviar
+        triggers = [btn_send.click, txt_in.submit]
+        
+        for trig in triggers:
+            trig(
+                orquestrador,
+                inputs=[txt_in, file_in, chatbot, code_json],
+                outputs=[chatbot, out_dna, out_logs]
+            ).then(
+                lambda: (None, ""), # Limpa input e label do arquivo após envio
+                outputs=[txt_in, file_status] 
+            )
+
+    return app
+
+if __name__ == "__main__":
+    ui_clean().launch()