Update app.py

app.py

@@ -41,42 +41,11 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
             "content": f"{template['prompt']}\nCaso: {informacoes_caso}"
         }
     ]
-        },
-        "petição inicial": {
-            "estrutura": [
-                "QUALIFICAÇÃO",
-                "DOS FATOS",
-                "DO DIREITO",
-                "DOS PEDIDOS"
-            ],
-            "prompt": "Elabore uma petição inicial criminal contemplando os requisitos do Art. 41 do CPP"
-        }
-    }
-
-    template = templates.get(tipo_peca.lower(), {
-        "estrutura": ["INTRODUÇÃO", "DESENVOLVIMENTO", "CONCLUSÃO"],
-        "prompt": f"Elabore uma {tipo_peca} criminal fundamentada."
-    })
-
-    mensagens_estrutura = [
-        {
-            "role": "system",
-            "content": (
-                "Você é um advogado criminalista especialista. "
-                "Formate a peça conforme a estrutura jurídica brasileira."
-            )
-        },
-        {
-            "role": "user",
-            "content": f"{template['prompt']}\n\nCaso: {informacoes_caso}\n\n"
-                      f"Estruture com:\n{chr(10).join(template['estrutura'])}"
-        }
-    ]
 
     resposta_estrutura = client.chat.completions.create(
         model="mistralai/Mistral-7B-Instruct-v0.3",
         messages=mensagens_estrutura,
-        max_tokens=1500,
+        max_tokens=1000,
         temperature=0.7
     )
 
@@ -92,44 +61,33 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
             "content": f"Revise mantendo apenas argumentos essenciais:\n{estrutura_base}"
         }
     ]
-        },
-        {
-            "role": "user",
-            "content": f"Aprimore esta peça:\n{estrutura_base}"
-        }
-    ]
 
     resposta_elaboracao = client.chat.completions.create(
         model="mistralai/Mistral-7B-Instruct-v0.3",
         messages=mensagens_elaboracao,
-        max_tokens=1500,
+        max_tokens=1000,
         temperature=0.7
     )
 
     mensagens_final = [
         {
             "role": "system",
-            "content": "Faça a revisão técnica final garantindo precisão e formatação."
+            "content": "Revise tecnicamente mantendo concisão."
         },
         {
             "role": "user",
-            "content": (
-                f"Revise e formate:\n"
-                f"{estrutura_base}\n\n"
-                f"Complementos:\n{resposta_elaboracao.choices[0].message['content']}"
-            )
+            "content": f"Finalize:\n{estrutura_base}\n\n{resposta_elaboracao.choices[0].message['content']}"
         }
     ]
 
     resposta_final = client.chat.completions.create(
         model="mistralai/Mistral-7B-Instruct-v0.3",
         messages=mensagens_final,
-        max_tokens=1500,
+        max_tokens=1000,
         temperature=0.5
     )
 
     return resposta_final.choices[0].message["content"]
-
 # Função para redefinir os campos
 def redefinir_campos():
     return "", "", ""