Update app.py

app.py

@@ -1,218 +1,274 @@
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
+from datetime import datetime
+from typing import Dict, Optional
 
-# Obtendo a API Key da variável de ambiente
-api_key = os.getenv("HF_API_KEY")
-if not api_key:
-    raise EnvironmentError("A variável de ambiente 'HF_API_KEY' não está configurada.")
-
-# Criação do cliente com a API Key
-client = InferenceClient(api_key=api_key)
-
-# Função para gerar a peça jurídica em português do Brasil
-def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
-    if not informacoes_caso or not tipo_peca:
-        return "Por favor, forneça todas as informações necessárias."
-
-    templates = {
-        "habeas corpus": {
-            "estrutura": [
-                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA [VARA] DA [COMARCA]",
-                "QUALIFICAÇÃO DO PACIENTE\n"
-                "O acusado, [Nome do Cliente], é um cidadão brasileiro, nascido em [Data de Nascimento] no município de [Local de Nascimento], "
-                "titular do documento de identidade [Tipo de Documento de Identidade] no número [Número do Documento de Identidade], "
-                "efetivo domiciliado na [Endereço de Domicílio do Cliente].",
-                "DOS FATOS\n"
-                "A Polícia de [Local do Crime] detém o acusado desde [Data da Detenção], por motivos de suposto furto qualificado, "
-                "cometido no dia [Data do Crime] às [Horas do Crime], no local de [Local do Crime]. "
-                "Entretanto, há provas documentais e testemunha ocular que comprovam a presença do acusado na cidade distante [Local Distante do Crime] no momento do crime.",
-                "DO DIREITO\n"
-                "O artigo 5º LXVIII da Constituição Federal dispõe que 'toda pessoa pode ser presa quando há fundamentos razoáveis para acreditar que ela cometeu ou está prestes a cometer um crime'. "
-                "O Código de Processo Penal estabelece, nos artigos 647 e 648, que 'a prisão não pode ser imposta a menos que haja fundamentos razoáveis...'",
-                "DO CONSTRANGIMENTO ILEGAL\n"
-                "A prisão do acusado, sem a devida fundamentação, caracteriza constrangimento ilegal, uma vez que há evidências de sua inocência.",
-                "DA MEDIDA LIMINAR\n"
-                "Requer-se a concessão de medida liminar para a liberdade do paciente.",
-                "DOS PEDIDOS\n"
-                "Diante do exposto, requer-se a concessão do habeas corpus.",
-                "PROVAS E DOCUMENTOS\n"
-                "Anexamos documentos que comprovam a inocência do acusado.",
-                "FECHAMENTO\n"
-                "Nestes termos, pede deferimento."
-            ],
-            "max_tokens": 500
-        },
-        # Outros templates podem ser adicionados aqui
-    }
-
-    template = templates.get(tipo_peca.lower())
-    if not template:
-        return "Tipo de peça não suportado."
-
-    # Preencher os campos da peça jurídica com base nas informações do caso
-    informacoes = {
-        "Nome do Cliente": "João da Silva",  # Exemplo
-        "Data de Nascimento": "01/01/1990",  # Exemplo
-        "Local de Nascimento": "São Paulo",
-        "Tipo de Documento de Identidade": "RG",
-        "Número do Documento de Identidade": "123456789",
-        "Endereço de Domicílio do Cliente": "Rua Exemplo, 123, São Paulo - SP",
-        "Local do Crime": "São Paulo",
-        "Data da Detenção": "15/12/2023",
-        "Data do Crime": "14/12/2023",
-        "Horas do Crime": "14:00",
-        "Local Distante do Crime": "Campinas",
-        "VARA": "1ª Vara Criminal",  # Exemplo
-        "COMARCA": "São Paulo"  # Exemplo
-    }
-
-    # Montar o texto da peça jurídica
-    texto_final = "\n\n".join([
-        linha.replace(placeholder, informacoes[placeholder])
-        for linha in template["estrutura"]
-        for placeholder in informacoes.keys()
-        if placeholder in linha
-    ])
-
-    # Realizar a requisição única para aprimoramento
-    try:
-        mensagens = [
-            {
-                "role": "system",
-                "content": (
-                    f"Como advogado criminalista, redija um {tipo_peca} técnica seguindo:\n"
-                    f"1. Estrutura formal\n"
-                    f"2. Linguagem jurídica\n"
-                    f"3. Fundamentação precisa\n"
-                    f"4. Formatação correta\n"
-                    f"5. Considere as informações do caso: {informacoes_caso}"
-                )
-            },
-            {
-                "role": "user",
-                "content": texto_final
-            }
-        ]
+class LegalDocumentGenerator:
+    def __init__(self):
+        self.api_key = os.getenv("HF_API_KEY")
+        if not self.api_key:
+            raise EnvironmentError("HF_API_KEY não configurada")
+        self.client = InferenceClient(api_key=self.api_key)
+        
+    def generate_section(self, messages: list, section: str) -> str:
+        try:
+            response = self.client.chat.completions.create(
+                model="mistralai/Mistral-7B-Instruct-v0.3",
+                messages=[*messages, {"role": "user", "content": f"Gere a seção {section}"}],
+                max_tokens=500,
+                temperature=0.3
+            )
+            return response.choices[0].message["content"]
+        except Exception as e:
+            raise Exception(f"Erro na geração da seção {section}: {str(e)}")
 
