caarleexx commited on
Commit
cd5c320
·
verified ·
1 Parent(s): 9cb7ac3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +174 -412
app.py CHANGED
@@ -1,444 +1,206 @@
 
 
 
 
1
 
2
- # -*- coding: utf-8 -*-
3
- """
4
- Pipeline v10 - CÓDIGO FINAL E COESO
5
-
6
- Características:
7
- - Memória API: Passagem de histórico em genai.types.Content para contexto.
8
- - Robustez: Chamada de API V1 com correções para Tools/ThinkingConfig.
9
- - Prompts Dinâmicos: Carregamento, salvamento e edição dinâmica.
10
- - Auditoria: Histórico de Governança compactado no DNA (últimos 10 turnos).
11
- """
12
-
13
- # ============================================================================
14
- # 1. IMPORTAÇÕES E CONFIGURAÇÃO INICIAL
15
- # ============================================================================
16
- import json
17
  import os
 
18
  import re
19
- import warnings
20
- from datetime import datetime
21
- from typing import Dict, List, Tuple, Any
22
-
23
  import gradio as gr
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
 
25
- from google import genai
26
- from google.genai import types
27
-
28
- # Define o nome do arquivo de prompts para uso centralizado
29
- PROMPT_FILENAME = "prompts_pipeline.json"
30
-
31
- warnings.filterwarnings("ignore", category=FutureWarning, module="google.api_core")
32
-
33
- # --- Classes de Utilidade: LOGGER (Movido para garantir a definição inicial) ---
34
- class Logger:
35
- def __init__(self, verbose: bool = True): self.verbose = verbose
36
- def log(self, msg: str, level: str = "INFO"):
37
- log_msg = f"[{datetime.now().strftime('%H:%M:%S')}] [{level.upper()}] {msg}"; print(log_msg)
38
- if level.upper() in ["TASK", "START", "SUCCESS", "ERROR", "WARN"]: print("=" * 70)
39
-
40
- # INSTÂNCIA DO LOGGER
41
- logger = Logger(verbose=True)
42
-
43
- # --- Configuração da API ---
44
- API_KEY = os.getenv("GOOGLE_API_KEY")
45
- if not API_KEY:
46
- raise ValueError("A variável de ambiente GOOGLE_API_KEY não foi configurada.")
47
-
48
- CLIENT = genai.Client(api_key=API_KEY)
49
-
50
- # --- Definição dos Modelos e Título ---
51
- COUNSELOR_MODEL_NAME = "gemini-2.5-pro"
52
- SUPERVISOR_MODEL_NAME = "gemini-2.5-pro"
53
- TITLE = "# 🚀 Pipeline v10 | Governanca e Memória API"
54
-
55
- # ============================================================================
56
- # 2. PROMPTS CENTRALIZADOS (FUNÇÕES DE CARREGAMENTO/SALVAMENTO)
57
- # ============================================================================
58
- PROMPTS = {} # Variável global para os prompts em memória
59
-
60
- # Placeholder de fallback completo (para uso direto)
61
- FALLBACK_PROMPTS = {
62
- "P1_TRIAGEM": "METACOGNIÇÃO - TRIAGEM INICIAL.\nPERGUNTA: {pergunta}\n---\nAnalise a PERGUNTA do usuário e o histórico de conversa. Sua função é CLASSIFICAR e TOMAR UMA DECISÃO.\n\n### CRITÉRIOS DE CLASSIFICAÇÃO\n1. **tipo**: 'factual', 'objetiva_tecnica', ou 'subjetiva_complexa'.\n2. **confianca**: 'extrema', 'alta', 'media', 'baixa', ou 'insuficiente'.\n### DECISÃO\n- 'responder_direto': Use para 'factual' com confiança 'extrema' ou 'alta'.\n- 'analisar_profundamente': Use para todas as outras combinações.\nRETORNE JSON: {{\"tipo\": \"...\", \"confianca\": \"...\", \"decisao\": \"responder_direto|analisar_profundamente\"}}",
63
- "GERAR_RESPOSTA_DIRETA": "TAREFA: Resposta Direta (Bypass).\nPERGUNTA: \"{pergunta}\"\n---\nConsiderando o histórico de conversa (memória), forneça uma resposta completa, precisa e concisa para a última PERGUNTA do usuário.\nRETORNE JSON: {{\"resposta_direta\": \"Sua resposta concisa e precisa aqui.\"}}",
64
- "JUSTIFICAR_BYPASS": "METACOGNIÇÃO - JUSTIFICATIVA DE BYPASS.\nANÁLISE (P1): {p1}\n---\nJustifique, para a governança interna, por que a pipeline de raciocínio profundo foi pulada com base na Triagem P1 (JSON fornecido).\nRETORNE JSON: {{\"justificativa_bypass\": {{\"motivo\": \"A pergunta foi classificada como ... e a confiança era ....\", \"acao_tomada\": \"...\", \"proximo_passo\": \"...\", \"p1_decisao\": {p1}}}}}",
65
- "P2_CENARIOS": "METACOGNIÇÃO - GERAÇÃO DE CENÁRIOS.\nANÁLISE (P1): {p1}, PERGUNTA: {pergunta}\n---\nCrie 2 cenários prováveis que VALIDAM e 2 cenários IMPROVÁVEIS que REFUTAM o princípio inerente da pergunta.\nRETORNE JSON: {{\"cenarios\": {{\"provaveis\": [\"Fato/Cenário que Valida 1\", \"Fato/Cenário que Valida 2\"], \"improvaveis\": [\"Fato/Cenário que Refuta 1\", \"Fato/Cenário que Refuta 2\"]}}, \"decisao\": \"prosseguir para as próximas validações.\"}}",
66
- "P4_CRUZAR_VALIDACOES": "METACOGNIÇÃO - ABSTRAÇÃO E FUNDAMENTO.\nANÁLISES ANTERIORES: P1(Triagem): {p1}, P2(Cenários): {p2}, P3(Isolamento Simulado): {p3}\n---\nCruze as análises. Identifique o Princípio Central/Fundamento que guiará a resposta.\nRETORNE JSON: {{\"principio_central\": \"O Princípio Central da discussão é ...\", \"analise_de_inconsistencia\": \"...\", \"status_analise\": \"FUNDAMENTO_OK\"}}",
67
- "P5_LACUNAS_FINAIS": "METACOGNIÇÃO - ANÁLISE DE INCERTEZA.\nPRINCÍPIO CENTRAL (P4): {p4}, PERGUNTA: {pergunta}\n---\nIdentifique a(s) lacuna(s) de informação.\n- Decisão: 'continuar' (pode resolver a incerteza) ou 'questionar' (precisa do dado do usuário).\nRETORNE JSON: {{\"pontos_de_incerteza\": [\"Detalhe 1\", \"Detalhe 2\"], \"decisao_interna\": \"questionar|continuar\", \"pergunta_chave_para_usuario\": \"Sua pergunta para o usuário (ou null se for 'continuar').\"}}",
68
- "P7_SINTETIZAR": "SINTETIZADOR.\nDADOS CRÍTICOS (P6: Ponderação Final): {p6}\n---\nConverta todo o raciocínio profundo em uma RESPOSTA FINAL coesa e completa.\nRETORNE JSON: {{\"resposta\": \"Sua resposta completa aqui. Deve ser direta e refletir a profundidade da análise executada.\"}}",
69
- "P8_VERIFICAR": "VERIFICADOR FINAL (SUPERVISOR DE QUALIDADE).\nRESPOSTA A VERIFICAR: {resposta_a_verificar}\n---\nRealize a verificação tripla (Factual, Lógica, Clareza). Se for necessário, corrija.\nRETORNE JSON: {{\"todas_aprovadas\": true|false, \"erros_encontrados\": \"...\", \"resposta_corrigida\": null}} (Se true) ou {{\"todas_aprovadas\": false, \"erros_encontrados\": \"...\", \"resposta_corrigida\": \"[Novo Texto Final e Corrigido].\"}} (Se false)"
70
- }
71
-
72
-
73
- def carregar_prompts_externos(filename: str = PROMPT_FILENAME) -> Dict:
74
- """Carrega prompts do arquivo ou retorna o fallback."""
75
- try:
76
- with open(filename, 'r', encoding='utf-8') as f:
77
- full_prompts = json.load(f)
78
- if not all(key in full_prompts for key in FALLBACK_PROMPTS.keys()):
79
- raise ValueError("O arquivo de prompts está incompleto ou malformatado.")
80
- logger.log(f"Prompts carregados com sucesso do arquivo: {filename}", "SUCCESS")
81
- return full_prompts
82
- except Exception as e:
83
- logger.log(f"Não foi possível carregar os prompts externos: {e}. Usando FALLBACK.", "ERROR")
84
- return FALLBACK_PROMPTS
85
 
