Refactor chat history formatting in app.py

app.py

@@ -82,9 +82,11 @@ 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)
-        # Remove roles from the chat history
-        formatted_chat_history = [[msg["content"] for msg in chat_history if msg["role"] == "user"],
-                                [msg["content"] for msg in chat_history if msg["role"] == "assistant"]]
+        # Format the chat history for the gr.Chatbot component
+        formatted_chat_history = []
+        for user_msg, assistant_msg in zip([msg["content"] for msg in chat_history if msg["role"] == "user"],
+                                        [msg["content"] for msg in chat_history if msg["role"] == "assistant"]):
+            formatted_chat_history.append([user_msg, assistant_msg])
         return formatted_chat_history, chat_history, ""