Update app.py

app.py

@@ -1,6 +1,6 @@
 # ╔════════════════════════════════════════════════════════════════════════════╗
-# ║             PIPELINE v33: DEBUG LOGS + GRADIO 5.x MESSAGES FORMAT         ║
-# ║                Prints completos: input/output Gemini + erros + fases      ║
+# ║          PIPELINE v34: GEMINI + GROQ + DEBUG LOGS                         ║
+# ║          Suporte dual: Google Gemini e Groq API                           ║
 # ╚════════════════════════════════════════════════════════════════════════════╝
 
 import os
@@ -10,20 +10,27 @@ import time
 from datetime import datetime
 import gradio as gr
 import google.generativeai as genai
+from groq import Groq
 
 # ==================== 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")
+# GEMINI
+gemini_key = os.getenv("GOOGLE_API_KEY", "SUA_API_KEY_AQUI")
+if gemini_key:
+    genai.configure(api_key=gemini_key)
+model_flash = genai.GenerativeModel("gemini-flash-latest")
 model_pro = genai.GenerativeModel("gemini-pro-latest")
 
+# GROQ
+groq_key = os.getenv("GROQ_API_KEY", "SUA_GROQ_KEY_AQUI")
+groq_client = Groq(api_key=groq_key) if groq_key else None
+
 ARQUIVO_CONFIG = "protocolo.json"
 ARQUIVO_HELP = "help.md"
-DELAY_ENTRE_AGENTES = 12  # 3 minutos em segundos
+DELAY_ENTRE_AGENTES = 180  # 3 minutos
 
-print("🚀 App inicializada. Modelos carregados: flash/pro")
+print("🚀 App inicializada")
+print(f"  ✅ Gemini: {'OK' if gemini_key else '❌ sem key'}")
+print(f"  ✅ Groq: {'OK' if groq_client else '❌ sem key'}")
 
 # ==================== 2. UTILIDADES ====================
 def carregar_protocolo():
@@ -68,31 +75,77 @@ def verificar_stop(texto):
     if not texto:
         return False
     stop_detectado = bool(re.search(r'\bstop\b', str(texto), re.IGNORECASE))
-    print(f"🛑 STOP detectado? {stop_detectado} em '{texto[:100]}...'")
+    print(f"🛑 STOP detectado? {stop_detectado} em '{str(texto)[:100]}...'")
     return stop_detectado
 
-# ==================== 3. ENGINE DE EXECUÇÃO ====================
+# ==================== 3. ENGINE DE EXECUÇÃO (DUAL) ====================
 def executar_no(timeline, config):
-    print(f"\n🔥 === EXECUTANDO {config['nome']} ({config.get('modelo', 'flash')}) ===")
-    modelo = model_pro if config.get("modelo") == "pro" else model_flash
+    provider = config.get('provider', 'gemini').lower()
+    print(f"\n🔥 === EXECUTANDO {config['nome']} ===")
+    print(f"   Provider: {provider.upper()} | Modelo: {config.get('modelo', 'auto')}")
+    
     contexto = json.dumps(timeline, ensure_ascii=False, indent=2)
     prompt = f"--- TIMELINE ---\n{contexto}\n----------------\nAGENTE: {config['nome']}\nMISSÃO: {config['missao']}"
     
-    print("📤 INPUT PROMPT (primeiros 800 chars):")
-    print(prompt[:800])
-    print("..." if len(prompt) > 800 else "")
+    print("📤 INPUT PROMPT (primeiros 600 chars):")
+    print(prompt[:600])
+    print("..." if len(prompt) > 600 else "")
     
     try:
         inicio = time.time()
-        resp = modelo.generate_content(prompt)
-        out_raw = resp.text
-        tempo_exec = time.time() - inicio
         