86
- def get_prompts_raw_text(filename: str = PROMPT_FILENAME) -> str:
87
- """ o arquivo de prompts como texto bruto para o editor do Gradio, ou retorna fallback."""
88
  try:
89
- if not os.path.exists(filename):
90
- return json.dumps(FALLBACK_PROMPTS, indent=2, ensure_ascii=False)
91
- with open(filename, 'r', encoding='utf-8') as f:
92
- return f.read()
93
  except:
94
- return json.dumps({"erro": "Não foi possível ler o arquivo."}, indent=2, ensure_ascii=False)
95
 
96
- def salvar_e_recarregar_prompts(new_prompts_json_str: str) -> str:
97
- """Salva o novo JSON no arquivo e recarrega a variável PROMPTS global."""
98
- global PROMPTS
99
- try:
100
- novo_dicionario = json.loads(new_prompts_json_str)
101
- with open(PROMPT_FILENAME, 'w', encoding='utf-8') as f:
102
- json.dump(novo_dicionario, f, indent=2, ensure_ascii=False)
103
- PROMPTS = novo_dicionario
104
- logger.log(f"Novo arquivo de prompts salvo e recarregado com sucesso.", "SUCCESS")
105
- return f"✅ SUCESSO! Prompts salvos em {PROMPT_FILENAME} e variáveis em memória atualizadas. Última atualização: {datetime.now().strftime('%H:%M:%S')}."
106
- except json.JSONDecodeError:
107
- logger.log("ERRO: O JSON enviado é inválido. Não foi possível salvar ou recarregar.", "ERROR")
108
- return "❌ ERRO: O JSON enviado está malformado. Corrija o JSON antes de salvar."
109
- except Exception as e:
110
- logger.log(f"ERRO CRÍTICO no servidor: Falha ao escrever o arquivo. Detalhes: {e}", "ERROR")
111
- return f"❌ ERRO CRÍTICO no servidor: Falha ao escrever o arquivo. Detalhes: {e}"
112
-
113
- # INICIALIZAÇÃO OBRIGATÓRIA DA VARIÁVEL GLOBAL DE PROMPTS
114
- PROMPTS = carregar_prompts_externos()
115
-
116
-
117
- # ============================================================================
118
- # 3. CLASSES E FUNÇÕES HELPERS ADICIONAIS
119
- # ============================================================================
120
- def sanitizar_texto(texto: str) -> str:
121
- if not isinstance(texto, str): return ""
122
- texto_limpo = re.sub(r'[\x00-\x08\x0b\x0c\x0e-\x1f]', '', texto)
123
- texto_limpo = re.sub(r'\s{2,}', ' ', texto_limpo).replace('\\n', '\n')
124
- return texto_limpo.strip()
125
-
126
- def to_gemini_contents(historico: List[Dict]) -> List[types.Content]:
127
- """Converte o histórico Gradio-compatível para o formato genai.types.Content (user/model)."""
128
- gemini_contents = []
129
- for entry in historico:
130
- role = "model" if entry["role"] == "assistant" else "user"
131
- content_for_api = re.sub(r'\*\*JUSTIFICATIVA DE RESPOSTA DIRETA:\*\*.*?\-\-\-\n\n', '', entry["content"], flags=re.DOTALL)
132
- if content_for_api:
133
- gemini_contents.append(
134
- types.Content(
135
- role=role,
136
- parts=[types.Part.from_text(text=content_for_api)]
137
- )
138
- )
139
- return gemini_contents
140
-
141
- def chamar_gemini_json(model_name: str, prompt: str,
142
- historico_conversa: List[types.Content] = None,
143
- temperatura: float = 0.4, max_tokens: int = 8192) -> Dict:
144
- prompt_completo = f"{prompt}\n\n---\n\n**INSTRUÇÃO CRÍTICA: Não use formatação Markdown. Sua resposta DEVE ser estritamente um único objeto JSON válido.**"
145
  try:
