Update app.py

app.py

@@ -88,16 +88,81 @@ class PromptManager:
             3. Mérito
             4. Análise das provas
             5. Pedidos
+            """,
+            
+            "resposta_acusacao": f"""
+            Atue como um advogado criminalista e gere uma resposta à acusação:
+            
+            DADOS DO PROCESSO:
+            Cliente: {context.get('client_name')}
+            Processo: {context.get('process_number')}
+            Tribunal: {context.get('court')}
+            
+            FATOS:
+            {context.get('facts')}
+            
+            TESE DEFENSIVA:
+            {context.get('legal_basis')}
+            
+            Elabore uma resposta à acusação completa, incluindo:
+            1. Preliminares
+            2. Mérito
+            3. Requerimentos probatórios
+            4. Teses defensivas
+            5. Pedidos
+            """,
+            
+            "recurso_sentido_estrito": f"""
+            Atue como um advogado criminalista e gere um recurso em sentido estrito:
+            
+            DADOS DO PROCESSO:
+            Cliente: {context.get('client_name')}
+            Processo: {context.get('process_number')}
+            Tribunal: {context.get('court')}
+            
+            FATOS:
+            {context.get('facts')}
+            
+            FUNDAMENTOS RECURSAIS:
+            {context.get('legal_basis')}
+            
+            Elabore um recurso em sentido estrito completo, incluindo:
+            1. Tempestividade
+            2. Cabimento
+            3. Fundamentação
+            4. Pedidos
+            """,
+            
+            "apelacao_criminal": f"""
+            Atue como um advogado criminalista e gere uma apelação criminal:
+            
+            DADOS DO PROCESSO:
+            Cliente: {context.get('client_name')}
+            Processo: {context.get('process_number')}
+            Tribunal: {context.get('court')}
+            
+            FATOS E SENTENÇA:
+            {context.get('facts')}
+            
+            RAZÕES RECURSAIS:
+            {context.get('legal_basis')}
+            
+            Elabore uma apelação criminal completa, incluindo:
+            1. Tempestividade
+            2. Preliminares
+            3. Mérito recursal
+            4. Pedidos
             """
         }
         
         return prompts.get(doc_type, "Tipo de documento não suportado")
 
 class DocumentGenerator:
-    """Gerencia a geração de documentos usando a API do LLama"""
+    """Gerencia a geração de documentos usando modelo Mistral público"""
     
     def __init__(self):
-        self.client = Client("ysharma/Chat_with_Meta_llama3_8b")
+        # Usando modelo Mistral público
+        self.client = Client("HuggingFaceH4/zephyr-7b-beta")
         self.prompt_manager = PromptManager()
     
     def generate(self, doc_type: str, context: Dict[str, str]) -> str:
@@ -105,18 +170,28 @@ class DocumentGenerator:
         try:
             prompt = self.prompt_manager.get_base_prompt(doc_type, context)
             
+            # Configuração específica para o modelo
             result = self.client.predict(
-                message=prompt,
-                request=0.85,  # temperatura mais baixa para maior consistência
-                param_3=3072,  # tokens aumentados para documentos longos
-                api_name="/chat"
+                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:
-            logger.error(f"Erro na geração do documento: {str(e)}")
-            return f"Erro na geração do documento: {str(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}"
     
     def _format_output(self, text: str) -> str:
         """Formata o texto gerado para melhor apresentação"""
@@ -135,7 +210,7 @@ class WebInterface:
         with gr.Blocks(theme=gr.themes.Soft()) as self.app:
             gr.Markdown("""
             # Gerador de Documentos Criminais
-            ### Powered by LLM Technology
+            ### Powered by AI Technology
             """)
             
             with gr.Row():
@@ -188,6 +263,12 @@ class WebInterface:
                             placeholder="Indique os fundamentos legais principais..."
                         )
                     
+                    # Status e botões
+                    status = gr.Textbox(
+                        label="Status",
+                        interactive=False
+                    )
+                    
                     generate_btn = gr.Button(
                         "Gerar Documento",
                         variant="primary"
@@ -226,7 +307,7 @@ class WebInterface:
                     doc_type, client_name, process_number,
                     court, jurisdiction, facts, legal_basis
                 ],
-                outputs=output
+                outputs=[output, status]
             )
     
     def _generate_document(
@@ -234,31 +315,46 @@ class WebInterface:
         process_number: str, court: str,
         jurisdiction: str, facts: str,
         legal_basis: str
-    ) -> str:
+    ) -> tuple:
         """Gera o documento com os parâmetros fornecidos"""
         
-        # Validação básica
-        if not all([client_name, process_number, facts, legal_basis]):
-            return "Erro: Todos os campos obrigatórios devem ser preenchidos"
-        
-        context = {
-            "client_name": client_name,
-            "process_number": process_number,
-            "court": court,
-            "jurisdiction": jurisdiction,
-            "facts": facts,
-            "legal_basis": legal_basis
-        }
-        
-        return self.generator.generate(
-            doc_type.lower().replace(" ", "_"),
-            context
-        )
+        try:
+            # Validação básica
+            if not all([client_name, process_number, facts, legal_basis]):
+                return "Erro: Todos os campos obrigatórios devem ser preenchidos", "⚠️ Preencha todos os campos obrigatórios"
+            
+            context = {
+                "client_name": client_name,
+                "process_number": process_number,
+                "court": court,
+                "jurisdiction": jurisdiction,
+                "facts": facts,
+                "legal_basis": legal_basis
+            }
+            
+            # Atualiza status
+            yield "", "⏳ Gerando documento..."
+            
+            # Gera documento
+            result = self.generator.generate(
+                doc_type.lower().replace(" ", "_"),
+                context
+            )
+            
+            return result, "✅ Documento gerado com sucesso!"
+            
+        except Exception as e:
+            logger.error(f"Erro: {str(e)}")
+            return "", f"❌ Erro: {str(e)}"
     
     def launch(self):
         """Inicia a interface web"""
         self.app.launch(share=True)
 
 if __name__ == "__main__":
+    # Arquivo requirements.txt necessário:
+    # gradio==4.19.2
+    # gradio_client==0.10.1
+    
     interface = WebInterface()
     interface.launch()