Update app.py

app.py

@@ -10,6 +10,50 @@ class DocumentGenerator:
             raise EnvironmentError("HF_API_KEY não configurada")
         self.client = InferenceClient(api_key=self.api_key)
         
+        self.templates = {
+            "denuncia": {
+                "system": """Você é um promotor de justiça experiente, especializado em redigir denúncias criminais.
+                Seu objetivo é gerar denúncias precisas e tecnicamente corretas.""",
+                "sections": {
+                    "cabecalho": "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} VARA CRIMINAL DA COMARCA DE {comarca}",
+                    "qualificacao": """O MINISTÉRIO PÚBLICO DO ESTADO DE {estado}, por seu Promotor de Justiça que esta subscreve, no uso de suas atribuições legais, vem perante Vossa Excelência oferecer DENÚNCIA em face de:
+
+{nome_reu}, {qualificacao_reu}""",
+                    "narrativa": "{narrativa}",
+                    "tipificacao": """Assim agindo, {nome_reu} incorreu nas sanções do artigo {artigo} do Código Penal.""",
+                    "pedido": """Ante o exposto, o Ministério Público requer:
+
+1. O recebimento e autuação da presente denúncia;
+2. A citação do denunciado para responder à acusação por escrito, no prazo de 10 (dez) dias;
+3. A intimação das testemunhas abaixo arroladas;
+4. Ao final, a condenação do denunciado nas penas do artigo {artigo} do Código Penal.
+
+Rol de Testemunhas:
+{testemunhas}
+
+{comarca}, {data}
+
+Promotor(a) de Justiça"""
+                }
+            },
+            "alegacoes_finais": {
+                "system": """Você é um promotor de justiça experiente, especializado em redigir alegações finais criminais.
+                Seu objetivo é gerar alegações finais precisas e tecnicamente corretas.""",
+                "sections": {
+                    "cabecalho": """EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} VARA CRIMINAL DA COMARCA DE {comarca}
+
+Processo nº: {numero_processo}""",
+                    "relatorio": "{narrativa_processual}",
+                    "fundamentacao": "{fundamentacao}",
+                    "pedido": """Ante o exposto, o Ministério Público requer a {pedido_final}.
+
+{comarca}, {data}
+
+Promotor(a) de Justiça"""
+                }
+            }
+        }
+
     def generate_section(self, messages: list, section: str) -> str:
         try:
             response = self.client.chat.completions.create(
@@ -23,21 +67,30 @@ class DocumentGenerator:
             raise Exception(f"Erro na geração da seção {section}: {str(e)}")
 
     def gerar_documento(self, tipo_documento: str, dados: dict) -> str:
+        template = self.templates.get(tipo_documento)
+        if not template:
+            return "Tipo de documento não encontrado"
+            
+        dados['data'] = datetime.now().strftime("%d/%m/%Y")
+        
         messages = [
-            {"role": "system", "content": "Você é um assistente especializado em peças jurídicas criminais."},
+            {"role": "system", "content": template["system"]},
             {"role": "user", "content": f"Dados para geração de {tipo_documento}:\n{str(dados)}"}
         ]
         
-        if tipo_documento == "denuncia":
-            sections = ["qualificacao", "narrativa", "tipificacao", "pedido"]
-        else:
-            sections = ["relatorio", "fundamentacao", "pedido"]
-            
         documento = ""
-        for section in sections:
-            documento += self.generate_section(messages, section) + "\n\n"
-            
-        dados['data'] = datetime.now().strftime("%d/%m/%Y")
+        for section_name, section_template in template["sections"].items():
+            try:
+                if any(key in section_template for key in dados.keys()):
+                    # Se o template tem placeholders, usa os dados fornecidos
+                    section_content = section_template.format(**dados)
+                else:
+                    # Senão, gera conteúdo via Mistral
+                    section_content = self.generate_section(messages, section_name)
+                documento += section_content + "\n\n"
+            except Exception as e:
+                documento += f"[Erro na seção {section_name}: {str(e)}]\n\n"
+                
         return documento
 
 def interface():
@@ -63,7 +116,9 @@ def interface():
         return gerador.gerar_documento(tipo_doc, dados)
 
     with gr.Blocks(title="Gerador de Peças Jurídicas Criminais") as demo:
-        gr.Markdown("# Gerador de Peças Jurídicas Criminais")
+        gr.Markdown("""# Gerador de Peças Jurídicas Criminais
+        
+Preencha os campos abaixo para gerar automaticamente denúncias e alegações finais criminais.""")
         
         with gr.Row():
             tipo_documento = gr.Radio(
@@ -74,41 +129,62 @@ def interface():
             
         with gr.Row():
             with gr.Column():
-                comarca = gr.Textbox(label="Comarca")
-                vara = gr.Textbox(label="Vara")
-                estado = gr.Textbox(label="Estado")
-                nome_reu = gr.Textbox(label="Nome do Réu")
+                comarca = gr.Textbox(
+                    label="Comarca",
+                    placeholder="São Paulo"
+                )
+                vara = gr.Textbox(
+                    label="Vara",
+                    placeholder="1ª"
+                )
+                estado = gr.Textbox(
+                    label="Estado",
+                    placeholder="São Paulo"
+                )
+                nome_reu = gr.Textbox(
+                    label="Nome do Réu",
+                    placeholder="João da Silva"
+                )
                 qualificacao = gr.Textbox(
                     label="Qualificação do Réu",
-                    placeholder="brasileiro, casado, etc..."
+                    placeholder="brasileiro, casado, comerciante, RG 12.345.678-9 SSP/SP"
                 )
                 
             with gr.Column():
                 narrativa = gr.Textbox(
                     label="Narrativa dos Fatos",
-                    lines=5
+                    lines=5,
+                    placeholder="No dia 10 de janeiro de 2024, por volta das 14h..."
+                )
+                artigo = gr.Textbox(
+                    label="Artigo(s) do CP",
+                    placeholder="155, caput"    
                 )
-                artigo = gr.Textbox(label="Artigo(s) do CP")
                 testemunhas = gr.Textbox(
                     label="Testemunhas",
                     lines=3,
-                    placeholder="1. Nome da testemunha\n2. Nome da testemunha"
+                    placeholder="1. Maria Santos - Policial Militar\n2. José Oliveira - Testemunha presencial"
                 )
 
         with gr.Row(visible=False) as campos_alegacoes:
             with gr.Column():
-                numero_processo = gr.Textbox(label="Número do Processo")
+                numero_processo = gr.Textbox(
+                    label="Número do Processo",
+                    placeholder="0000000-00.0000.0.00.0000"
+                )
                 narrativa_processual = gr.Textbox(
                     label="Narrativa Processual",
-                    lines=5
+                    lines=5,
+                    placeholder="Trata-se de ação penal..."
                 )
                 fundamentacao = gr.Textbox(
                     label="Fundamentação",
-                    lines=5
+                    lines=5,
+                    placeholder="A materialidade delitiva está comprovada..."
                 )
                 pedido_final = gr.Textbox(
                     label="Pedido Final",
-                    placeholder="condenação do réu nas sanções do art. X..."
+                    placeholder="condenação do réu nas sanções do art. 155, caput, do Código Penal"
                 )
 
         def atualizar_campos(escolha):
@@ -117,9 +193,45 @@ def interface():
         tipo_documento.change(atualizar_campos, tipo_documento, campos_alegacoes)
         
         with gr.Row():
-            gerar_btn = gr.Button("Gerar Documento")
+            gerar_btn = gr.Button("Gerar Documento", variant="primary")
+            limpar_btn = gr.Button("Limpar Campos")
             
-        saida = gr.Textbox(label="Documento Gerado", lines=10)
+        saida = gr.Textbox(
+            label="Documento Gerado",
+            lines=15,
+            show_copy_button=True
+        )
+
+        # Exemplo de dados
+        exemplo_denuncia = {
+            "tipo_doc": "denuncia",
+            "comarca": "São Paulo",
+            "vara": "1ª",
+            "estado": "São Paulo",
+            "nome_reu": "João da Silva",
+            "qualificacao": "brasileiro, casado, comerciante, RG 12.345.678-9 SSP/SP",
+            "narrativa": "No dia 10 de janeiro de 2024, por volta das 14h, na Rua das Flores, 123, Centro, nesta cidade, o denunciado subtraiu para si um telefone celular iPhone 13, avaliado em R$ 5.000,00, pertencente à vítima Maria Santos.",
+            "artigo": "155, caput",
+            "testemunhas": "1. Maria Santos - Vítima\n2. Pedro Oliveira - Policial Militar\n3. Ana Costa - Testemunha presencial",
+            "numero_processo": "",
+            "narrativa_processual": "",
+            "fundamentacao": "",
+            "pedido_final": ""
+        }
+        
+        gr.Examples(
+            examples=[[exemplo_denuncia[k] for k in [
+                "tipo_doc", "comarca", "vara", "estado", "nome_reu", "qualificacao",
+                "narrativa", "artigo", "testemunhas", "numero_processo",
+                "narrativa_processual", "fundamentacao", "pedido_final"
+            ]]],
+            inputs=[
+                tipo_documento, comarca, vara, estado, nome_reu, qualificacao,
+                narrativa, artigo, testemunhas, numero_processo,
+                narrativa_processual, fundamentacao, pedido_final
+            ],
+            label="Exemplos"
+        )
         
         gerar_btn.click(
             processar,
@@ -130,6 +242,19 @@ def interface():
             ],
             outputs=saida
         )
+        
+        def limpar():
+            return [""] * 13
+            
+        limpar_btn.click(
+            limpar,
+            inputs=None,
+            outputs=[
+                tipo_documento, comarca, vara, estado, nome_reu, qualificacao,
+                narrativa, artigo, testemunhas, numero_processo,
+                narrativa_processual, fundamentacao, pedido_final
+            ]
+        )
     
     return demo