Update app.py

app.py

@@ -18,19 +18,19 @@ class DocumentGenerator:
     def __init__(self):
         self.client = InferenceClient(api_key=HF_TOKEN)
         self.model = "mistralai/Mistral-7B-Instruct-v0.2"
-    
+        
     def generate(self, doc_type: str, context: Dict[str, str]) -> str:
         """Gera o documento usando o modelo"""
         try:
             messages = [
                 {
                     "role": "system",
-                    "content": """Você é um advogado criminalista brasileiro experiente. 
-                    Gere documentos jurídicos formais no formato do direito brasileiro."""
+                    "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."""
                 },
                 {
                     "role": "user",
-                    "content": f"""Gere um {doc_type} completo e tecnicamente preciso com:
+                    "content": f"""Gere um {doc_type} objetivo e sem repetições com:
                     
                     QUALIFICAÇÃO:
                     Cliente: {context.get('client_name')}
@@ -42,17 +42,26 @@ class DocumentGenerator:
                     {context.get('facts')}
                     
                     FUNDAMENTOS JURÍDICOS:
-                    {context.get('legal_basis')}"""
+                    {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"""
                 }
             ]
             
-            # Gera o documento
+            # Gera o documento com parâmetros ajustados
             completion = self.client.chat.completions.create(
                 model=self.model,
                 messages=messages,
-                temperature=0.3,
-                top_p=0.85,
-                max_tokens=2048
+                temperature=0.3,           # Reduzido para maior consistência
+                top_p=0.85,               # 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
             )
             
             return self._format_output(completion.choices[0].message.content)