146
- contents = historico_conversa if historico_conversa else []
147
- contents.append(types.Content(role="user", parts=[types.Part.from_text(text=prompt_completo)]))
148
- tools = [types.Tool(google_search=types.GoogleSearch())]
149
- generate_content_config = types.GenerateContentConfig(
150
- temperature=temperatura,
151
- max_output_tokens=max_tokens,
152
- thinking_config=types.ThinkingConfig(thinking_budget=8192),
153
- tools=tools
154
- )
155
- stream = CLIENT.models.generate_content_stream(model=model_name, contents=contents, config=generate_content_config)
156
 
157
- resposta_bruta = "".join(chunk.text for chunk in stream if chunk.text)
158
- resposta_sanitizada = sanitizar_texto(resposta_bruta)
159
-
160
- if not resposta_sanitizada:
161
- return {"erro": "API_EMPTY_RESPONSE"}
162
-
163
- match = re.search(r'\{.*\}', resposta_sanitizada, re.DOTALL)
164
- if match:
165
- return json.loads(match.group(0))
166
  else:
167
- return json.loads(resposta_sanitizada)
168
-
 
 
 
 
 
 
169
  except Exception as e:
170
- logger.log(f"Falha na chamada da API ou no parse do JSON: {e}", "ERROR")
171
- return {"erro": "API_CALL_FAILED", "detalhes": str(e)}
172
 
173
- def criar_dna() -> Dict:
174
- return {
175
- "historico_governanca": [],
176
- "meta": {"total_turnos": 0},
177
- "pipeline_state": { "status": "completed", "paused_at_step": None, "saved_data": {} },
178
- "P1_data": {}, "P2_data": {}, "P4_data": {}, "P5_data": {}, "P8_data": {},
179
- }
180
 
