Update app.py

app.py

@@ -4,54 +4,65 @@ import os
 from datetime import datetime
 
 DOCUMENT_TYPES = {
-    "resposta_acusacao": "Resposta à Acusação",
-    "habeas_corpus": "Habeas Corpus", 
-    "revogacao_preventiva": "Revogação de Prisão Preventiva",
-    "liberdade_provisoria": "Liberdade Provisória",
-    "relaxamento_prisao": "Relaxamento de Prisão",
-    "apelacao": "Recurso de Apelação",
-    "embargos": "Embargos de Declaração"
+    "resposta_acusacao": "Resposta à Acusação (Art. 396-A do CPP)",
+    "habeas_corpus": "Habeas Corpus (Art. 647 do CPP)", 
+    "revogacao_preventiva": "Revogação de Prisão Preventiva (Art. 316 do CPP)",
+    "liberdade_provisoria": "Liberdade Provisória (Art. 321 do CPP)",
+    "relaxamento_prisao": "Relaxamento de Prisão (Art. 5º, LXV, CF)",
+    "apelacao": "Recurso de Apelação (Art. 593 do CPP)",
+    "embargos": "Embargos de Declaração (Art. 382 do CPP)"
 }
 
-DEFAULT_SECTIONS = """AUTOS Nº [NUMERO_PROCESSO]
+TEMPLATE_PENAL = """EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} VARA CRIMINAL DA COMARCA DE {comarca}/{estado}
+
+AUTOS Nº {num_processo}
+
+{tipo_peca}
 
 I - QUALIFICAÇÃO DAS PARTES
-[Detalhes do acusado e seu advogado]
+{nome}, brasileiro(a), {idade} anos, {estado_civil}, {profissao}, RG nº {rg}, CPF nº {cpf}, residente e domiciliado à {endereco}, {cidade}/{estado}, vem, por seu advogado que esta subscreve, com fundamento na legislação aplicável, apresentar {tipo_documento}, pelos fatos e fundamentos a seguir expostos.
 
 II - FUNDAMENTOS JURÍDICOS
-1. [Base legal específica do caso]
-2. [Elementos necessários para configuração do crime]
+1. {fundamentos}
+2. {elementos_tipo_penal}
 
 III - JURISPRUDÊNCIA APLICÁVEL
-1. [Precedentes favoráveis]
-2. [Entendimentos dos tribunais]
+1. {jurisprudencia_1}
+2. {jurisprudencia_2}
 
 IV - DOUTRINA APLICÁVEL
-1. [Citações doutrinárias relevantes]
-2. [Posicionamentos doutrinários]
+1. {doutrina_1}
+2. {doutrina_2}
 
 V - PRELIMINARES
-1. [Questões processuais]
-2. [Nulidades e vícios]
+1. {preliminar_1}
+2. {preliminar_2}
 
 VI - MÉRITO
-1. [Argumentos principais]
-2. [Provas e contraprovas]
+1. {merito_1}
+2. {merito_2}
+3. {merito_3}
 
 VII - PEDIDOS
-1. [Pedido principal]
-2. [Pedidos subsidiários]
-3. [Provas a produzir]
+1. PRELIMINARMENTE:
+   a) {pedido_preliminar_1}
+   b) {pedido_preliminar_2}
+
+2. NO MÉRITO:
+   a) {pedido_merito_1}
+   b) {pedido_merito_2}
 
-[CIDADE], [DATA]
+3. PROVAS A PRODUZIR:
+   {provas}
 
-[ADVOGADO]
-OAB/[ESTADO] [NÚMERO]
+{cidade}, {data}
+
+{advogado}
+OAB/{estado} {oab}
 
 ROL DE TESTEMUNHAS:
-1. [Nome e qualificação]
-2. [Nome e qualificação]
-3. [Nome e qualificação]"""
+{testemunhas}
+"""
 
 class LegalDocGenerator:
     def __init__(self):
@@ -60,117 +71,144 @@ class LegalDocGenerator:
             raise EnvironmentError("HF_API_KEY não configurada")
         self.client = InferenceClient(api_key=self.api_key)
 
