Update app.py

app.py

@@ -13,8 +13,7 @@ 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,
-    realizando duas requisições para uma resposta mais completa e ampla.
+    Gera uma peça jurídica criminal em português do Brasil com base no tipo e nas informações fornecidas.
 
     Parâmetros:
         informacoes_caso (str): Descrição detalhada do caso.
@@ -27,6 +26,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 para o modelo
     mensagens_iniciais = [
         {
             "role": "system",
@@ -62,8 +62,7 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
         },
         {
             "role": "user",
-            "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}"
+            "content": f"Agora, com base na peça inicial a seguir, adicione mais detalhes, justificativas e argumentos legais:\n{peca_inicial}"
         }
     ]
 
@@ -74,14 +73,16 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
         max_tokens=1000
     )
 
-    peca_expansao = resposta_expansao.choices[0].message["content"]
+    peca_expandida = resposta_expansao.choices[0].message["content"]
 
-    # 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()}"
+    # Combina a peça inicial com a expansão, garantindo um documento completo e coerente
+    peca_completa = f"**Peça Jurídica: {tipo_peca.capitalize()}**\n\n{peca_inicial}\n\n{peca_expandida}"
 
+    # Retorna a peça completa
     return peca_completa
 
 
+
 # Função para redefinir os campos
 def redefinir_campos():
     return "", "", ""