Update app.py

app.py

@@ -138,7 +138,7 @@ def render_message(history):
         if user_message and ("user", user_message) not in seen_messages:
             seen_messages.add(("user", user_message))
             user_message_html = markdown.markdown(
-                escape_html(user_message),
+                user_message,  # Directly pass user_message
                 extensions=["fenced_code", "codehilite"]
             )
             user_message_html = bleach.clean(user_message_html, tags=allowed_tags, attributes=allowed_attributes, strip=True)
@@ -151,7 +151,7 @@ def render_message(history):
         if assistant_message and ("assistant", assistant_message) not in seen_messages:
             seen_messages.add(("assistant", assistant_message))
             assistant_message_html = markdown.markdown(
-                escape_html(assistant_message),
+                assistant_message,  # Directly pass assistant_message
                 extensions=["fenced_code", "codehilite"]
             )
             assistant_message_html = bleach.clean(assistant_message_html, tags=allowed_tags, attributes=allowed_attributes, strip=True)
@@ -166,8 +166,8 @@ def render_message(history):
 
 
 def escape_html(unsafe_text):
-    escaped_text = ''.join(f"&#{ord(char)};" if char not in ('\n', '\r') else '<br>' for char in unsafe_text)
-    return escaped_text
+    return unsafe_text.replace('&', '&amp;').replace('<', '&lt;').replace('>', '&gt;')
+
 css="""
 .chatbox {height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex;  flex-direction: column-reverse;}
 """