Refactor chat history formatting logic in app.py

app.py

@@ -91,20 +91,23 @@ with gr.Blocks() as demo:
         last_role = None
         for entry in chat_history:
             role = entry["role"]
-            if role != last_role:
-                if role == "user":
-                    formatted_chat_history.append([entry["content"], ""])
-                elif role == "assistant":
-                    formatted_chat_history.append(["", entry["content"]])
-                last_role = role
-            else:
-                if role == "user":
-                    formatted_chat_history[-1][0] += entry["content"]
+            content = entry["content"].strip()  # Trim whitespace to check for empty content
+            if content:  # Only proceed if content is not empty
+                if role != last_role:
+                    if role == "user":
+                        formatted_chat_history.append([content, ""])
+                    elif role == "assistant":
+                        formatted_chat_history.append(["", content])
+                    last_role = role
                 else:
-                    formatted_chat_history[-1][1] += entry["content"]
+                    if role == "user" and content:
+                        formatted_chat_history[-1][0] += "\n" + content
+                    elif role == "assistant" and content:
+                        formatted_chat_history[-1][1] += "\n" + content
 
         return formatted_chat_history, chat_history, ""
 
+
     submit.click(
         fn=update_chatbot,
         inputs=[message, chat_history_state],