v2

app.py

@@ -1,100 +1,78 @@
 import os
 from openai import AzureOpenAI
 import gradio as gr
-from dotenv import load_dotenv
 
-load_dotenv()
 client = AzureOpenAI(
-  azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT"), 
-  api_key=os.getenv("AZURE_OPENAI_API_KEY"),  
-  api_version="2024-02-01"
+    azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"), 
+    api_key=os.getenv("AZURE_OPENAI_API_KEY"),  
+    api_version="2024-02-01"
 )
 
 # Define three prompt options with user-friendly names
 prompts = [
-    os.getenv("prompt1"),
-    os.getenv("prompt2"),
-    "Você é um chatbot especializado em direito brasileiro, treinado em legislação, jurisprudência e doutrina nacionais. Sua principal função é fornecer informações jurídicas precisas e confiáveis aos usuários. Para cumprir essa tarefa, siga estas diretrizes: 1. Base de Conhecimento: Legislação: Consulte a Constituição Federal, leis, decretos, portarias e demais normas jurídicas brasileiras atualizadas. Jurisprudência: Utilize julgados dos tribunais superiores brasileiros (STF, STJ, etc.), tribunais estaduais e tribunais federais, sempre buscando decisões recentes e relevantes para o tema em questão. Doutrina: Recorra a livros, artigos acadêmicos e obras de renomados juristas brasileiros para aprofundar sua análise e fornecer diferentes perspectivas sobre o assunto. 2. Metodologia de Resposta: Compreensão da Pergunta: Leia atentamente a pergunta do usuário para identificar o tema central, as dúvidas específicas e o contexto em que a questão se insere. Pesquisa e Análise: Utilize sua base de conhecimento para pesquisar as normas, julgados e doutrinas relevantes para o tema. Analise as informações encontradas, identificando os pontos convergentes e divergentes, e avalie a aplicabilidade das normas e precedentes ao caso concreto. Elaboração da Resposta: Estruture sua resposta de forma clara, objetiva e didática, utilizando linguagem jurídica acessível ao público leigo. Apresente as normas, julgados e doutrinas relevantes, explicando sua aplicação ao caso e destacando os pontos mais importantes. Precisão e Confiabilidade: Priorize a precisão e a confiabilidade das informações fornecidas, evitando especulações ou opiniões pessoais. Se houver dúvidas ou controvérsias sobre o tema, apresente as diferentes correntes e perspectivas, sem tomar partido.Atualização: Mantenha-se atualizado sobre as mudanças legislativas, jurisprudenciais e doutrinárias, buscando sempre as informações mais recentes e relevantes. Orientações adicionais: 1. Priorize a precisão acima de tudo. * Se não tiver informações suficientes para responder com precisão, recomende fontes alternativas ao usuário. * Nunca invente informações ou apresente algo como factual se não tiver certeza. * Seja transparente sobre as limitações do seu conhecimento e quando a informação pode estar desatualizada. 2. Responda com base em evidências. * Sempre busque suporte confiável para suas respostas, seja de seu conhecimento interno ou de pesquisas. * Cite suas fontes sempre que possível, especialmente ao lidar com tópicos complexos ou controversos. * Evite fazer suposições ou tirar conclusões que não sejam diretamente apoiadas pelas evidências disponíveis. 3. Lide com a incerteza com cuidado. * Se a pergunta for ambígua, peça esclarecimentos ao usuário antes de responder. * Se houver várias respostas possíveis, apresente-as de forma neutra e explique os diferentes pontos de vista. * Em caso de dúvida, erre do lado da cautela e evite fornecer informações que possam ser enganosas ou prejudiciais. * Lembre-se de que sua principal responsabilidade é fornecer informações precisas e confiáveis. 5. Lembre-se de suas limitações. * Você não é capaz de realizar ações no mundo físico. * Você não tem opiniões pessoais. Em tópicos controversos, apresente os diferentes pontos de vista de forma neutra e evite tomar partido. * Se solicitado a representar um lado específico de uma questão controversa, siga as instruções do usuário, mas mantenha um tom neutro e distanciado. Ao seguir estas ordens, você minimizará o risco de alucinação e fornecerá aos usuários as informações mais precisas e confiáveis possíveis.",
-    "Seja um assistente de IA útil",
-    "You are an expert AI coding assistant. Follow these principles when assisting: 1. **Clarity First:** Write clear, well-structured code with comments explaining complex logic. 2. **Security is Paramount:** Never introduce security vulnerabilities. If unsure about potential risks, prioritize safety. 3. **Context Matters:** Utilize any provided code context or instructions to generate the most accurate and relevant code. 4. **User Preferences:** When applicable, adapt your coding style to align with the existing code or the user's explicit instructions. 5. **Efficiency:**  If multiple approaches exist, favor the most efficient solution, unless the user prioritizes readability or maintainability.",
-]
-
-option_names = [
-    "Resumir Sentença em linguagem simples",
-    "Transformar Transcrição de oitiva judicial em citação indireta",
-    "IA jurídica",
-    "IA genérica",
-    "IA para tecnologia",    
+    "be a helpful assistant",
+    "Crie um resumo de aproximadamente uma página da sentença abaixo em linguagem simples e acessível, sem jargões. O resumo deve conter: 1) Tipo de sentença; 2) fatos relevantes: 2.1) data do crime; 2.2) Local do crime; 2.3 vítimas; 2.4) descrição do fato apurado; 3) A decisão 4) A fundamentação.",
+    """Abaixo consta uma transcrição de uma oitiva judicial. Faça uma citação indireta do que a depoente respondeu, separado por ponto e vírgula. O resultado deve constar tudo que foi dito pela testemunha, podendo ser omitido apenas o que for repetitivo.  O texto não deve ser interrompido até exaurir todo o conteúdo da oitiva.
+    exemplo de reposta: Estava trabalhando durante os fatos; Trabalhava em um depósito de cerveja ao lado de um posto de gasolina; Quando soube dos fatos a vítima estava no hospital; Fechou o comércio e foi até o Hospital de Acopiara, todavia a vítima foi transferida; Ouviu dizer que Leandro chegou alterado e começou a espancar Vitória, tendo a mãe da adolescente a defendido; Neste momento o réu sacou um punhal e perfurou a ofendida; Não sabe de nenhuma desavença entre a vítima e o réu; Acredita que o réu estava "virado" de uma vaquejada; Conhece o réu apenas de vista; É irmão e sobrinho de Luciene."""
 ]
 
 # Initialize a dictionary to store chat histories per user
 chat_histories = {}
 
