Update app.py

app.py

@@ -15,29 +15,32 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     if not tipo_peca or not informacoes_caso:
         return "Por favor, forneça todas as informações necessárias."
 
-    templates = {
-        "habeas corpus": {
-            "estrutura": [
-                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO",
-                "QUALIFICAÇÃO",
-                "DOS FATOS",
-                "FUNDAMENTOS JURÍDICOS",
-                "DO CABIMENTO DO HABEAS CORPUS",
-                "DA ILEGALIDADE DA PRISÃO",
-                "DOS PEDIDOS",
-                "FECHAMENTO"
-            ],
-            "prompt": (
-                "Gere um Habeas Corpus tecnicamente preciso incluindo:\n"
-                "- Qualificação do paciente\n"
-                "- Narrativa dos fatos\n"
-                "- CPP Arts. 647/648 e jurisprudência:\n"
-                "  * STF HC 148.459/DF sobre excesso de prazo\n"
-                "  * STJ HC 509.030/SP sobre ausência de fundamentação\n"
-                "  * STF HC 137.728/RJ sobre presunção de inocência\n"
-                "- Súmulas 691 e 692 do STF\n"
-                "- Pedidos liminar e principal detalhados"
-            )
+    template = {
+        "estrutura": [
+            "EXCELENTÍSSIMO(A) JUIZ(A)",
+            "FATOS",
+            "DIREITO",
+            "PEDIDOS"
+        ],
+        "prompt": (
+            "Gere HC conciso com:\n"
+            "- Fatos essenciais\n"
+            "- Art. 5º LXVIII CF, 647/648 CPP\n"
+            "- 1-2 precedentes chave\n"
+            "- Pedido liminar e principal"
+        )
+    }
+
+    mensagens_estrutura = [
+        {
+            "role": "system",
+            "content": "Redija HC objetivo. Foque no constrangimento ilegal."
+        },
+        {
+            "role": "user",
+            "content": f"{template['prompt']}\nCaso: {informacoes_caso}"
+        }
+    ]
         },
         "petição inicial": {
             "estrutura": [
@@ -82,17 +85,13 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     mensagens_elaboracao = [
         {
             "role": "system",
-            "content": (
-                "Atue como advogado criminalista especialista em habeas corpus. Para cada seção:\n"
-                "1. PREÂMBULO: Qualificação completa e formal\n"
-                "2. DOS FATOS: Narrativa cronológica e objetiva\n"
-                "3. DO DIREITO:\n"
-                "   - Art. 5º, LXVIII CF/88 (texto completo)\n" 
-                "   - Arts. 647/648 CPP (transcrição)\n"
-                "   - Jurisprudência com ementas\n"
-                "4. PEDIDOS: Principal e liminar fundamentados\n"
-                "Use linguagem técnica e formal. Numere parágrafos."
-            )
+            "content": "Fundamente com artigos e 1 precedente chave por argumento."
+        },
+        {
+            "role": "user",
+            "content": f"Revise mantendo apenas argumentos essenciais:\n{estrutura_base}"
+        }
+    ]
         },
         {
             "role": "user",