Update app.py

app.py

@@ -158,11 +158,11 @@ class PromptManager:
         return prompts.get(doc_type, "Tipo de documento não suportado")
 
 class DocumentGenerator:
-    """Gerencia a geração de documentos usando modelo BERT público"""
+    """Gerencia a geração de documentos usando um modelo público"""
     
     def __init__(self):
-        # Usando modelo BERT público e estável
-        self.client = Client("microsoft/phi-1_5")  # Modelo mais leve e estável
+        # Usando um Space público de geração de texto
+        self.client = Client("abidlabs/gpt4all")  # Space público e estável
         self.prompt_manager = PromptManager()
         self.max_retries = 3
     
@@ -172,14 +172,13 @@ class DocumentGenerator:
             try:
                 prompt = self.prompt_manager.get_base_prompt(doc_type, context)
                 
-                # Configuração específica para o modelo
+                # Configuração específica para o GPT4All
                 result = self.client.predict(
-                    prompt,  # texto principal
-                    max_length=2048,  # tamanho máximo
-                    temperature=0.7,  # criatividade moderada
-                    top_p=0.95,  # coerência
-                    do_sample=True,  # amostragem
-                    api_name="/run"  # endpoint padrão
+                    prompt,            # Prompt para geração
+                    "portuguese",      # Idioma
+                    0.7,              # Temperatura
+                    1024,             # Tamanho máximo
+                    api_name="/predict"
                 )
                 
                 return self._format_output(result)
@@ -196,7 +195,7 @@ class DocumentGenerator:
                     elif "timeout" in error_msg.lower():
                         return "O servidor está demorando para responder. Por favor, tente novamente."
                     else:
-                        return f"Não foi possível gerar o documento. Por favor, verifique os dados e tente novamente."
+                        return "Não foi possível gerar o documento. Por favor, tente novamente mais tarde."
     
     def _format_output(self, text: str) -> str:
         """Formata o texto gerado para melhor apresentação"""