Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
+import json
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 
@@ -9,7 +10,7 @@ client = InferenceClient(
     token=HF_TOKEN
 )
 
-# Aqui você cola o JSON normalmente, sem escapar nada
+# Seu AGENTE_JSON completo
 AGENTE_JSON = """
 {
   "nome_agente": "IndenizaAI",
@@ -31,7 +32,7 @@ AGENTE_JSON = """
     "Total geral da indenização"
   ],
   "comportamento_resposta": {
-    "estilo": 'técnico, claro e organizado',
+    "estilo": "técnico, claro e organizado",
     "estrutura": [
       "👉 Resumo inicial do caso",
       "👉 Tabela detalhada de cada cálculo",
@@ -47,12 +48,26 @@ AGENTE_JSON = """
 }
 """
 
-def montar_mensagem_usuario(msg):
+# Campos que queremos armazenar do usuário
+CAMPOS_NECESSARIOS = [
+    "nome_trabalhador", "idade", "cargo", "tempo_servico",
+    "renda_mensal", "contrato_trabalho",
+    "data_acidente", "tipo_acidente", "descricao_acidente", "tempo_afastamento",
+    "sequelas", "custos_medicos", "perda_capacidade",
+    "gastos_tratamento", "perda_rendimentos"
+]
+
+# Armazenamento em memória (pode ser substituído por DB)
+usuarios_dados = {}
+
+def montar_mensagem_usuario(msg, dados_completos):
     return (
         "Você é o IndenizaAI.\n"
         "Siga integralmente todas as instruções do JSON abaixo:\n"
         "### CONFIGURAÇÃO DO AGENTE ###\n"
         + AGENTE_JSON +
+        "\n### DADOS COLETADOS DO USUÁRIO ###\n"
+        + json.dumps(dados_completos, indent=2) +
         "\n### MENSAGEM DO USUÁRIO ###\n"
         + msg +
         "\n### INSTRUÇÃO CRÍTICA ###\n"
@@ -63,32 +78,49 @@ def montar_mensagem_usuario(msg):
         "Responda como **IndenizaAI**."
     )
 
-def responder(mensagem, historico):
-    prompt = montar_mensagem_usuario(mensagem)
+def verificar_dados_completos(dados):
+    faltando = [campo for campo in CAMPOS_NECESSARIOS if campo not in dados or not dados[campo]]
+    return faltando
 
-    messages = [
-        {"role": "system", "content": "Você é o IndenizaAI, especialista em cálculos de indenizações trabalhistas."},
-        {"role": "user", "content": prompt}
-    ]
+def responder(mensagem, historico, usuario_id="default"):
+    # Recupera dados armazenados do usuário
+    dados_usuario = usuarios_dados.get(usuario_id, {})
 
-    result = client.chat_completion(
-        messages=messages,
-        max_tokens=1200,
-        temperature=0.1,
-        top_p=0.9
-    )
+    # Tenta atualizar dados a partir de JSON enviado pelo usuário
+    try:
+        dados_parciais = json.loads(mensagem)
+        dados_usuario.update(dados_parciais)
+    except:
+        pass  # mensagem normal, sem dados estruturados
+
+    usuarios_dados[usuario_id] = dados_usuario
+
+    faltando = verificar_dados_completos(dados_usuario)
+
+    if faltando:
+        resposta = f"⚠️ Ainda faltam os seguintes dados para realizar o cálculo: {', '.join(faltando)}"
+    else:
+        prompt = montar_mensagem_usuario(mensagem, dados_usuario)
+        messages = [
+            {"role": "system", "content": "Você é o IndenizaAI, especialista em cálculos de indenizações trabalhistas."},
+            {"role": "user", "content": prompt}
+        ]
+        result = client.chat_completion(
+            messages=messages,
+            max_tokens=1200,
+            temperature=0.1,
+            top_p=0.9
+        )
+        resposta = result.choices[0].message["content"]
 
-    resposta = result.choices[0].message["content"]
     historico.append({"role": "user", "content": mensagem})
     historico.append({"role": "assistant", "content": resposta})
     return historico, ""
 
 with gr.Blocks() as demo:
     gr.Markdown("## ⚖️ IndenizaAI — Assistente Jurídico Especializado")
-
     chat = gr.Chatbot(type="messages")
     inp = gr.Textbox(label="Digite sua mensagem")
-
     inp.submit(responder, [inp, chat], [chat, inp])
 
 demo.launch()