Update app.py

app.py

@@ -5,6 +5,8 @@ import os
 import markdown
 from markdown.extensions.codehilite import CodeHiliteExtension
 import markdown.extensions.fenced_code
+from markdown_it import MarkdownIt
+from markdown_it.plugins import plugin
 
 API_URL = "https://host.palple.polrambora.com/pmsq"
 
@@ -147,14 +149,16 @@ def render_message(history):
 
 
 def escape_and_format(text):
-    text = (
-        text.replace("**", "<b>").replace("__", "</b>") 
-        .replace("`", "<code>").replace("```", "</code>")  
-        .replace("\n", "<br>") 
-    )
-
-    return ''.join(f"&#{ord(char)};" if char not in ('<', '>', '<br>', '<b>', '</b>', '<code>', '</code>') else char for char in text)
-
+    md = MarkdownIt()
+    
+    html = md.render(text)
+    
+    html = html.replace("\n", "<br>")
+    
+    allowed_tags = {'<b>', '</b>', '<i>', '</i>', '<h1>', '</h1>', '<h2>', '</h2>', '<h3>', '</h3>', '<code>', '</code>', '<br>', '<ul>', '</ul>', '<li>', '</li>', '<blockquote>', '</blockquote>', '<p>', '</p>'}
+    html = ''.join(f"&#{ord(char)};" if char not in allowed_tags else char for char in html)
+    
+    return html
 
 css="""
 .chatbox {height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex;  flex-direction: column-reverse;}