181
- def formatar_dna_json(data: Dict) -> str:
182
- """Retorna um JSON formatado com dois espaços de indentação."""
183
- try:
184
- return json.dumps(data, indent=2, ensure_ascii=False)
185
- except:
186
- return f"Erro ao serializar dados: {data}"
187
-
188
-
189
- # ============================================================================
190
- # 4. PASSOS DA PIPELINE
191
- # ============================================================================
192
- def passo_1_triagem(pergunta: str, dna: Dict, historico_memoria: List[types.Content]) -> Tuple[Dict, Dict]:
193
- logger.log("📊 PASSO 1: TRIAGEM INICIAL", "TASK")
194
- p1 = chamar_gemini_json(COUNSELOR_MODEL_NAME, PROMPTS["P1_TRIAGEM"].format(pergunta=pergunta), historico_conversa=historico_memoria)
195
- dna["P1_data"] = p1
196
- return p1, dna
197
- def passo_gerar_resposta_direta(pergunta: str, historico_memoria: List[types.Content]) -> Dict:
198
- logger.log("⚡ FAST PATH: GERANDO RESPOSTA DIRETA", "TASK")
199
- return chamar_gemini_json(COUNSELOR_MODEL_NAME, PROMPTS["GERAR_RESPOSTA_DIRETA"].format(pergunta=pergunta), historico_conversa=historico_memoria)
200
- def passo_justificar_bypass(p1: Dict) -> Dict:
201
- logger.log("⚡ FAST PATH: GERANDO JUSTIFICATIVA", "TASK")
202
- return chamar_gemini_json(COUNSELOR_MODEL_NAME, PROMPTS["JUSTIFICAR_BYPASS"].format(p1=json.dumps(p1)))
203
- def passo_2_cenarios(pergunta: str, p1: Dict, dna: Dict) -> Tuple[Dict, Dict]:
204
- logger.log("🧠 PASSO 2: GERAÇÃO DE CENÁRIOS", "TASK")
205
- p2 = chamar_gemini_json(COUNSELOR_MODEL_NAME, PROMPTS["P2_CENARIOS"].format(p1=json.dumps(p1), pergunta=pergunta))
206
- dna["P2_data"] = p2
207
- return p2, dna
208
- def passo_4_cruzar_validacoes(p1: Dict, p2: Dict, p3: Dict, dna: Dict) -> Tuple[Dict, Dict]:
209
- logger.log("🧠 PASSO 4: IDENTIFICAÇÃO DO PRINCÍPIO CENTRAL", "TASK")
210
- p4 = chamar_gemini_json(COUNSELOR_MODEL_NAME, PROMPTS["P4_CRUZAR_VALIDACOES"].format(p1=json.dumps(p1), p2=json.dumps(p2), p3=json.dumps(p3)))
211
- dna["P4_data"] = p4
212
- return p4, dna
213
- def passo_5_lacunas_finais(pergunta: str, p4: Dict, dna: Dict) -> Tuple[Dict, Dict]:
214
- logger.log("🧠 PASSO 5: ANÁLISE DE INCERTEZA", "TASK")
215
- p5 = chamar_gemini_json(COUNSELOR_MODEL_NAME, PROMPTS["P5_LACUNAS_FINAIS"].format(p4=json.dumps(p4), pergunta=pergunta))
216
- dna["P5_data"] = p5
217
- return p5, dna
218
- def passo_8_verificar(resposta_a_verificar: str, dna: Dict) -> Tuple[Dict, Dict]:
219
- logger.log("✅ PASSO 8: VERIFICAÇÃO FINAL", "TASK")
220
- p8 = chamar_gemini_json(SUPERVISOR_MODEL_NAME, PROMPTS["P8_VERIFICAR"].format(resposta_a_verificar=resposta_a_verificar))
221
- dna["P8_data"] = p8
222
- return p8, dna
223
- # Simulado
224
- def passo_3_isolar_cenarios(p2: Dict) -> Dict: return {"simulado": True}
225
- def passo_6_ponderar(p2: Dict, p4: Dict, p5: Dict) -> Dict: return {"simulado": True}
226
- def passo_7_sintetizar(p6: Dict) -> Dict: return {"resposta": "Resposta sintetizada vinda da pipeline completa (simulado)."}
227
-
228
-
229
- # ============================================================================
230
- # 5. ORQUESTRADOR PRINCIPAL
231
- # ============================================================================
232
- def salvar_governanca(dna: Dict, pergunta: str, tipo: str, **data) -> None:
233
- """Salva o registro da iteração de governança."""
234
- entry = {"turno": dna['meta']['total_turnos'], "timestamp": datetime.now().strftime('%H:%M:%S'), "tipo": tipo, "pergunta": pergunta, **data}
235
- dna["historico_governanca"].append(entry)
236
- if len(dna["historico_governanca"]) > 10:
237
- dna["historico_governanca"] = dna["historico_governanca"][-10:]
238
-
239
- def iniciar_nova_pipeline(pergunta_original: str, historico: List[Dict], anexo: Any, dna: Dict, historico_memoria: List[types.Content]) -> Tuple[str, List, Dict]:
240
- pergunta = sanitizar_texto(pergunta_original)
241
 
242
- p1, dna = passo_1_triagem(pergunta, dna, historico_memoria)
243
- if "erro" in p1: return f"Erro na Triagem: {p1.get('detalhes', 'Não foi possível classificar a pergunta.')}", historico, dna
244
 
245
- # FAST PATH
246
- if p1.get("decisao") == "responder_direto":
247
- resposta_direta_data = passo_gerar_resposta_direta(pergunta, historico_memoria)
248
- justificativa_data = passo_justificar_bypass(p1)
249
- if "erro" in resposta_direta_data:
250
- return f"Erro ao gerar a resposta direta: {resposta_direta_data['detalhes']}", historico, dna
251
 
252
- resposta_direta = resposta_direta_data.get("resposta_direta", "Não foi possível gerar a resposta.")
253
- justificativa = justificativa_data.get("justificativa_bypass", {})
254
- verificacao, dna = passo_8_verificar(resposta_direta, dna)
255
- resposta_final = verificacao.get("resposta_corrigida") or resposta_direta
256
 
