Update app.py

app.py

@@ -1,24 +1,24 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Conexão com o modelo
+# Cliente do modelo
 client = InferenceClient("meta-llama/Llama-3.1-8B-Instruct")
 
-SYSTEM_PROMPT = """
-Você é JuriX, uma IA formal, profissional e objetiva.
-Responda sempre com clareza, precisão e linguagem jurídica técnica.
-"""
+SYSTEM_PROMPT = (
+    "Você é JuriX, um assistente jurídico formal, profissional e técnico. "
+    "Responda sempre com precisão e clareza."
+)
 
 def chat_fn(message, history):
     if history is None:
         history = []
 
-    # Monta as mensagens para o modelo
+    # Monta mensagens
     msgs = [{"role": "system", "content": SYSTEM_PROMPT}]
 
-    for h in history:
-        msgs.append({"role": "user", "content": h[0]})
-        msgs.append({"role": "assistant", "content": h[1]})
+    for user_msg, bot_msg in history:
+        msgs.append({"role": "user", "content": user_msg})
+        msgs.append({"role": "assistant", "content": bot_msg})
 
     msgs.append({"role": "user", "content": message})
 
@@ -29,20 +29,19 @@ def chat_fn(message, history):
         temperature=0.4
     )
 
-    bot_msg = response.choices[0].message["content"]
+    bot_reply = response.choices[0].message["content"]
+    history.append((message, bot_reply))
 
-    # Atualiza o histórico
-    history.append((message, bot_msg))
     return "", history
 
 
-with gr.Blocks(theme="gradio/soft") as demo:
+with gr.Blocks() as demo:
     gr.Markdown("# ⚖️ JuriX — Assistente Jurídico Profissional")
-    gr.Markdown("Bem-vindo! Como posso ajudar?")
+    gr.Markdown("Olá! Sou a JuriX e estou aqui para ajudar.")
 
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Digite sua mensagem")
+    text = gr.Textbox(label="Digite sua mensagem e pressione ENTER")
 
-    msg.submit(chat_fn, [msg, chatbot], [msg, chatbot])
+    text.submit(chat_fn, [text, chatbot], [text, chatbot])
 
-demo.launch()
+demo.launch()