Update app.py

app.py

@@ -13,7 +13,8 @@ client = InferenceClient(api_key=api_key)
 # Função para gerar a peça jurídica em português do Brasil
 def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     """
-    Gera uma peça jurídica criminal em português do Brasil com base no tipo e nas informações fornecidas.
+    Gera uma peça jurídica criminal em português do Brasil com base no tipo e nas informações fornecidas,
+    realizando duas requisições para uma resposta mais completa e ampla.
 
     Parâmetros:
         informacoes_caso (str): Descrição detalhada do caso.
@@ -26,7 +27,7 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     if not tipo_peca:
         return "Por favor, selecione o tipo de peça jurídica."
 
-    mensagens = [
+    mensagens_iniciais = [
         {
             "role": "system",
             "content": (
@@ -40,16 +41,44 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
         }
     ]
 
-    # Chamada ao modelo para gerar a resposta
-    resposta = client.chat.completions.create(
+    # Primeira requisição: Geração inicial da peça
+    resposta_inicial = client.chat.completions.create(
         model="mistralai/Mistral-7B-Instruct-v0.3",
-        messages=mensagens,
+        messages=mensagens_iniciais,
         max_tokens=1000
     )
+    
+    peca_inicial = resposta_inicial.choices[0].message["content"]
+
+    # Segunda requisição: Expansão da peça jurídica
+    mensagens_expansao = [
+        {
+            "role": "system",
+            "content": (
+                "Você é um especialista em direito criminal no Brasil e deve expandir a peça jurídica gerada, "
+                "adicionando mais detalhes, justificativas e argumentos legais. "
+                "Todas as respostas devem estar em português do Brasil, considerando o ordenamento jurídico brasileiro."
+            )
+        },
+        {
+            "role": "user",
+            "content": f"Agora, com base na peça inicial a seguir, adicione mais detalhes, justificativas e argumentos legais:\n{peca_inicial}"
+        }
+    ]
+
+    # Segunda requisição: Expansão e aprofundamento da peça jurídica
+    resposta_expansao = client.chat.completions.create(
+        model="mistralai/Mistral-7B-Instruct-v0.3",
+        messages=mensagens_expansao,
+        max_tokens=1000
+    )
+
+    peca_expandida = resposta_expansao.choices[0].message["content"]
+
+    # Formatação da peça jurídica final
+    peca_completa = f"**Peça Jurídica: {tipo_peca.capitalize()}**\n\n{peca_expandida}"
+    return peca_completa
 
-    # Formatação da peça jurídica
-    peca = resposta.choices[0].message["content"]
-    return f"**Peça Jurídica: {tipo_peca.capitalize()}**\n\n{peca}"
 
 # Função para redefinir os campos
 def redefinir_campos():