-        resposta_final = client.chat.completions.create(
-            model="mistralai/Mistral-7B-Instruct-v0.3",
-            messages=mensagens,
-            max_tokens=template['max_tokens'],
-            temperature=0.7
-        )
+    def generate_document(self, case_info: str, doc_type: str, client_data: Dict) -> str:
+        if not all([case_info, doc_type, client_data]):
+            return "Erro: Dados incompletos"
 
-        return resposta_final.choices[0].message["content"]
+        templates = {
+            "habeas corpus": self._get_habeas_corpus_template(),
+            "defesa prévia": self._get_defesa_previa_template(),
+            "apelação criminal": self._get_apelacao_template(),
+            "petição inicial": self._get_peticao_inicial_template()
+        }
+
+        template = templates.get(doc_type.lower())
+        if not template:
+            return "Erro: Tipo de documento não suportado"
+
+        try:
+            messages = [
+                {
+                    "role": "system",
+                    "content": (
+                        "Você é um advogado criminalista especializado. Gere uma peça jurídica observando:\n"
+                        "1. Formalidade e técnica jurídica\n"
+                        "2. Fundamentação legal precisa e atualizada\n"
+                        "3. Jurisprudência relevante\n"
+                        "4. Adequação ao caso concreto\n"
+                        f"Tipo de peça: {doc_type}\n"
+                        f"Informações do caso: {case_info}"
+                    )
+                }
+            ]
+
+            sections = {}
+            for section in template["required_sections"]:
+                sections[section] = self.generate_section(messages, section)
+
+            document = template["structure"].format(
+                **sections,
+                **client_data,
+                data=datetime.now().strftime("%d/%m/%Y")
+            )
+
+            return document
+
+        except Exception as e:
+            return f"Erro na geração do documento: {str(e)}"
+
+    def _get_habeas_corpus_template(self):
+        return {
+            "structure": (
+                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} DA {comarca}\n\n"
+                "HABEAS CORPUS COM PEDIDO LIMINAR\n\n"
+                "QUALIFICAÇÃO DO PACIENTE\n{qualificacao}\n\n"
+                "DOS FATOS\n{fatos}\n\n"
+                "DO DIREITO\n{fundamentacao}\n\n"
+                "DO CONSTRANGIMENTO ILEGAL\n{constrangimento}\n\n"
+                "DA MEDIDA LIMINAR\n{liminar}\n\n"
+                "DOS PEDIDOS\n{pedidos}\n\n"
+                "PROVAS E DOCUMENTOS\n{provas}\n\n"
+                "{comarca}, {data}\n\n"
+                "{advogado}\n"
+                "OAB/{estado} {numero_oab}"
+            ),
+            "required_sections": [
+                "qualificacao", "fatos", "fundamentacao", 
+                "constrangimento", "liminar", "pedidos", "provas"
+            ]
+        }
+
+    def _get_defesa_previa_template(self):
+        return {
+            "structure": (
+                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} DA {comarca}\n\n"
+                "DEFESA PRÉVIA\n\n"
+                "Autos nº {numero_processo}\n\n"
+                "QUALIFICAÇÃO DO ACUSADO\n{qualificacao}\n\n"
+                "DOS FATOS\n{fatos}\n\n"
+                "DO DIREITO\n{fundamentacao}\n\n"
+                "DAS PRELIMINARES\n{preliminares}\n\n"
+                "DO MÉRITO\n{merito}\n\n"
+                "DAS PROVAS\n{provas}\n\n"
+                "DOS PEDIDOS\n{pedidos}\n\n"
+                "{comarca}, {data}\n\n"
+                "{advogado}\n"
+                "OAB/{estado} {numero_oab}"
+            ),
+            "required_sections": [
+                "qualificacao", "fatos", "fundamentacao", "preliminares",
+                "merito", "provas", "pedidos"
+            ]
+        }
+
+    def _get_apelacao_template(self):
+        return {
+            "structure": (
+                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) DESEMBARGADOR(A) RELATOR(A) DO TRIBUNAL DE JUSTIÇA DO ESTADO DE {estado}\n\n"
+                "RAZÕES DE APELAÇÃO\n\n"
+                "Autos nº {numero_processo}\n\n"
+                "QUALIFICAÇÃO DO APELANTE\n{qualificacao}\n\n"
+                "DOS FATOS\n{fatos}\n\n"
+                "DAS PRELIMINARES\n{preliminares}\n\n"
+                "DO MÉRITO\n{merito}\n\n"
+                "DO DIREITO\n{fundamentacao}\n\n"
+                "DOS PEDIDOS\n{pedidos}\n\n"
+                "{comarca}, {data}\n\n"
+                "{advogado}\n"
+                "OAB/{estado} {numero_oab}"
+            ),
+            "required_sections": [
+                "qualificacao", "fatos", "preliminares", "merito",
+                "fundamentacao", "pedidos"
+            ]
+        }
+
+    def _get_peticao_inicial_template(self):
+        return {
+            "structure": (
+                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} DA {comarca}\n\n"
+                "PETIÇÃO INICIAL\n\n"
+                "QUALIFICAÇÃO DO AUTOR\n{qualificacao}\n\n"
+                "DOS FATOS\n{fatos}\n\n"
+                "DO DIREITO\n{fundamentacao}\n\n"
+                "DO PEDIDO LIMINAR\n{liminar}\n\n"
+                "DOS PEDIDOS\n{pedidos}\n\n"
+                "DAS PROVAS\n{provas}\n\n"
+                "DO VALOR DA CAUSA\n{valor_causa}\n\n"
+                "{comarca}, {data}\n\n"
+                "{advogado}\n"
+                "OAB/{estado} {numero_oab}"
+            ),
+            "required_sections": [
+                "qualificacao", "fatos", "fundamentacao", "liminar",
+                "pedidos", "provas", "valor_causa"
+            ]
+        }
+
+def create_interface():
+    generator = LegalDocumentGenerator()
 
