From 55d25632dfca0dfeecc910e070c40ff4c8ec803a Mon Sep 17 00:00:00 2001
From: Maxime Gasse <maxime.gasse@gmail.com>
Date: Wed, 29 May 2024 09:52:31 -0400
Subject: [PATCH] chat user messages no escape html (#53)

---
 core/src/browsergym/core/chat_files/chatbox_modern.html | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/core/src/browsergym/core/chat_files/chatbox_modern.html b/core/src/browsergym/core/chat_files/chatbox_modern.html
index 55b6b4de..3659d320 100644
--- a/core/src/browsergym/core/chat_files/chatbox_modern.html
+++ b/core/src/browsergym/core/chat_files/chatbox_modern.html
@@ -280,7 +280,6 @@
             msgContainer.className = 'message';
 
             const text = document.createElement('div');
-            text.innerHTML = addHtmlLineBreaks(escapeHtml(msg));
 
             // const assistant_img = document.createElement('img');
             // assistant_img.src = assistant_image_data;
@@ -290,12 +289,14 @@
             switch (role) {
                 case "user":
                     text.className = 'user-message';
+                    text.innerHTML = addHtmlLineBreaks(msg);
                     text.style.setProperty('--before-content', `"${timeString} - You"`);
                     msgContainer.appendChild(text);
                     chatBody.appendChild(msgContainer);
                     break;
                 case "assistant":
                     text.className = 'assistant-message';
+                    text.innerHTML = addHtmlLineBreaks(escapeHtml(msg));
                     text.style.setProperty('--before-content', `"${timeString} - Bot"`);
                     // msgContainer.appendChild(assistant_img); // Add the image to the message container
                     msgContainer.appendChild(text);
@@ -303,12 +304,14 @@
                     break;
                 case "infeasible":
                     text.className = 'assistant-message';
+                    text.innerHTML = addHtmlLineBreaks(escapeHtml(msg));
                     text.style.setProperty('--before-content', `"${timeString} - Bot (abort)"`);
                     msgContainer.appendChild(text);
                     chatBody.appendChild(msgContainer);
                     break;
                 case "info":
                     text.className = 'info-message';
+                    text.innerHTML = addHtmlLineBreaks(escapeHtml(msg));
                     msgContainer.appendChild(text);
                     // hide previous debug messages
                     for (const msg of chatDebug.children) {