Update app.py

app.py

@@ -17,16 +17,17 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
 
     template = {
         "estrutura": [
-            "EXCELENTÍSSIMO(A) JUIZ(A)",
-            "FATOS",
-            "DIREITO",
-            "PEDIDOS"
+            "EXCELENTÍSSIMO(A) SENHOR(A) JUIZ(A) DE DIREITO",
+            "DOS FATOS",
+            "DO DIREITO",
+            "DOS PEDIDOS"
         ],
         "prompt": (
-            "Gere HC conciso com:\n"
-            "- Fatos essenciais\n"
-            "- Art. 5º LXVIII CF, 647/648 CPP\n"
-            "- 1-2 precedentes chave\n"
+            "Elabore um Habeas Corpus conciso incluindo:\n"
+            "- Fatos essenciais do caso\n"
+            "- Artigo 5º, LXVIII da Constituição Federal\n" 
+            "- Artigos 647 e 648 do Código de Processo Penal\n"
+            "- Jurisprudência relevante\n"
             "- Pedido liminar e principal"
         )
     }
@@ -34,7 +35,7 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     mensagens_estrutura = [
         {
             "role": "system",
-            "content": "Redija HC objetivo. Foque no constrangimento ilegal."
+            "content": "Como advogado criminalista brasileiro, redija um HC objetivo em português formal. Foque no constrangimento ilegal."
         },
         {
             "role": "user",
@@ -54,11 +55,11 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     mensagens_elaboracao = [
         {
             "role": "system",
-            "content": "Fundamente com artigos e 1 precedente chave por argumento."
+            "content": "Como advogado criminalista brasileiro, fundamente cada argumento com legislação e jurisprudência pertinente."
         },
         {
             "role": "user",
-            "content": f"Revise mantendo apenas argumentos essenciais:\n{estrutura_base}"
+            "content": f"Aprimore esta peça mantendo os argumentos essenciais:\n{estrutura_base}"
         }
     ]
 
@@ -72,11 +73,11 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     mensagens_final = [
         {
             "role": "system",
-            "content": "Revise tecnicamente mantendo concisão."
+            "content": "Faça a revisão técnica final em português jurídico formal."
         },
         {
             "role": "user",
-            "content": f"Finalize:\n{estrutura_base}\n\n{resposta_elaboracao.choices[0].message['content']}"
+            "content": f"Finalize a peça:\n{estrutura_base}\n\n{resposta_elaboracao.choices[0].message['content']}"
         }
     ]
 
@@ -88,6 +89,7 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     )
 
     return resposta_final.choices[0].message["content"]
+    
 # Função para redefinir os campos
 def redefinir_campos():
     return "", "", ""