-def predict(message, history, selected_option, username):
-    # Get the user's chat history or create a new one
-    if username not in chat_histories:
-        chat_histories[username] = []
-    history = chat_histories[username]
+def predict(message, prompt_index):
+    history = chat_histories.setdefault(prompt_index, [])
 
-    # Get the index of the selected option
-    selected_index = option_names.index(selected_option)
-    # Get the corresponding prompt based on the selected index
-    selected_prompt = prompts[selected_index]
-    history_openai_format = []
-    history_openai_format.append({"role": "system", "content": selected_prompt})
+    selected_prompt = prompts[prompt_index]
+    history_openai_format = [{"role": "system", "content": selected_prompt}]
     for human, assistant in history:
-        history_openai_format.append({"role": "user", "content": human })
-        history_openai_format.append({"role": "assistant", "content":assistant})
+        history_openai_format.append({"role": "user", "content": human})
+        history_openai_format.append({"role": "assistant", "content": assistant})
     history_openai_format.append({"role": "user", "content": message})
 
-
-
     response = client.chat.completions.create(
-         model='gpt-4o-mini',
-    messages= history_openai_format,
-    temperature=1,
-    max_tokens=2000,   
-    stream=True)
+        model='gpt-4o-mini',
+        messages=history_openai_format,
+        temperature=1.0,
+        stream=True
+    )
 
     partial_message = ""
     for chunk in response:
-        # Check if choices exist and are not empty
-        if chunk.choices and len(chunk.choices) > 0: 
-            if chunk.choices[0].delta.content is not None:
-                partial_message = partial_message + chunk.choices[0].delta.content
-                yield partial_message
+        if chunk.choices and chunk.choices[0].delta.content is not None:
+            partial_message += chunk.choices[0].delta.content
+            yield partial_message
 
-    # Update the chat history for the user
-    chat_histories[username].append((message, partial_message))
+    chat_histories[prompt_index].append((message, partial_message))
 
-option_dropdown = gr.Dropdown(choices=option_names, label="Select Option")
-username_textbox = gr.Textbox(label="Username", placeholder="Enter your username") 
+def predict_wrapper(message, prompt_index):
+    return list(predict(message, prompt_index))[-1]
 
-bot = gr.ChatInterface(
-    predict, 
-    additional_inputs=[option_dropdown, username_textbox],
-    title="Chatbot do direito 👩‍⚖️",
-    description="""Feito para auxiliar profissionais do direito no dia-a-dia.
-\n\n
-Instruções: 
-1. no canto inferior da tela clique em "Additional Input", selecione a tarefa desejada:
-    a) Resumir Sentença em linguagem simples;
-    b) Transformar Transcrição de oitiva judicial em citação indireta;
-    c) IA jurídica;
-    d) IA genérica;  
-2. Digite um nome de usuário (O nome de usuário serve apenas para não embaralhar as mensagens entre usuários concomitantes).
-3. Copie o conteúdo para o campo "insira sua mensagem" e depois clique em "enviar". \n
-e-mail: isa.[email protected]
-""",
-    theme="Soft",
-    examples=[
-        ["quais as teorias para o início da vida", "IA jurídica", "usuário1"], 
-        ["O que é exceção de pré-executividade", "IA jurídica", "Usuário1"], 
-    ],
-    retry_btn="🔄Refazer pergunta",
-    undo_btn=None,
-    clear_btn="🗑️Apagar tudo",
-    submit_btn="enviar",
-    stop_btn="parar",
-)
-bot.launch(auth=([(os.getenv("username"), os.getenv("password")), (os.getenv("username2"), os.getenv("password2"))]))
+def clear_history():
+    global chat_histories
+    chat_histories = {}
+
+# Gradio Blocks interface
+with gr.Blocks() as demo:
+    with gr.Tab("Genérica"):
+        message_input = gr.Textbox(label="Message")
+        output = gr.Textbox()
+        message_input.submit(predict_wrapper, inputs=[message_input, gr.State(0)], outputs=output)
+    
+    with gr.Tab("Resumir Sentença em linguagem simples"):
+        message_input = gr.Textbox(label="Message")
+        output = gr.Textbox()
+        message_input.submit(predict_wrapper, inputs=[message_input, gr.State(1)], outputs=output)
+    
+    with gr.Tab("Transformar Transcrição de oitiva judicial em citação indireta"):
+        message_input = gr.Textbox(label="Message")
+        output = gr.Textbox()
+        message_input.submit(predict_wrapper, inputs=[message_input, gr.State(2)], outputs=output)
+    
+    with gr.Tab("Clear Memory"):
+        clear_button = gr.Button("Clear Memory")
+        clear_button.click(fn=clear_history, inputs=[], outputs=[])
+
+try:
+    demo.launch(auth=([(os.getenv("username"), os.getenv("password"))]))
+finally:
+    clear_history()