Update app.py

app.py

@@ -5,8 +5,7 @@ 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
+import mistune
 
 API_URL = "https://host.palple.polrambora.com/pmsq"
 
@@ -149,16 +148,34 @@ def render_message(history):
 
 
 def escape_and_format(text):
-    md = MarkdownIt()
+    allowed_tags = {'<b>', '</b>', '<i>', '</i>', '<h1>', '</h1>', '<h2>', '</h2>', '<h3>', '</h3>', 
+                    '<code>', '</code>', '<br>', '<ul>', '</ul>', '<li>', '</li>', '<blockquote>', '</blockquote>', '<p>', '</p>'}
     
-    html = md.render(text)
+    markdown = mistune.create_markdown(renderer="html")
     
-    html = html.replace("\n", "<br>")
+    html = markdown(text)
     
-    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)
+    filtered_html = ""
+    i = 0
+    while i < len(html):
+        if html[i] == '<':
+            tag_end = html.find('>', i) + 1
+            if tag_end == 0: 
+                filtered_html += f"&#{ord(html[i])};"
+                i += 1
+                continue
+            
+            tag = html[i:tag_end]
+            if tag in allowed_tags:
+                filtered_html += tag
+            else:
+                filtered_html += ''.join(f"&#{ord(c)};" for c in tag)
+            i = tag_end
+        else:
+            filtered_html += f"&#{ord(html[i])};"
+            i += 1
     
-    return html
+    return filtered_html
 
 css="""
 .chatbox {height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex;  flex-direction: column-reverse;}