257
- justificativa_texto = f"**JUSTIFICATIVA DE RESPOSTA DIRETA:**\n- **Tipo/Confiança (P1):** {p1.get('tipo', 'N/A')}/{p1.get('confianca', 'N/A')}\n- **Motivo:** {justificativa.get('motivo', 'N/A')}\n\n---\n"
258
- resposta_formatada = justificativa_texto + resposta_final
259
-
260
- novo_historico = historico + [{"role": "user", "content": pergunta_original}, {"role": "assistant", "content": resposta_formatada}]
261
- salvar_governanca(dna, pergunta_original, "FAST_PATH", p1=p1, p8=verificacao)
262
- return "PIPELINE_COMPLETED", novo_historico, dna
263
-
264
- # FULL PATH
265
- else:
266
- p2, dna = passo_2_cenarios(pergunta, p1, dna)
267
- p3 = passo_3_isolar_cenarios(p2)
268
- p4, dna = passo_4_cruzar_validacoes(p1, p2, p3, dna)
269
- p5, dna = passo_5_lacunas_finais(pergunta, p4, dna)
270
 
271
- if p5.get("decisao_interna") == "questionar":
272
- salvar_governanca(dna, pergunta_original, "FULL_PATH_PAUSA", p1=p1, p5=p5)
273
- dna['pipeline_state'] = {"status": "paused", "paused_at_step": "P5", "saved_data": {'p1':p1, 'p2':p2, 'p3':p3, 'p4':p4, 'pergunta_original': pergunta, 'historico_original': historico}}
274
- pergunta_do_bot = p5.get('pergunta_chave_para_usuario', 'Preciso de mais informações.')
275
- historico_atualizado = historico + [{"role": "user", "content": pergunta_original}, {"role": "assistant", "content": pergunta_do_bot}]
276
- return "PIPELINE_PAUSED", historico_atualizado, dna
277
 
278
- p6 = passo_6_ponderar(p2, p4, p5)
279
- p7 = passo_7_sintetizar(p6)
280
- p8, dna = passo_8_verificar(p7.get("resposta", ""), dna)
281
 
282
- resposta_final = p8.get("resposta_corrigida") or p7.get("resposta", "Não foi possível gerar a resposta.")
283
- novo_historico = historico + [{"role": "user", "content": pergunta_original}, {"role": "assistant", "content": resposta_final}]
284
- dna['pipeline_state']['status'] = 'completed'
 
 
 
 
285
 
286
- salvar_governanca(dna, pergunta_original, "FULL_PATH_CONCLUIDO", p1=p1, p2=p2, p4=p4, p5=p5, p8=p8)
287
- return "PIPELINE_COMPLETED", novo_historico, dna
288
-
289
- def resumir_pipeline(esclarecimento_usuario_original: str, dna: Dict, historico_memoria: List[types.Content]) -> Tuple[str, List, Dict]:
290
- # Lógica de Retomada Simulada
291
- dados_salvos = dna['pipeline_state']['saved_data']
292
- resposta = "RESUMIDO! A pipeline (pausada) foi retomada com o seu esclarecimento, mas a lógica de retomada completa precisa ser implementada."
293
- dna['pipeline_state']['status'] = 'completed'
294
- historico_atualizado = dados_salvos['historico_original'] + [{"role": "user", "content": esclarecimento_usuario_original}, {"role": "assistant", "content": resposta}]
295
- salvar_governanca(dna, esclarecimento_usuario_original, "RETOMADA_SIMULADA", p_resumido=True, status="P_COMPLETED")
296
- return "PIPELINE_COMPLETED", historico_atualizado, dna
297
-
298
- def executar_pipeline(pergunta: str, historico: List[Dict], anexo: Any, dna: Dict, historico_memoria: List[types.Content]) -> Tuple[str, List, Dict]:
299
- try:
300
- if 'pipeline_state' not in dna: dna.update(criar_dna())
301
- dna['meta']['total_turnos'] = dna['meta'].get('total_turnos', 0) + 1
302
 
303
- if dna['pipeline_state']['status'] == 'paused':
304
- return resumir_pipeline(pergunta, dna, historico_memoria)
305
-
306
- for key in ["P1_data", "P2_data", "P4_data", "P5_data", "P8_data"]:
307
- dna[key] = {}
308
 
309
- return iniciar_nova_pipeline(pergunta, historico, anexo, dna, historico_memoria)
310
-
311
- except Exception as e:
312
- logger.log(f"Erro catastrófico no orquestrador: {e}", "ERROR")
313
- resposta_erro = f"Ocorreu um erro inesperado na pipeline: {e}"
314
- dna['pipeline_state']['status'] = 'error'
315
- salvar_governanca(dna, pergunta, "ERRO_CATASTRÓFICO", erro=str(e))
316
- return "PIPELINE_ERROR", historico + [{"role": "user", "content": pergunta}, {"role": "assistant", "content": resposta_erro}], dna
317
-
318
-
319
- # ============================================================================
320
- # 6. INTERFACE COM GRADIO
321
- # ============================================================================
322
-
323
- def chat_interface(pergunta: str, historico_gradio: List[List[str]], anexo: Any, dna_json_str: str) -> Tuple[List, str, str, None, str, str, str, str, str, str, str]:
324
-
325
- try:
326
- dna = json.loads(dna_json_str) if dna_json_str and dna_json_str.strip() else criar_dna()
327
- except:
328
- dna = criar_dna()
329
 
