Update app.py

app.py

@@ -25,7 +25,7 @@ class DefenseGenerator:
             response = self.client.chat.completions.create(
                 model="mistralai/Mistral-7B-Instruct-v0.3",
                 messages=[
-                    {"role": "system", "content": "Você é um advogado criminalista experiente."},
+                    {"role": "system", "content": "Você é um advogado criminalista experiente. Gere peças sem usar asteriscos ou marcadores. Use numeração romana para seções principais (I, II, III...) e números arábicos para subitens (1, 2, 3...). Mantenha formatação consistente e profissional."},
                     {"role": "user", "content": prompt}
                 ],
                 max_tokens=2000,
@@ -36,18 +36,49 @@ class DefenseGenerator:
             return f"Erro na geração: {str(e)}"
 
     def gerar_documento(self, tipo: str, fatos: str, dados_basicos: dict) -> str:
-        prompt = f"""Gere uma peça de {self.document_types[tipo]} completa com base nos seguintes fatos:
+        prompt = f"""Gere uma peça de {self.document_types[tipo]} completa, formatada e sem asteriscos, com base nos seguintes fatos:
 
 {fatos}
 
-Inclua todos os elementos necessários:
-- Qualificação completa das partes
-- Fundamentos jurídicos pertinentes
-- Jurisprudência relevante
-- Doutrina aplicável
-- Preliminares quando cabíveis
-- Mérito detalhado
-- Pedidos específicos
+A peça deve seguir este formato:
+
+EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) [AUTORIDADE]
+
+[Título da Peça em Maiúsculas]
+
+I - QUALIFICAÇÃO DAS PARTES
+[Nome e dados completos]
+
+II - FUNDAMENTOS JURÍDICOS
+1. [Primeiro fundamento]
+2. [Segundo fundamento]
+[...]
+
+III - JURISPRUDÊNCIA APLICÁVEL
+1. [Primeira jurisprudência]
+2. [Segunda jurisprudência]
+[...]
+
+IV - DOUTRINA APLICÁVEL
+[Citações doutrinárias relevantes]
+
+V - PRELIMINARES
+1. [Primeira preliminar]
+2. [Segunda preliminar]
+[...]
+
+VI - MÉRITO
+[Desenvolvimento dos argumentos]
+
+VII - PEDIDOS
+1. [Primeiro pedido]
+2. [Segundo pedido]
+[...]
+
+Local, Data.
+
+[Advogado]
+OAB/[Estado] [Número]
 
 Dados básicos para formatação:
 Comarca: {dados_basicos['comarca']}