Update app.py

app.py

@@ -1,5 +1,5 @@
 # ╔════════════════════════════════════════════════════════════════════════════╗
-# ║  PIPELINE v42: FRAG + VISÃO PAGINADA + CONFEXT_UPLOAD + LOGS DETALHADOS   ║
+# ║  PIPELINE v43: FRAG + VISÃO PAGINADA + CONFEXT_UPLOAD + PARSE ROBUSTO     ║
 # ╚════════════════════════════════════════════════════════════════════════════╝
 
 import os
@@ -33,17 +33,8 @@ def carregar_protocolo():
         with open(ARQUIVO_CONFIG, "r", encoding="utf-8") as f:
             return f.read()
     except:
-        return json.dumps([
-            {
-                "nome": "PAGINADOR_VISUAL",
-                "missao": (
-                    "Você recebe o texto bruto de um conjunto de páginas de um PDF. "
-                    "Separe o conteúdo por PÁGINA, na ordem original. "
-                    "Para cada página, produza um objeto com: 'pagina', "
-                    "'transcricao_fiel' (texto integral, sem resumo) e "
-                    "'descricao_visual' (imagens, tabelas, diagramas, layout, sem julgamentos). "
-                    "Se o fragmento tiver 5 páginas, devolva uma lista JSON com EXATAMENTE 5 objetos, "
-                    "um por página."
+        # fallback com instrução já reforçada para JSON limpo
+        return json.dumps(json```
                 ),
                 "tipo_saida": "json",
                 "modelo": "flash"
@@ -119,6 +110,23 @@ def ler_anexo_e_fragmentar(arquivo, paginas_por_fragmento=5, logs=""):
 
 # ==================== 3. ENGINE DE EXECUÇÃO ====================
 
+def _extrair_json_possivel(out_raw: str) -> str:
+    """
+    Tenta isolar só o bloco JSON de uma resposta que pode ter texto extra.
+    Procura o primeiro 'json.
+    """
+    cleaned = out_raw.strip()
+    idx_abre_col = cleaned.find("
+
+    # menor índice válido
+    candidatos = [i for i in [idx_abre_col, idx_abre_obj] if i != -1]
+    if candidatos:
+        start = min(candidatos)
+        cleaned = cleaned[start:]
+
+    cleaned = cleaned.replace("```json", "").replace("```
+    return cleaned
+
 def executar_no(timeline, config, fragmento_input=None, logs=""):
     logs = log_point(f"executar_no({config['nome']}) chamado", logs)
     modo = "input_fragmento" if fragmento_input is not None else "timeline"
@@ -149,7 +157,8 @@ def executar_no(timeline, config, fragmento_input=None, logs=""):
         logs = log_point(f"Saída bruta (120 chars): {out[:120]!r}", logs)
 
         if config["tipo_saida"] == "json":
-            cleaned = out.strip().replace("``````", "")
+            cleaned = _extrair_json_possivel(out)
+            logs = log_point(f"Trecho candidato a JSON (120): {cleaned[:120]!r}", logs)
             try:
                 content = json.loads(cleaned)
             except Exception as e:
@@ -187,7 +196,7 @@ def orquestrador(texto, arquivo, history, json_config, confext_state):
         protocolo = json.loads(json_config)
         logs = log_point("Protocolo JSON carregado", logs)
     except Exception as e:
-        history[-1][1] = "❌ Erro no JSON de Configuração."
+        history[-1] = "❌ Erro no JSON de Configuração."[3]
         logs = log_point(f"ERRO carregando protocolo: {e}", logs)
         yield history, {}, logs, confext_state
         return
@@ -205,17 +214,17 @@ def orquestrador(texto, arquivo, history, json_config, confext_state):
     )
 
     if fragmentos:
-        history[-1][1] = "⏳ Fragmentando + visão paginada..."
+        history[-1] = "⏳ Fragmentando + visão paginada..."[3]
         logs = log_point("Fragmentos disponíveis; iniciando visão paginada", logs)
         yield history, timeline, logs, confext_upload
 
     # PASSO PAGINADOR_VISUAL (primeiro agente, se existir)
     if protocolo and fragmentos:
-        cfg_visao = protocolo[0]
+        cfg_visao = protocolo
         logs = log_point(f"Agente de visão selecionado: {cfg_visao['nome']}", logs)
 
         for i, fragmento in enumerate(fragmentos):
-            history[-1][1] = f"👁️ {cfg_visao['nome']} frag {i+1}/{len(fragmentos)}..."
+            history[-1] = f"👁️ {cfg_visao['nome']} frag {i+1}/{len(fragmentos)}..."[3]
             logs = log_point(f"Enviando frag {i+1}", logs)
             yield history, timeline, logs, confext_upload
 
@@ -258,7 +267,7 @@ def orquestrador(texto, arquivo, history, json_config, confext_state):
     final_response = ""
 
     for cfg in restante:
-        history[-1][1] = f"⚙️ {cfg['nome']}..."
+        history[-1] = f"⚙️ {cfg['nome']}..."[3]
         logs = log_point(f"Iniciando passo adicional: {cfg['nome']}", logs)
         yield history, timeline, logs, confext_upload
 
@@ -267,14 +276,14 @@ def orquestrador(texto, arquivo, history, json_config, confext_state):
 
         if cfg["tipo_saida"] == "texto":
             final_response = res["content"]
-            history[-1][1] = final_response
+            history[-1] = final_response[3]
             logs = log_point(f"Passo {cfg['nome']} produziu texto final", logs)
 
         yield history, timeline, logs, confext_upload
 
     if not restante and not texto:
-        history[-1][1] = "✅ PDF processado. Pronto para perguntas usando confext_upload."
-        final_response = history[-1][1]
+        history[-1] = "✅ PDF processado. Pronto para perguntas usando confext_upload."[3]
+        final_response = history[-1][3]
         logs = log_point("Nenhum passo adicional; apenas pré-processamento", logs)
 
     logs = log_point("FIM orquestrador()", logs)
@@ -294,7 +303,6 @@ def ui_clean():
         confext_state = gr.State(value=None)
 
         with gr.Tabs():
-            # --- ABA 1 ---
             with gr.Tab("💬 Investigador"):
                 chatbot = gr.Chatbot(
                     label="",
@@ -335,14 +343,12 @@ def ui_clean():
                     outputs=file_status,
                 )
 
-            # --- ABA 2 ---
             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)
                 confext_view = gr.JSON(label="confext_upload")
 
-            # --- ABA 3 ---
             with gr.Tab("⚙️ Config"):
                 with gr.Row():
                     btn_save = gr.Button("Salvar Config")