330
- # Prepara o histórico
331
- historico_interno = []
332
- for turno in historico_gradio:
333
- if turno and turno[0]: historico_interno.append({"role": "user", "content": turno[0]})
334
- if turno and turno[1]: historico_interno.append({"role": "assistant", "content": turno[1]})
335
 
336
- historico_para_gemini = to_gemini_contents(historico_interno)
 
 
337
 
338
- # Execução
339
- _ , novo_historico_para_exibir, dna_atualizado = executar_pipeline(pergunta, historico_interno, anexo, dna, historico_para_gemini)
340
-
341
- # Conversão de Outputs
342
- novo_historico_gradio = []
343
- for i in range(0, len(novo_historico_para_exibir), 2):
344
- user_msg = novo_historico_para_exibir[i]['content']
345
- assistant_msg = novo_historico_para_exibir[i+1]['content'] if i + 1 < len(novo_historico_para_exibir) else ""
346
- novo_historico_gradio.append([user_msg, assistant_msg])
347
 
348
- dna_completo_str = formatar_dna_json(dna_atualizado)
349
- p1_data_str = formatar_dna_json(dna_atualizado.get('P1_data', {}))
350
- p2_data_str = formatar_dna_json(dna_atualizado.get('P2_data', {}))
351
- p4_data_str = formatar_dna_json(dna_atualizado.get('P4_data', {}))
352
- p5_data_str = formatar_dna_json(dna_atualizado.get('P5_data', {}))
353
- p8_data_str = formatar_dna_json(dna_atualizado.get('P8_data', {}))
354
- historico_gov_str = formatar_dna_json(dna_atualizado.get('historico_governanca', []))
355
 
356
- return (novo_historico_gradio,
357
- "",
358
- json.dumps(dna_atualizado, indent=2, ensure_ascii=False),
359
- None,
360
- dna_completo_str,
361
- p1_data_str,
362
- p2_data_str,
363
- p4_data_str,
364
- p5_data_str,
365
- p8_data_str,
366
- historico_gov_str
367
- )
368
-
369
 
370
  if __name__ == "__main__":
371
- prompts_text_inicial = get_prompts_raw_text()
372
-
373
- with gr.Blocks(title="Pipeline v10 - Raciocínio Adaptativo", theme=gr.themes.Soft()) as demo:
374
-
375
-
376
- #gr.Markdown(TITLE)
377
-
378
- with gr.Tab("💬 Chat e Resposta"):
379
- with gr.Row():
380
- with gr.Column(scale=3):
381
- chatbot = gr.Chatbot(label="Chat", height=500, bubble_full_width=False)
382
- input_textbox = gr.Textbox(label="Digite sua pergunta...", lines=3)
383
- with gr.Row():
384
- submit_button = gr.Button("🚀 Enviar", variant="primary", scale=1)
385
- file_upload = gr.File(label="Anexar Arquivo", scale=1)
386
-
387
- with gr.Column(scale=2):
388
- dna_json_hidden = gr.Textbox(value=json.dumps(criar_dna()), visible=False)
389
- dna_view = gr.Code(label="DNA (Estado Completo da Conversa/Meta-Dados)", language="json", interactive=False, value=json.dumps(criar_dna(), indent=2, ensure_ascii=False))
390
-
391
- with gr.Group():
392
- with gr.Accordion("📜 Ver Histórico de Governança", open=False):
393
- gov_out = gr.Code(label=f"Histórico de Governança (Últimos 10 Turnos)", language="json", interactive=False, value="[]")
394
- gov_out#.render()
395
- p1_out = gr.Code(label="1. TRIAGEM (P1)", language="json", interactive=False, value="{}")
396
- p2_out = gr.Code(label="2. CENÁRIOS (P2)", language="json", interactive=False, value="{}")
397
- p4_out = gr.Code(label="4. PRINCÍPIO (P4)", language="json", interactive=False, value="{}")
398
- p5_out = gr.Code(label="5. LACUNAS/DECISÃO (P5)", language="json", interactive=False, value="{}")
399
- p8_out = gr.Code(label="8. VERIFICAÇÃO (P8)", language="json", interactive=False, value="{}")
400
-
401
-
402
-
403
- with gr.Tab("🛠️ Debug da Pipeline"):
404
- gr.Markdown("## Visualização JSON por Passo do TURNO ATUAL")
405
- with gr.Row():
406
- p1_out#.render()
407
- p2_out#.render()
408
- with gr.Row():
409
- p4_out#.render()
410
- p5_out#.render()
411
- with gr.Row():
412
- p8_out#.render()
413
-
414
- with gr.Tab("📝 Editor de Prompts"):
415
- gr.Markdown(f"## Editor Dinâmico de Prompts ({PROMPT_FILENAME})")
416
- prompts_editor = gr.Code(label="Conteúdo do JSON de Prompts", language="json", value=prompts_text_inicial, interactive=True, lines=30)
417
- save_button = gr.Button("💾 SALVAR E RECARREGAR PROMPTS NO SERVIDOR", variant="stop")
418
- save_status = gr.Textbox(label="Status do Servidor", interactive=False)
419
-
420
- save_button.click(fn=salvar_e_recarregar_prompts, inputs=[prompts_editor], outputs=[save_status])
421
-
422
-
423
- # Mapeamento Completo de Inputs e Outputs (11 Saídas)
424
- fn_args = {
425
- "fn": chat_interface,
426
- "inputs": [input_textbox, chatbot, file_upload, dna_json_hidden],
427
- "outputs": [chatbot,
428
- input_textbox,
429
- dna_json_hidden,
430
- file_upload,
431
- dna_view,
432
- p1_out,
433
- p2_out,
434
- p4_out,
435
- p5_out,
436
- p8_out,
437
- gov_out
438
- ]
439
- }
440
-
441
- submit_button.click(**fn_args)
442
- input_textbox.submit(**fn_args)
443
-
444
- demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
 
