Update app.py

app.py

@@ -130,7 +130,7 @@ def render_message(history):
             messages_html += f"""
             <div style='display: flex; align-items: center; margin-bottom: 10px;'>
                 <img src='{user_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>
-                <span style='color: white;'>{user_message_html}</span>
+                <span style='color: white; white-space: pre-wrap;'>{user_message_html}</span>
             </div>"""
 
         if assistant_message and assistant_message not in seen_messages:
@@ -139,7 +139,7 @@ def render_message(history):
             messages_html += f"""
             <div style='display: flex; align-items: center; margin-bottom: 10px;'>
                 <img src='{assistant_pic}' style='width: 40px; height: 40px; border-radius: 50%; margin-right: 10px;'>
-                <span style='color: white;'>{assistant_message_html}</span>
+                <span style='color: white; white-space: pre-wrap;'>{assistant_message_html}</span>
             </div>"""
 
     messages_html += "</div></div>"
@@ -147,10 +147,11 @@ def render_message(history):
 
 
 def escape_html(unsafe_text):
-    escaped_text = ''.join(f"&#{ord(char)};" if char != '\n' else '<br>' for char in unsafe_text)
+    escaped_text = ''.join(f"&#{ord(char)};" if char not in ('\n', '\r') else '<br>' for char in unsafe_text)
     return escaped_text
 
 
+
 css="""
 .chatbox {height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex;  flex-direction: column-reverse;}
 """