-        print(f"📥 OUTPUT BRUTO GEMINI ({len(out_raw)} chars):")
+        # ===== GROQ =====
+        if provider == 'groq':
+            if not groq_client:
+                raise Exception("Groq client não configurado (falta GROQ_API_KEY)")
+            
+            modelo_groq = config.get('modelo', 'meta-llama/llama-4-maverick-17b-128e-instruct')
+            
+            # Converte timeline para formato chat do Groq
+            messages = []
+            for msg in timeline:
+                if msg.get('role') in ['user', 'assistant']:
+                    messages.append({
+                        "role": msg['role'],
+                        "content": str(msg.get('content', ''))
+                    })
+            
+            # Adiciona prompt da missão como system message
+            messages_groq = [
+                {"role": "system", "content": f"AGENTE: {config['nome']}\nMISSÃO: {config['missao']}"}
+            ] + messages
+            
+            print(f"🔧 Groq messages: {len(messages_groq)} msgs")
+            
+            completion = groq_client.chat.completions.create(
+                model=modelo_groq,
+                messages=messages_groq,
+                temperature=1,
+                max_completion_tokens=2048,
+                top_p=1,
+                stream=True,
+                stop=None
+            )
+            
+            out_raw = ""
+            for chunk in completion:
+                out_raw += chunk.choices[0].delta.content or ""
+            
+            print(f"📥 OUTPUT GROQ ({len(out_raw)} chars):")
+        
+        # ===== GEMINI =====
+        else:
+            modelo = model_pro if config.get("modelo") == "pro" else model_flash
+            resp = modelo.generate_content(prompt)
+            out_raw = resp.text
+            print(f"📥 OUTPUT GEMINI ({len(out_raw)} chars):")
+        
         print(out_raw[:1000])
         print("..." if len(out_raw) > 1000 else "")
         
-        # Tenta parsear JSON se tipo_saida for json
+        tempo_exec = time.time() - inicio
+        
+        # Parse JSON se necessário
         content = out_raw
         if config.get('tipo_saida') == 'json':
             try:
@@ -107,17 +160,19 @@ def executar_no(timeline, config):
         return {
             "role": "assistant",
             "agent": config['nome'],
+            "provider": provider,
             "content": content,
             "raw": out_raw,
             "tempo": tempo_exec
         }, True
         
     except Exception as e:
-        print(f"💥 ERRO GEMINI: {str(e)}")
+        print(f"💥 ERRO {provider.upper()}: {str(e)}")
         return {
             "role": "system",
             "error": str(e),
-            "agent": config['nome']
+            "agent": config['nome'],
+            "provider": provider
         }, False
 
 # ==================== 4. ORQUESTRADOR ====================
@@ -126,35 +181,28 @@ def orquestrador(texto, anexos_list, history, json_config, contexto_objetivo):
     print("🎬 INICIANDO ORQUESTRADOR - NOVA EXECUÇÃO")
     print(f"📝 Input texto: '{texto[:200]}...' ({len(texto)} chars)")
     print(f"📎 Anexos: {len(anexos_list)} arquivos")
-    print(f"🎯 Objetivo: '{contexto_objetivo[:100]}...'")
     
-    # 1. Input Check
     if not texto.strip():
         print("⚠️ Texto vazio, abortando")
         yield history, []
         return
     
-    # 2. Setup - FORMATO MESSAGES
     history = history + [{"role": "user", "content": texto}]
-    print(f"📊 History atualizado com user msg ({len(history)} mensagens)")
     
     try:
         protocolo = json.loads(json_config)
-        print(f"🔗 Protocolo carregado: {len(protocolo)} agentes - { [a['nome'] for a in protocolo] }")
+        print(f"🔗 Protocolo: {len(protocolo)} agentes - {[a['nome'] for a in protocolo]}")
     except Exception as e:
         print(f"💥 Erro JSON config: {e}")
         history.append({"role": "assistant", "content": f"❌ Erro no JSON de Configuração: {str(e)}"})
         yield history, []
         return
     
-    # Adiciona assistente vazio inicial
     history.append({"role": "assistant", "content": ""})
     
