Update app.py

app.py

@@ -50,13 +50,13 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     
     peca_inicial = resposta_inicial.choices[0].message["content"]
 
-    # Segunda requisição: Expansão da peça jurídica
+    # Segunda requisição: Expansão da peça jurídica com mais detalhes e argumentos
     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. "
+                "adicionando mais detalhes, justificativas, e argumentos legais. "
                 "Todas as respostas devem estar em português do Brasil, considerando o ordenamento jurídico brasileiro."
             )
         },
@@ -75,11 +75,13 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
 
     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}"
+    # Combina a peça inicial com a expansão, garantindo um documento completo
+    peca_completa = f"**Peça Jurídica: {tipo_peca.capitalize()}**\n\n{peca_inicial}\n\n{peca_expandida}"
+
     return peca_completa
 
 
+
 # Função para redefinir os campos
 def redefinir_campos():
     return "", "", ""