Update app.py

app.py

@@ -17,48 +17,94 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
 
     templates = {
         "habeas corpus": {
-            "estrutura": ["EXORDIAL", "FATOS", "DIREITO", "PEDIDOS"],
+            "estrutura": [
+                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA [VARA] DA [COMARCA]",
+                "QUALIFICAÇÃO DO PACIENTE",
+                "DOS FATOS",
+                "DO DIREITO",
+                "DO CONSTRANGIMENTO ILEGAL",
+                "DA MEDIDA LIMINAR",
+                "DOS PEDIDOS",
+                "PROVAS E DOCUMENTOS",
+                "FECHAMENTO",
+            ],
             "prompt": (
-                "HC conciso:\n"
-                "1. Fatos e constrangimento ilegal\n"
-                "2. CF art 5º LXVIII e CPP 647/648\n"
-                "3. Precedente STF/STJ\n"
-                "4. Liminar + principal"
+                "HC técnico:\n"
+                "1. Qualificação completa\n"
+                "2. Fatos e provas do constrangimento\n"
+                "3. CF art 5º LXVIII e CPP 647/648\n"
+                "4. STF/STJ sobre o tema\n"
+                "5. Pedido liminar fundamentado\n"
+                "6. Rol de documentos"
             ),
-            "max_tokens": 300
+            "max_tokens": 400
         },
         "petição inicial": {
-            "estrutura": ["EXORDIAL", "QUALIFICAÇÃO", "FATOS", "DIREITO", "PEDIDOS"],
+            "estrutura": [
+                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) PROMOTOR(A) DE JUSTIÇA",
+                "QUALIFICAÇÃO DAS PARTES",
+                "DOS FATOS E FUNDAMENTOS",
+                "DA TIPIFICAÇÃO LEGAL",
+                "DAS PROVAS E DILIGÊNCIAS",
+                "DO ROL DE TESTEMUNHAS",
+                "DOS PEDIDOS",
+                "DO VALOR DA CAUSA",
+                "FECHAMENTO"
+            ],
             "prompt": (
-                "Denúncia criminal:\n"
-                "1. Qualificação completa\n"
-                "2. Fatos e tipificação\n"
-                "3. Arts. 41/395 CPP\n"
-                "4. Provas e diligências"
+                "Denúncia Criminal art. 41 CPP:\n"
+                "1. Qualificação completa partes\n"
+                "2. Narrativa clara dos fatos\n"
+                "3. Tipificação detalhada\n"
+                "4. Provas e diligências\n"
+                "5. Testemunhas até 8\n"
+                "6. Pedidos condenatórios"
             ),
-            "max_tokens": 400
+            "max_tokens": 500
         },
         "defesa prévia": {
-            "estrutura": ["QUALIFICAÇÃO", "PRELIMINARES", "MÉRITO", "PROVAS", "PEDIDOS"],
+            "estrutura": [
+                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO",
+                "QUALIFICAÇÃO DO RÉU",
+                "DAS PRELIMINARES",
+                "DO MÉRITO",
+                "DAS PROVAS",
+                "DO ROL DE TESTEMUNHAS",
+                "DOS REQUERIMENTOS",
+                "FECHAMENTO"
+            ],
             "prompt": (
                 "Defesa art. 396-A CPP:\n"
-                "1. Preliminares\n"
-                "2. Tese defensiva\n"
-                "3. Provas a produzir\n"
-                "4. Testemunhas"
+                "1. Dados do processo\n"
+                "2. Preliminares processuais\n"
+                "3. Tese defensiva\n"
+                "4. Provas documentais\n"
+                "5. Testemunhas até 8\n"
+                "6. Requerimentos finais"
             ),
-            "max_tokens": 350
+            "max_tokens": 450
         },
         "apelação criminal": {
-            "estrutura": ["TEMPESTIVIDADE", "FATOS", "PRELIMINARES", "MÉRITO", "PEDIDOS"],
+            "estrutura": [
+                "EGRÉGIO TRIBUNAL DE JUSTIÇA",
+                "TEMPESTIVIDADE",
+                "RESUMO PROCESSUAL",
+                "DAS PRELIMINARES",
+                "DO MÉRITO RECURSAL",
+                "DAS CONTRARRAZÕES",
+                "DOS PEDIDOS",
+                "FECHAMENTO"
+            ],
             "prompt": (
-                "Apelação criminal:\n"
-                "1. Art. 593 CPP\n" 
-                "2. Preliminares\n"
-                "3. Error in judicando\n"
-                "4. Reforma/nulidade"
+                "Apelação art. 593 CPP:\n" 
+                "1. Tempestividade\n"
+                "2. Síntese processual\n"
+                "3. Preliminares\n"
+                "4. Razões de reforma\n"
+                "5. Contrarrazões\n"
+                "6. Pedido específico"
             ),
-            "max_tokens": 450
+            "max_tokens": 500
         }
     }
 
@@ -69,11 +115,22 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
     mensagens_estrutura = [
         {
             "role": "system",
-            "content": f"Redija {tipo_peca} técnica em português jurídico. Máximo {template['max_tokens']} tokens."
+            "content": (
+                f"Como advogado criminalista, redija {tipo_peca} técnica seguindo:\n"
+                f"1. Estrutura formal\n"
+                f"2. Linguagem jurídica\n"
+                f"3. Fundamentação precisa\n"
+                f"4. Formatação correta"
+            )
         },
         {
             "role": "user",
-            "content": f"{template['prompt']}\nCaso: {informacoes_caso}"
+            "content": (
+                f"Elabore seguindo estrutura:\n"
+                f"{chr(10).join(template['estrutura'])}\n\n"
+                f"Requisitos:\n{template['prompt']}\n\n"
+                f"Caso:\n{informacoes_caso}"
+            )
         }
     ]
 
@@ -89,11 +146,17 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
     mensagens_elaboracao = [
         {
             "role": "system",
-            "content": f"Aprimore {tipo_peca} com fundamentação precisa. Máximo {template['max_tokens']} tokens."
+            "content": (
+                f"Como especialista criminal, aprimore {tipo_peca} com:\n"
+                "1. Legislação pertinente\n"
+                "2. Jurisprudência atual\n"
+                "3. Doutrina relevante\n"
+                "4. Fundamentação técnica"
+            )
         },
         {
             "role": "user",
-            "content": f"Revise:\n{estrutura_base}"
+            "content": f"Revise e complemente:\n{estrutura_base}"
         }
     ]
 
@@ -107,11 +170,22 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
     mensagens_final = [
         {
             "role": "system",
-            "content": f"Finalize {tipo_peca} em formato técnico. Estrutura: {' > '.join(template['estrutura'])}"
+            "content": (
+                f"Finalize {tipo_peca} garantindo:\n"
+                "1. Estrutura correta\n"
+                "2. Numeração parágrafos\n"
+                "3. Formatação adequada\n"
+                "4. Citações precisas"
+            )
         },
         {
             "role": "user",
-            "content": f"Versão final:\n{estrutura_base}\n\n{resposta_elaboracao.choices[0].message['content']}"
+            "content": (
+                f"Versão final seguindo:\n"
+                f"{chr(10).join(template['estrutura'])}\n\n"
+                f"Conteúdo base:\n{estrutura_base}\n\n"
+                f"Complementos:\n{resposta_elaboracao.choices[0].message['content']}"
+            )
         }
     ]