1
+ # ╔════════════════════════════════════════════════════════════════════════════╗
2
+ # ║ PIPELINE v22: ORQUESTRAÇÃO DE AGENTES (JSON-LIST) ║
3
+ # ║ O Segredo: Uma lista de instruções que governa o pensamento. ║
4
+ # ╚════════════════════════════════════════════════════════════════════════════╝
5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  import os
7
+ import json
8
  import re
 
 
 
 
9
  import gradio as gr
10
+ import google.generativeai as genai
11
+
12
+ # ==================== 1. CONFIGURAÇÃO ====================
13
+ api_key = os.getenv("GOOGLE_API_KEY", "SUA_API_KEY_AQUI")
14
+ genai.configure(api_key=api_key)
15
+
16
+ # Modelo Flash para raciocínio rápido, Pro para redação final
17
+ model_fast = genai.GenerativeModel("gemini-flash-latest")
18
+ model_pro = genai.GenerativeModel("gemini-pro-latest")
19
+
20
+ # ==================== 2. O PROTOCOLO (A "ALMA" DA IA) ====================
21
+ # É aqui que você define a inteligência. O código abaixo apenas obedece esta lista.
22
+ # Adicione ou remova agentes aqui para mudar o comportamento do sistema.
23
+
24
+ PROTOCOLO_OPERACIONAL = [
25
+ {
26
+ "nome": "ESCRIVÃO_DIGITAL",
27
+ "tipo_saida": "json",
28
+ "modelo": model_fast,
29
+ "missao": """
30
+ Leia o input do Usuário. Sua tarefa é extrair FATOS frios.
31
+ Extraia:
32
+ 1. Quem são as partes (Vítima/Acusado).
33
+ 2. Cronologia (O que aconteceu e quando).
34
+ 3. Inventário de Provas (O que foi citado: áudio, vídeo, testemunha).
35
+ Retorne APENAS JSON.
36
+ """
37
+ },
38
+ {
39
+ "nome": "ANALISTA_COMPORTAMENTAL",
40
+ "tipo_saida": "json",
41
+ "modelo": model_fast,
42
+ "missao": """
43
+ Leia o input original e os fatos do Escrivão.
44
+ Sua tarefa é analisar o SUBTEXTO e a PSICOLOGIA.
45
+ 1. Analise o tom de voz descrito ou transcrito.
46
+ 2. Identifique manipulação (Gaslighting, Inversão de Culpa).
47
+ 3. Avalie a coerência emocional (Alguém calmo demais? Alguém desesperado?).
48
+ Retorne APENAS JSON.
49
+ """
50
+ },
51
+ {
52
+ "nome": "AUDITOR_INVESTIGATIVO",
53
+ "tipo_saida": "json",
54
+ "modelo": model_fast,
55
+ "missao": """
56
+ Sua tarefa é encontrar a MENTIRA.
57
+ Cruze o depoimento do acusado (Fatos) com a análise psicológica (Comportamental).
58
+ Liste as contradições. Onde a versão dele não bate com a realidade do áudio/texto?
59
+ Retorne APENAS JSON.
60
+ """
61
+ },
62
+ {
63
+ "nome": "PROMOTOR_SENIOR",
64
+ "tipo_saida": "texto", # Gera o relatório final visível
65
+ "modelo": model_pro, # Modelo mais potente para escrever bem
66
+ "missao": """
67
+ Escreva o PARECER TÉCNICO FINAL (Relatório Jurídico).
68
+ Use Markdown. Seja detalhado (+600 palavras).
69
+ Estrutura Obrigatória:
70
+ 1. Cabeçalho (Data/Assunto).
71
+ 2. Sumário dos Fatos (Baseado no Escrivão).
72
+ 3. Análise Forense (Baseado no Analista e Auditor). Explique a manipulação.
73
+ 4. Fundamentação Jurídica (Cite leis: Maria da Penha, Ameaça, Violência Psicológica).
74
+ 5. Conclusão e Pedidos (Prisão? Medida Protetiva?).
75
+ """
76
+ }
77
+ ]
78
 
79
+ # ==================== 3. O MOTOR (CÓDIGO AGNÓSTICO) ====================
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
80
 
81
+ def limpar_json(texto: str) -> dict:
82
+ """Garante que o JSON venha limpo, sem markdown."""
83
  try:
84
+ if "```" in texto:
85
+ texto = re.search(r"```json?(.*?)```", texto, re.DOTALL).group(1)
86
+ return json.loads(texto.strip())
 
87
  except:
88
+ return {"conteudo_bruto": texto}
89
 
