Update app.py

app.py

@@ -56,13 +56,14 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
             "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."
             )
         },
         {
             "role": "user",
-            "content": f"Agora, com base na peça inicial a seguir, adicione mais detalhes, justificativas e argumentos legais:\n{peca_inicial}"
+            "content": f"Agora, com base na peça inicial a seguir, adicione mais detalhes, justificativas e argumentos legais, "
+                       "sem repetir o conteúdo já gerado:\n{peca_inicial}"
         }
     ]
 
@@ -73,16 +74,14 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
         max_tokens=1000
     )
 
-    peca_expandida = resposta_expansao.choices[0].message["content"]
+    peca_expansao = resposta_expansao.choices[0].message["content"]
 
-    # 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}"
+    # Combina a peça inicial com a expansão, garantindo um documento completo sem duplicação
+    peca_completa = f"**Peça Jurídica: {tipo_peca.capitalize()}**\n\n{peca_inicial.strip()}\n\n{peca_expansao.strip()}"
 
     return peca_completa
 
 
-
-
 # Função para redefinir os campos
 def redefinir_campos():
     return "", "", ""