-    # 3. Monta contexto inicial
     contexto_inicial = ""
     if contexto_objetivo and contexto_objetivo.strip():
         contexto_inicial += f"[OBJETIVO DO MODELO]\n{contexto_objetivo.strip()}\n[FIM OBJETIVO]\n\n"
-        print("🎯 Contexto objetivo adicionado")
     
     if anexos_list:
         for anexo in anexos_list:
@@ -164,16 +212,12 @@ def orquestrador(texto, anexos_list, history, json_config, contexto_objetivo):
     
     full_input = f"{contexto_inicial}{texto}".strip()
     timeline = [{"role": "user", "content": full_input}]
-    print(f"🌐 Full input timeline inicial: {len(full_input)} chars")
-    
     audit_data = []
     
-    # 4. Loop pelos agentes
     for idx, cfg in enumerate(protocolo):
         print(f"\n{'='*50}")
-        print(f"🚀 FASE {idx+1}/{len(protocolo)}: {cfg['nome']} ({cfg.get('tipo_saida', 'texto')})")
+        print(f"🚀 FASE {idx+1}/{len(protocolo)}: {cfg['nome']} ({cfg.get('provider', 'gemini')})")
         
-        # Mensagem aguardo
         history[-1]["content"] = f"⏳ Chamando agente **{cfg['nome']}**... Aguarde."
         yield history, audit_data
         
@@ -183,16 +227,14 @@ def orquestrador(texto, anexos_list, history, json_config, contexto_objetivo):
         history[-1]["content"] = f"⏳ Agente **{cfg['nome']}** processando..."
         yield history, audit_data
         
-        # Executa agente
         res, sucesso = executar_no(timeline, cfg)
         timeline.append(res)
         
-        # Audit
         audit_entry = {
             "step": idx + 1,
             "agent": cfg['nome'],
-            "mission": cfg['missao'][:100] + "...",
-            "model": cfg.get('modelo', 'flash'),
+            "provider": cfg.get('provider', 'gemini'),
+            "model": cfg.get('modelo', 'auto'),
             "type": cfg.get('tipo_saida', 'texto'),
             "response": str(res.get('content'))[:100] + "...",
             "raw_len": len(res.get('raw', '')),
@@ -201,7 +243,6 @@ def orquestrador(texto, anexos_list, history, json_config, contexto_objetivo):
             "timestamp": datetime.now().strftime('%H:%M:%S')
         }
         audit_data.append(audit_entry)
-        print(f"📋 Audit entry adicionado: {audit_entry['step']}")
         
         conteudo_resposta = res.get('content', '')
         if verificar_stop(conteudo_resposta):
@@ -212,8 +253,6 @@ def orquestrador(texto, anexos_list, history, json_config, contexto_objetivo):
         
         if idx == len(protocolo) - 1 or cfg.get('tipo_saida') == 'texto':
             texto_final = str(conteudo_resposta)
-            print(f"✨ Resposta final agente {idx+1}: {len(texto_final)} chars")
-            # Typewriter effect
             for i in range(0, len(texto_final), 5):
                 history[-1]["content"] = texto_final[:i+5]
                 yield history, audit_data
@@ -222,9 +261,8 @@ def orquestrador(texto, anexos_list, history, json_config, contexto_objetivo):
             yield history, audit_data
     
     print("🏁 Orquestrador concluído")
-    print("="*80)
 
-# Resto do código UI inalterado...
+# ==================== 5. UI ====================
 def ui_clean():
     config_init = carregar_protocolo()
     help_init = carregar_help()
@@ -234,16 +272,16 @@ def ui_clean():
         
         with gr.Tabs():
             with gr.Tab("💬 Chat"):
-                gr.Markdown("## Investigador AI (v33 DEBUG)")
-                chatbot = gr.Chatbot(label="Histórico", show_label=True, height=500)
+                gr.Markdown("## Investigador AI (v34 Gemini+Groq)")
+                chatbot = gr.Chatbot(label="Histórico", height=500)
                 
                 with gr.Row():
