Update app.py

app.py

@@ -25,50 +25,70 @@ class DocumentGenerator:
             messages = [
                 {
                     "role": "system",
-                    "content": """Você é um advogado criminalista brasileiro. Gere documentos jurídicos concisos e objetivos, 
-                    sem repetições desnecessárias. Use linguagem técnica apropriada e citações específicas."""
+                    "content": """Você é um advogado criminalista brasileiro altamente experiente.
+                    Gere documentos jurídicos seguindo rigorosamente o formato e padrão das peças processuais brasileiras.
+                    Use linguagem formal, técnica e apropriada ao meio jurídico brasileiro."""
                 },
                 {
                     "role": "user",
-                    "content": f"""Gere um {doc_type} objetivo e sem repetições com:
-                    
-                    QUALIFICAÇÃO:
-                    Cliente: {context.get('client_name')}
-                    Processo: {context.get('process_number')}
-                    Tribunal: {context.get('court')}
-                    Comarca: {context.get('jurisdiction')}
-                    
-                    FATOS:
-                    {context.get('facts')}
-                    
-                    FUNDAMENTOS JURÍDICOS:
-                    {context.get('legal_basis')}
-                    
-                    Regras:
-                    1. Seja conciso e direto
-                    2. Evite repetir informações
-                    3. Cite apenas uma vez cada artigo/jurisprudência
-                    4. Use formatação jurídica padrão
-                    5. Inclua apenas informações relevantes"""
+                    "content": f"""Gere um {doc_type} seguindo estritamente o formato jurídico brasileiro padrão:
+
+EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA VARA CRIMINAL DA COMARCA DE {context.get('jurisdiction').upper()}
+
+Processo nº {context.get('process_number')}
+
+{context.get('client_name')}, brasileiro, devidamente qualificado nos autos do processo em epígrafe, por seu advogado que esta subscreve, vem, respeitosamente, à presença de Vossa Excelência, com fundamento no artigo 5º, LXVIII, da Constituição Federal e artigos 647 e seguintes do Código de Processo Penal, impetrar:
+
+HABEAS CORPUS
+com pedido de liminar
+
+em face do constrangimento ilegal sofrido, pelos fatos e fundamentos a seguir expostos:
+
+I - DOS FATOS
+{context.get('facts')}
+
+II - DO DIREITO
+{context.get('legal_basis')}
+
+III - DOS PEDIDOS
+
+Ante o exposto, requer-se:
+
+a) A concessão de LIMINAR para determinar a imediata expedição de alvará de soltura em favor do paciente;
+
+b) No mérito, a concessão definitiva da ordem para revogar a prisão preventiva, expedindo-se o competente alvará de soltura em favor do paciente, se por outro motivo não estiver preso.
+
+Nestes termos,
+Pede deferimento.
+
+{context.get('jurisdiction')}, [DATA]
+
+[Nome do Advogado]
+OAB/XX XXX.XXX"""
                 }
             ]
             
-            # Gera o documento com parâmetros ajustados
             completion = self.client.chat.completions.create(
                 model=self.model,
                 messages=messages,
-                temperature=0.3,           # Reduzido para maior consistência
-                top_p=0.85,               # Mantém coerência
+                temperature=0.1,           # Reduzido para maior formalidade
+                top_p=0.8,                # Mantém coerência
                 max_tokens=2048,          # Limite de tamanho
-                presence_penalty=0.6,     # Penaliza repetições
-                frequency_penalty=0.8     # Penaliza fortemente palavras repetidas
+                presence_penalty=0.3,     # Reduz repetições mantendo formalidade
+                frequency_penalty=0.5     # Reduz repetições de palavras
             )
             
-            return self._format_output(completion.choices[0].message.content)
+            # Formata a data
+            result = completion.choices[0].message.content
+            current_date = datetime.now().strftime('%d de %B de %Y')
+            result = result.replace('[DATA]', current_date)
+            
+            return self._format_output(result)
             
         except Exception as e:
             logger.error(f"Erro na geração: {str(e)}")
             return f"Erro na geração do documento: {str(e)}"
+
     
     def _format_output(self, text: str) -> str:
         """Formata o texto gerado"""