Update app.py

app.py

@@ -206,15 +206,21 @@ with gr.Blocks(css=css) as demo:
         # Create a new history with emojis added
         history_with_emojis = []
         for user_msg, assistant_msg in history:
-            # Add the user emoji to the user message
+            # Add the user emoji to the user message (only if it doesn't already have one)
             if user_msg:
-                user_msg_with_emoji = f"👤 {user_msg}"
+                if not user_msg.startswith("👤 "):
+                    user_msg_with_emoji = f"👤 {user_msg}"
+                else:
+                    user_msg_with_emoji = user_msg
             else:
                 user_msg_with_emoji = user_msg
                 
-            # Add the assistant emoji to the assistant message
+            # Add the assistant emoji to the assistant message (only if it doesn't already have one)
             if assistant_msg:
-                assistant_msg_with_emoji = f"🛡️ {assistant_msg}"
+                if not assistant_msg.startswith("🛡️ "):
+                    assistant_msg_with_emoji = f"🛡️ {assistant_msg}"
+                else:
+                    assistant_msg_with_emoji = assistant_msg
             else:
                 assistant_msg_with_emoji = assistant_msg