Update app.py

app.py

@@ -158,43 +158,53 @@ class PromptManager:
         return prompts.get(doc_type, "Tipo de documento não suportado")
 
 class DocumentGenerator:
-    """Gerencia a geração de documentos usando modelo Mistral público"""
+    """Gerencia a geração de documentos usando modelo BERT público"""
     
     def __init__(self):
-        # Usando modelo Mistral público
-        self.client = Client("HuggingFaceH4/zephyr-7b-beta")
+        # Usando modelo BERT público e estável
+        self.client = Client("microsoft/phi-1_5")  # Modelo mais leve e estável
         self.prompt_manager = PromptManager()
+        self.max_retries = 3
     
     def generate(self, doc_type: str, context: Dict[str, str]) -> str:
         """Gera o documento baseado no tipo e contexto"""
-        try:
-            prompt = self.prompt_manager.get_base_prompt(doc_type, context)
-            
-            # Configuração específica para o modelo
-            result = self.client.predict(
-                prompt,  # prompt principal
-                max_new_tokens=2048,  # limite de tokens de saída
-                temperature=0.7,  # criatividade moderada
-                top_p=0.95,  # mantém a coerência
-                repetition_penalty=1.15,  # evita repetições
-                api_name="/generate"  # endpoint específico
-            )
-            
-            return self._format_output(result)
-            
-        except Exception as e:
-            error_msg = str(e)
-            logger.error(f"Erro na geração do documento: {error_msg}")
-            
-            if "quota" in error_msg.lower():
-                return "O serviço está temporariamente sobrecarregado. Por favor, tente novamente em alguns minutos."
-            elif "timeout" in error_msg.lower():
-                return "O servidor demorou muito para responder. Por favor, tente novamente."
-            else:
-                return f"Erro na geração do documento. Por favor, verifique os dados e tente novamente. Detalhes: {error_msg}"
+        for attempt in range(self.max_retries):
+            try:
+                prompt = self.prompt_manager.get_base_prompt(doc_type, context)
+                
+                # Configuração específica para o modelo
+                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
+                )
+                
+                return self._format_output(result)
+                
+            except Exception as e:
+                error_msg = str(e)
+                logger.error(f"Tentativa {attempt + 1} falhou: {error_msg}")
+                
+                if attempt < self.max_retries - 1:
+                    continue
+                else:
+                    if "quota" in error_msg.lower():
+                        return "Serviço temporariamente indisponível. Por favor, tente novamente em alguns minutos."
+                    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."
     
     def _format_output(self, text: str) -> str:
         """Formata o texto gerado para melhor apresentação"""
+        if not isinstance(text, str):
+            if isinstance(text, (list, tuple)) and len(text) > 0:
+                text = str(text[0])
+            else:
+                text = str(text)
         return text.strip()
 
 class WebInterface: