Adjust chat history formatting for Gradio Chatbot component

app.py

@@ -82,10 +82,16 @@ with gr.Blocks() as demo:
         if not chat_history or (chat_history and chat_history[-1]["role"] != "user"):
             chat_history = user(message, chat_history)
         chat_history, _ = chat(chat_history, BASE_SYSTEM_MESSAGE, max_tokens.value, temperature.value, top_p.value, 40, 1.1)
-        # Ajusta a formatação do histórico do chat para o formato esperado pelo gr.Chatbot
-        formatted_chat_history = [[msg["role"], msg["content"]] for msg in chat_history]
+        # Adjust the formatting of the chat history for the Gradio Chatbot component
+        formatted_chat_history = []
+        for msg in chat_history:
+            if msg["role"] == "user":
+                formatted_chat_history.append(["user", msg["content"]])
+            else:
+                formatted_chat_history.append(["bot", msg["content"]])  # Change 'assistant' to 'bot' for clarity in the UI
         return formatted_chat_history, chat_history, ""
 
+
     submit.click(
         fn=update_chatbot,
         inputs=[message, chat_history_state],