initial commit

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,12 @@
+---
+title: Chatbot Judicial
+emoji: ⚡
+colorFrom: green
+colorTo: purple
+sdk: gradio
+sdk_version: 4.42.0
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,70 @@
+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"
+)
+
+# Define three prompt options with user-friendly names
+prompts = [
+    os.getenv("prompt1"),
+    os.getenv("prompt2"),
+    os.getenv("prompt3"),
+]
+
+option_names = [
+    "Sentença em linguagem simples",
+    "Sentença em linguagem simples (Júri)",
+    "Transcrição de oitiva",
+]
+
+# 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]
+
+    # 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})
+    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": message})
+
+
+
+    response = client.chat.completions.create(
+         model='gpt-4o-mini',
+    messages= history_openai_format,
+    temperature=1,
+    max_tokens=2000,   
+    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
+
+    # Update the chat history for the user
+    chat_histories[username].append((message, partial_message))
+
+option_dropdown = gr.Dropdown(choices=option_names, label="Select Option")
+username_textbox = gr.Textbox(label="Username", placeholder="Enter your username") 
+
+gr.ChatInterface(predict, additional_inputs=[option_dropdown, username_textbox]
+).launch(share=True, auth=(os.getenv("username"), os.getenv("password")))

requirements.txt

@@ -0,0 +1,3 @@
+openai
+gradio
+python-dotenv