-    def _build_prompt(self, doc_type: str, caso: dict) -> str:
-        return f"""Atue como um advogado criminalista experiente gerando uma peça de {DOCUMENT_TYPES[doc_type]}.
-
-CASO CONCRETO:
-- Acusado: {caso['cliente']}
-- Fatos: {caso['fatos']}
-- Comarca: {caso['comarca']}
-- Vara: {caso['vara']} Vara Criminal
-- OAB: {caso['oab']}
-
-INSTRUÇÕES ESPECÍFICAS:
-1. Use linguagem jurídica formal
-2. Cite jurisprudência real e atualizada
-3. Fundamente com doutrina pertinente
-4. Destaque elementos favoráveis ao acusado
-5. Mantenha estrutura com numeração romana para seções principais
-6. Inclua elementos de prova mencionados nos fatos
-7. Desenvolva argumentos técnicos para cada pedido
-8. Mantenha formatação profissional
-9. Use artigos corretos do Código Penal e CPP
-
-ESTRUTURA BASE:
-{DEFAULT_SECTIONS}"""
-
-    def generate(self, doc_type: str, caso: dict) -> str:
+    def _format_date(self) -> str:
+        return datetime.now().strftime("%d de %B de %Y")
+
+    def _build_prompt(self, doc_type: str, dados: dict) -> str:
+        # Instruções base para o modelo
+        system_prompt = """Você é um advogado criminalista especialista com anos de experiência.
+Ao gerar peças processuais:
+1. Use linguagem jurídica formal e técnica
+2. Cite jurisprudência atual e relevante
+3. Faça referências doutrinárias precisas
+4. Desenvolva argumentos específicos para o caso
+5. Mantenha coerência em toda a peça
+6. Adapte os argumentos ao tipo de documento
+7. Use legislação atualizada e pertinente"""
+
+        # Prompt específico para o caso
+        case_prompt = f"""TIPO DE PEÇA: {DOCUMENT_TYPES[doc_type]}
+
+DADOS DO CASO:
+Cliente: {dados['nome']}
+Fatos: {dados['fatos']}
+Comarca: {dados['comarca']}
+Vara: {dados['vara']}
+
+ELEMENTOS ESPECÍFICOS A INCLUIR:
+1. Fundamentos específicos sobre {doc_type}
+2. Jurisprudência favorável recente
+3. Argumentos técnicos sobre {doc_type}
+4. Pedidos adequados ao tipo de peça
+5. Provas pertinentes ao caso
+
+BASE PARA GERAÇÃO:
+{TEMPLATE_PENAL}"""
+
+        return system_prompt, case_prompt
+
+    def generate(self, doc_type: str, dados: dict) -> str:
         try:
-            prompt = self._build_prompt(doc_type, caso)
+            system_prompt, case_prompt = self._build_prompt(doc_type, dados)
+            dados['data'] = self._format_date()
+            
             response = self.client.chat.completions.create(
                 model="mistralai/Mistral-7B-Instruct-v0.3",
                 messages=[
-                    {"role": "system", "content": "Você é um advogado criminalista especialista em gerar peças jurídicas técnicas e bem fundamentadas."},
-                    {"role": "user", "content": prompt}
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": case_prompt}
                 ],
                 max_tokens=2500,
                 temperature=0.7
             )
+            
             return response.choices[0].message["content"]
         except Exception as e:
             return f"Erro na geração: {str(e)}"
 
 def create_interface():