-                    txt_in = gr.Textbox(show_label=False, placeholder="Digite sua mensagem...", lines=2, scale=9)
+                    txt_in = gr.Textbox(show_label=False, placeholder="Digite...", lines=2, scale=9)
                     btn_send = gr.Button("📤 Enviar", variant="primary", scale=1)
             
             with gr.Tab("📎 Anexos & Contexto"):
                 objetivo_text = gr.Textbox(label="Objetivo do Modelo", lines=5)
-                anexos_upload = gr.File(file_count="multiple", file_types=[".txt", ".md", ".csv", ".json", ".pdf"])
+                anexos_upload = gr.File(file_count="multiple", file_types=[".txt", ".md", ".csv", ".json"])
                 anexos_display = gr.Textbox(label="Arquivos", interactive=False, lines=3)
                 
                 def atualizar_anexos(files):
@@ -254,27 +292,49 @@ def ui_clean():
                 anexos_upload.change(atualizar_anexos, anexos_upload, [anexos_state, anexos_display])
             
             with gr.Tab("⚙️ Protocolo"):
+                gr.Markdown("""
+                ## Edição do Protocolo
+                
+                **Novo:** Adicione `"provider": "groq"` ou `"provider": "gemini"` em cada agente.
+                
+                Modelos Groq suportados:
+                - `meta-llama/llama-4-maverick-17b-128e-instruct`
+                - `meta-llama/llama-3.3-70b-versatile`
+                - `deepseek-r1-distill-llama-70b`
+                
+                Exemplo:
+                ```
+                {
+                  "nome": "Agente Groq",
+                  "provider": "groq",
+                  "modelo": "meta-llama/llama-4-maverick-17b-128e-instruct",
+                  "tipo_saida": "texto",
+                  "missao": "..."
+                }
+                ```
+                """)
+                
                 with gr.Row():
-                    btn_save_proto = gr.Button("💾 Salvar")
+                    btn_save_proto = gr.Button("💾 Salvar", variant="primary")
                     btn_reload_proto = gr.Button("🔄 Recarregar")
                 proto_status = gr.Markdown("")
                 code_json = gr.Code(value=config_init, language="json", lines=30)
                 
                 btn_save_proto.click(salvar_protocolo, code_json, proto_status)
-                btn_reload_proto.click(lambda: carregar_protocolo(), code_json)
+                btn_reload_proto.click(lambda: carregar_protocolo(), outputs=code_json)
             
             with gr.Tab("🔍 Auditoria"):
                 audit_display = gr.JSON(label="Dados de Auditoria")
             
             with gr.Tab("❓ Ajuda"):
                 help_content = gr.Markdown(help_init)
-                gr.Button("🔄 Recarregar").click(lambda: carregar_help(), help_content)
+                gr.Button("🔄 Recarregar").click(lambda: carregar_help(), outputs=help_content)
         
-        btn_send.click(orquestrador, [txt_in, anexos_state, chatbot, code_json, objetivo_text], [chatbot, audit_display]).then(lambda: "", txt_in)
-        txt_in.submit(orquestrador, [txt_in, anexos_state, chatbot, code_json, objetivo_text], [chatbot, audit_display]).then(lambda: "", txt_in)
+        btn_send.click(orquestrador, [txt_in, anexos_state, chatbot, code_json, objetivo_text], [chatbot, audit_display]).then(lambda: "", outputs=txt_in)
+        txt_in.submit(orquestrador, [txt_in, anexos_state, chatbot, code_json, objetivo_text], [chatbot, audit_display]).then(lambda: "", outputs=txt_in)
     
     return app
 
 if __name__ == "__main__":
     print("🎉 Lançando app...")
-    ui_clean().launch(css="footer{display:none!important;}.contain{border:none!important;}")
+    ui_clean().launch(css="footer{display:none!important;}")