90
+ def executar_agente(timeline: list, config: dict) -> dict:
91
+ """Um agente entra na sala, a timeline e sua opinião."""
92
+
93
+ # 1. O agente recebe todo o histórico anterior como contexto
94
+ contexto_str = json.dumps(timeline, ensure_ascii=False, indent=2)
95
+
96
+ prompt = f"""
97
+ --- CONTEXTO DO PROCESSO (TIMELINE) ---
98
+ {contexto_str}
99
+ ---------------------------------------
100
+
101
+ VOCÊ É O AGENTE: {config['nome']}
102
+ SUA MISSÃO: {config['missao']}
103
+
104
+ IMPORTANTE: Não converse. Apenas execute a missão.
105
+ """
106
+
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
107
  try:
108
+ # 2. Chamada à API
109
+ response = config['modelo'].generate_content(prompt)
110
+ conteudo_raw = response.text
 
 
 
 
 
 
 
111
 
112
+ # 3. Processamento da Saída
113
+ if config['tipo_saida'] == 'json':
114
+ conteudo_processado = limpar_json(conteudo_raw)
 
 
 
 
 
 
115
  else:
116
+ conteudo_processado = conteudo_raw # Texto/Markdown
117
+
118
+ return {
119
+ "role": "assistant",
120
+ "agent_name": config['nome'],
121
+ "content": conteudo_processado
122
+ }
123
+
124
  except Exception as e:
125
+ return {"role": "system", "error": str(e)}
 
126
 
127
+ def orquestrador(texto_usuario: str):
128
+ """Loop principal que roda a lista de agentes."""
129
+ if not texto_usuario.strip(): return "Insira um caso.", []
 
 
 
 
130
 
131
+ # 1. Inicia a Timeline com a entrada do usuário
132
+ timeline = [
133
+ {"role": "user", "content": texto_usuario}
134
+ ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
135
 
136
+ relatorio_final = ""
 
137
 
138
+ # 2. O Loop Mágico (Itera sobre o Protocolo)
139
+ for passo in PROTOCOLO_OPERACIONAL:
140
+ # Executa o agente atual
141
+ resultado = executar_agente(timeline, passo)
 
 
142
 
143
+ # Adiciona o resultado à timeline para os próximos agentes verem
144
+ timeline.append(resultado)
 
 
145
 
146
+ # Se for o Promotor (texto), salvamos para exibir na tela principal
147
+ if passo['tipo_saida'] == 'texto':
148
+ relatorio_final = resultado['content']
149
+
150
+ # Retorna o texto formatado e a timeline completa (JSON) para inspeção
151
+ return relatorio_final, timeline
 
 
 
 
 
 
 
152
 
153
+ # ==================== 4. INTERFACE GRÁFICA ====================
154
+
155
+ def interface():
156
+ with gr.Blocks(title="Orquestrador v22", theme=gr.themes.Soft()) as app:
 
 
157
 
158
+ gr.Markdown("# ⚖️ Pipeline Cognitiva: Orquestração de Agentes")
159
+ gr.Markdown("Este sistema não usa um prompt único. Ele usa uma cadeia de especialistas virtuais definidos em JSON.")
 
160
 
161
+ with gr.Row():
162
+ # Entrada
163
+ inp = gr.Textbox(
164
+ label="Dossiê do Caso (Texto/Transcrições)",
165
+ lines=8,
166
+ placeholder="Cole aqui o BO, depoimentos e transcrições..."
167
+ )
168
 
169
+ btn = gr.Button("🚀 INICIAR ORQUESTRAÇÃO", variant="primary", size="lg")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
170
 
171
+ with gr.Row():
172
+ # Saída Final (Markdown Renderizado)
173
+ out_relatorio = gr.Markdown(label="Parecer Final do Promotor")
 
 
174
 
175
+ # Saída de Debug (O "Cérebro" da máquina)
176
+ out_timeline = gr.JSON(label="Timeline de Consciência (Passo a Passo)")
177
+
178
+ btn.click(orquestrador, inp, [out_relatorio, out_timeline])
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
179
 
180
+ return app
 
 
 
 
181
 
182
+ # ==================== PSEUDO-CÓDIGO (DOCUMENTAÇÃO) ====================
183
+ """
184
+ COMO ESTE CÓDIGO PENSA (Lógica Humanizada):
185
 
186
+ 1. RECEBER(Caso) -> Cria uma Lista: [Caso]
 
 
 
 
 
 
 
 
187
 
188
+ 2. PARA CADA Agente NA Lista_de_Instruções:
189
+ A. O Agente LÊ toda a Lista até agora (memória perfeita).
190
+ B. O Agente EXECUTA sua instrução específica (ex: "Ache mentiras").
191
+ C. O Agente ESCREVE sua conclusão no final da Lista.
 
 
 
192
 
193
+ Exemplo Visual da Lista crescendo:
194
+ [
195
+ {User: "Ele disse que não bateu..."},
196
+ {Escrivão: "Fatos: Agressão dia 20..."},
197
+ {Psicólogo: "Tom: Ele está manipulando..."},
198
+ {Investigador: "Mentira detectada: Ele nega mas o áudio prova..."},
199
+ {Promotor: "Relatório Final: Com base na mentira detectada pelo Investigador..."}
200
+ ]
201
+
202
+ 3. EXIBIR o trabalho do último Agente (Promotor).
203
+ """
 
 
204
 
205
  if __name__ == "__main__":
206
+ interface().launch()