-    except Exception as e:
-        return f"Erro ao gerar a peça: {str(e)}"
-        
-# Função para redefinir os campos
-def redefinir_campos():
-    return "", "", ""
-
-# Função para exportar a peça jurídica gerada
-def exportar_peca(peca_texto):
-    """
-    Exporta a peça jurídica gerada para um arquivo .txt.
-
-    Parâmetros:
-        peca_texto (str): Texto da peça jurídica gerada.
-    
-    Retorna:
-        str: Caminho do arquivo exportado.
-    """
-    arquivo_nome = "peca_juridica_gerada.txt"
-    with open(arquivo_nome, "w", encoding="utf-8") as arquivo:
-        arquivo.write(peca_texto)
-    return f"Arquivo salvo como: {arquivo_nome}"
-
-# Configuração do app Gradio com funcionalidades adicionais
-def gerar_interface_gradio():
-    """
-    Configura a interface do Gradio para gerar peças jurídicas com exemplos pré-definidos e funcionalidades adicionais.
-    """
     with gr.Blocks() as interface:
         gr.Markdown("# Gerador de Peças Jurídicas Criminais")
-        gr.Markdown(
-            "### Preencha os campos abaixo ou escolha um exemplo, e selecione o tipo de peça jurídica que deseja gerar."
-        )
         
