Refactor chat history formatting in app.py

app.py

@@ -81,22 +81,19 @@ with gr.Blocks() as demo:
 
     def update_chatbot(message, chat_history, system_message=BASE_SYSTEM_MESSAGE):
         print("Updating chatbot...")
-        # Check if we're starting a new conversation or continuing an existing one
         if not chat_history or (chat_history and chat_history[-1]["role"] != "user"):
-            # Include the system message only if starting a new conversation
             chat_history = user(message, chat_history, system_message if not chat_history else None)
         else:
-            # If continuing an existing conversation, don't add the system message again
             chat_history = user(message, chat_history)
         chat_history, _ = chat(chat_history, system_message, max_tokens.value, temperature.value, top_p.value, 40, 1.1)
-        # Format the chat history for the gr.Chatbot component
         formatted_chat_history = []
         for entry in chat_history:
-            if entry["role"] in ["user", "assistant"]:
-                formatted_chat_history.append([entry["content"] if entry["role"] == "user" else "", entry["content"] if entry["role"] == "assistant" else ""])
+            if entry["role"] == "user":
+                formatted_chat_history.append([entry["content"], ""])
+            elif entry["role"] == "assistant":
+                formatted_chat_history.append(["", entry["content"]])
         return formatted_chat_history, chat_history, ""
 
-
     submit.click(
         fn=update_chatbot,
         inputs=[message, chat_history_state],