app.py

import os
import gradio as gr
from huggingface_hub import InferenceClient

# 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
        },
        # Add other templates as needed...
    }

    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"
                )
            },
            {
                "role": "user",
                "content": texto_final
            }
        ]

        resposta_final = client.chat.completions.create(
            model="mistralai/Mistral-7B-Instruct-v0.3",
            messages=mensagens,
            max_tokens=template['max_tokens'],
            temperature=0.7
        )

        return resposta_final.choices[0].message["content"]

    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(
                label="Informações do Caso",
                placeholder="Descreva o caso detalhadamente ou selecione um exemplo abaixo.",
                lines=5
            )
            
            exemplo_caso = gr.Dropdown(
                label="Exemplos de Casos",
                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
            )
            
            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.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."
        )

        botao_gerar.click(
            gerar_peca_juridica_criminal, 
            inputs=[entrada_caso, entrada_tipo], 
            outputs=saida_texto
        )

        botao_redefinir.click(
            redefinir_campos, 
            inputs=None, 
            outputs=[entrada_caso, entrada_tipo, saida_texto]
        )
        
        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
        )
    
    return interface

# Executa o app
if __name__ == "__main__":
    app = gerar_interface_gradio()
    app.launch()