-    def process(tipo, fatos, comarca, vara, cliente, oab, num_processo="", testemunhas=""):
-        caso = {
-            "cliente": cliente,
+    def process_submit(doc_type, fatos, comarca, vara, cliente, oab, proc_num, 
+                      idade, estado_civil, profissao, rg, cpf, endereco, testemunhas):
+        dados = {
+            "tipo_peca": DOCUMENT_TYPES[doc_type],
+            "nome": cliente,
             "fatos": fatos,
             "comarca": comarca,
             "vara": vara,
             "oab": oab,
-            "num_processo": num_processo,
+            "num_processo": proc_num,
+            "idade": idade,
+            "estado_civil": estado_civil,
+            "profissao": profissao,
+            "rg": rg,
+            "cpf": cpf,
+            "endereco": endereco,
             "testemunhas": testemunhas
         }
-        return LegalDocGenerator().generate(tipo, caso)
+        return LegalDocGenerator().generate(doc_type, dados)
 
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown("# Sistema Criminal.AI - Gerador de Peças Jurídicas")
-        
-        with gr.Row():
-            tipo = gr.Dropdown(
-                choices=list(DOCUMENT_TYPES.keys()),
-                value="resposta_acusacao",
-                label="Tipo de Peça"
-            )
-            num_processo = gr.Textbox(
-                label="Número do Processo",
-                placeholder="Ex: 0000000-00.0000.0.00.0000"
-            )
-
-        with gr.Row():
-            with gr.Column():
-                comarca = gr.Textbox(label="Comarca", placeholder="Ex: São Paulo")
-                vara = gr.Textbox(label="Vara", placeholder="Ex: 1ª")
-            with gr.Column():
-                cliente = gr.Textbox(label="Nome do Cliente", placeholder="Ex: João da Silva")
-                oab = gr.Textbox(label="OAB", placeholder="Ex: SP123456")
-        
-        fatos = gr.Textbox(
-            label="Descrição dos Fatos",
-            placeholder="Descreva detalhadamente a situação do cliente...",
-            lines=5
-        )
+        gr.Markdown("# Sistema de Geração de Peças Criminais v2.0")
         
-        testemunhas = gr.Textbox(
-            label="Rol de Testemunhas (opcional)",
-            placeholder="Nome e qualificação das testemunhas, uma por linha",
-            lines=3
-        )
+        with gr.Tab("Dados do Processo"):
+            with gr.Row():
+                tipo = gr.Dropdown(choices=list(DOCUMENT_TYPES.keys()), 
+                                 value="resposta_acusacao",
+                                 label="Tipo de Peça")
+                proc_num = gr.Textbox(label="Número do Processo",
+                                    placeholder="0000000-00.0000.0.00.0000")
+            
+            with gr.Row():
+                comarca = gr.Textbox(label="Comarca")
+                vara = gr.Textbox(label="Vara Criminal")
+
+        with gr.Tab("Dados do Cliente"):
+            with gr.Row():
+                cliente = gr.Textbox(label="Nome Completo")
+                idade = gr.Number(label="Idade")
+                estado_civil = gr.Dropdown(label="Estado Civil",
+                                         choices=["Solteiro(a)", "Casado(a)", "Divorciado(a)", "Viúvo(a)"])
+            
+            with gr.Row():
+                profissao = gr.Textbox(label="Profissão")
+                rg = gr.Textbox(label="RG")
+                cpf = gr.Textbox(label="CPF")
+            
+            endereco = gr.Textbox(label="Endereço Completo")
+
+        with gr.Tab("Dados do Caso"):
+            fatos = gr.TextArea(label="Descrição dos Fatos",
+                              placeholder="Descreva detalhadamente a situação...",
+                              lines=5)
+            
+            testemunhas = gr.TextArea(label="Rol de Testemunhas",
+                                    placeholder="Nome completo e qualificação de cada testemunha",
+                                    lines=3)
+
+        with gr.Tab("Dados do Advogado"):
+            oab = gr.Textbox(label="Número da OAB com Estado")
+
+        doc_output = gr.TextArea(label="Documento Gerado",
+                               lines=30,
+                               show_copy_button=True)
         
-        output = gr.TextArea(
-            label="Peça Gerada",
-            lines=30,
-            show_copy_button=True
+        submit_btn = gr.Button("Gerar Peça Processual", variant="primary")
+        submit_btn.click(
+            fn=process_submit,
+            inputs=[tipo, fatos, comarca, vara, cliente, oab, proc_num,
+                   idade, estado_civil, profissao, rg, cpf, endereco, testemunhas],
+            outputs=doc_output
         )
 
         gr.Examples([
-            [
-                "resposta_acusacao",
-                "Cliente preso em flagrante por furto de celular iPhone 13. É primário, tem residência fixa e trabalho lícito comprovado. A vítima não o reconheceu com certeza.",
-                "São Paulo",
-                "1ª",
-                "João Silva",
-                "SP123456",
-                "0001234-56.2024.8.26.0000",
-                "1. Maria Oliveira - Testemunha ocular\n2. José Santos - Empregador do acusado"
-            ],
-        ], inputs=[tipo, fatos, comarca, vara, cliente, oab, num_processo, testemunhas])
-
-        gr.Button("Gerar Peça").click(
-            fn=process,
-            inputs=[tipo, fatos, comarca, vara, cliente, oab, num_processo, testemunhas],
-            outputs=output
-        )
+            ["resposta_acusacao", 
+             "Cliente preso em flagrante por furto de celular iPhone 13. É primário, tem residência fixa e trabalho lícito comprovado. A vítima não o reconheceu com certeza.",
+             "São Paulo", "1ª", "João Silva", "SP123456", "0001234-56.2024.8.26.0000",
+             35, "Casado(a)", "Comerciante", "12.345.678-9", "123.456.789-00",
+             "Rua das Flores, 123, Centro", 
+             "1. Maria Oliveira - Testemunha ocular\n2. José Santos - Empregador do acusado"]
+        ], inputs=[tipo, fatos, comarca, vara, cliente, oab, proc_num,
+                  idade, estado_civil, profissao, rg, cpf, endereco, testemunhas])
 
     return demo