-        with gr.Row():
-            entrada_caso = gr.Textbox(
+        with gr.Tab("Dados do Caso"):
+            case_info = gr.Textbox(
                 label="Informações do Caso",
-                placeholder="Descreva o caso detalhadamente ou selecione um exemplo abaixo.",
+                placeholder="Descreva o caso detalhadamente",
                 lines=5
             )
-            
-            exemplo_caso = gr.Dropdown(
-                label="Exemplos de Casos",
+            doc_type = gr.Dropdown(
+                label="Tipo de Peça",
                 choices=[
-                    "O cliente foi acusado injustamente de furto qualificado, mas há provas de que estava em outro local no momento do crime.",
-                    "O cliente foi preso em flagrante por porte de entorpecentes, mas os policiais não apresentaram o mandado de busca.",
-                    "O cliente foi acusado de agressão, mas existem testemunhas que comprovam sua ausência no local do ocorrido."
-                ],
-                value=None,
-                interactive=True
+                    "habeas corpus",
+                    "defesa prévia",
+                    "apelação criminal",
+                    "petição inicial"
+                ]
             )
-            
-            entrada_tipo = gr.Dropdown(
-                label="Tipo de Peça Jurídica",
-                choices=["habeas corpus", "petição inicial", "defesa prévia", "apelação criminal"],
-                value=None  # Define o valor inicial como nenhum
-            )
-        
-        def atualizar_caso(caso_exemplo):
-            """
-            Atualiza o campo de texto com o exemplo escolhido pelo usuário.
-            """
-            return caso_exemplo
-
-        exemplo_caso.change(
-            atualizar_caso, 
-            inputs=exemplo_caso, 
-            outputs=entrada_caso
-        )
-        
+
+        with gr.Tab("Dados do Cliente"):
+            with gr.Row():
+                client_name = gr.Textbox(label="Nome")
+                client_nationality = gr.Textbox(label="Nacionalidade")
+            with gr.Row():
+                client_profession = gr.Textbox(label="Profissão")
+                client_id = gr.Textbox(label="RG")
+            with gr.Row():
+                client_address = gr.Textbox(label="Endereço")
+                client_city = gr.Textbox(label="Cidade")
+            with gr.Row():
+                client_state = gr.Textbox(label="Estado")
+                client_zip = gr.Textbox(label="CEP")
+
+        with gr.Tab("Dados Processuais"):
+            with gr.Row():
+                court = gr.Textbox(label="Vara")
+                jurisdiction = gr.Textbox(label="Comarca")
+            with gr.Row():
+                process_number = gr.Textbox(label="Número do Processo")
+                lawyer_name = gr.Textbox(label="Nome do Advogado")
+            with gr.Row():
+                lawyer_oab = gr.Textbox(label="Número OAB")
+                lawyer_state = gr.Textbox(label="Estado OAB")
+
         with gr.Row():
-            botao_gerar = gr.Button("Gerar Peça Jurídica")
-            botao_redefinir = gr.Button("Redefinir Campos")
-        
-        saida_texto = gr.Textbox(
-            label="Peça Jurídica Gerada",
-            lines=15,
-            placeholder="A peça jurídica gerada aparecerá aqui."
-        )
+            generate_btn = gr.Button("Gerar Documento")
+            clear_btn = gr.Button("Limpar Campos")
 
-        botao_gerar.click(
-            gerar_peca_juridica_criminal, 
-            inputs=[entrada_caso, entrada_tipo], 
-            outputs=saida_texto
+        output = gr.Textbox(
+            label="Documento Gerado",
+            lines=20,
+            placeholder="O documento gerado aparecerá aqui"
         )
 
-        botao_redefinir.click(
-            redefinir_campos, 
-            inputs=None, 
-            outputs=[entrada_caso, entrada_tipo, saida_texto]
+        def generate(
+            case_info, doc_type, 
+            name, nationality, profession, id_num,
+            address, city, state, zip_code,
+            court, jurisdiction, process_num,
+            lawyer_name, lawyer_oab, lawyer_state
+        ):
+            client_data = {
+                "nome": name,
+                "nacionalidade": nationality,
+                "profissao": profession,
+                "rg": id_num,
+                "endereco": address,
+                "cidade": city,
+                "estado": state,
+                "cep": zip_code,
+                "vara": court,
+                "comarca": jurisdiction,
+                "numero_processo": process_num,
+                "advogado": lawyer_name,
+                "numero_oab": lawyer_oab,
+                "estado_oab": lawyer_state
+            }
+            
+            return generator.generate_document(case_info, doc_type, client_data)
+
+        def clear_fields():
+            return [""] * 16
+
+        generate_btn.click(
+            generate,
+            inputs=[
+                case_info, doc_type,
+                client_name, client_nationality, client_profession, client_id,
+                client_address, client_city, client_state, client_zip,
+                court, jurisdiction, process_number,
+                lawyer_name, lawyer_oab, lawyer_state
+            ],
+            outputs=output
         )
-        
-        with gr.Row():
-            botao_exportar = gr.Button("Exportar Peça Jurídica para .txt")
-            saida_arquivo = gr.Textbox(
-                label="Arquivo Exportado",
-                interactive=False,
-                placeholder="O caminho do arquivo exportado será exibido aqui."
-            )
-        
-        botao_exportar.click(
-            exportar_peca,
-            inputs=saida_texto,
-            outputs=saida_arquivo
+
+        clear_btn.click(
+            clear_fields,
+            outputs=[
+                case_info, doc_type,
+                client_name, client_nationality, client_profession, client_id,
+                client_address, client_city, client_state, client_zip,
+                court, jurisdiction, process_number,
+                lawyer_name, lawyer_oab, lawyer_state
+            ]
         )
-    
+
     return interface
 
-# Executa o app
 if __name__ == "__main__":
-    app = gerar_interface_gradio()
-    app.launch()
+    interface = create_interface()
+    interface.launch()