Update app.py

app.py

@@ -140,37 +140,48 @@ def render_message(history):
         </div>
     </div>
 
-    <!-- MathJax Script -->
-    <script type="text/javascript" id="MathJax-script" async
-        src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/3.2.2/es5/tex-mml-chtml.js">
-    </script>
-
-    <!-- Trigger MathJax Typeset -->
+    
     <script>
-        document.addEventListener("DOMContentLoaded", function() {
-            if (typeof MathJax !== 'undefined') {
-                MathJax.typeset();
+            function escapeHtml(unsafe) {
+                return unsafe
+                    .replace(/&/g, "&amp;")
+                    .replace(/</g, "&lt;")
+                    .replace(/>/g, "&gt;")
+                    .replace(/"/g, "&quot;")
+                    .replace(/'/g, "&#039;");
             }
-        });
+            let messages = document.querySelectorAll('#message-content');
+            messages.forEach((message) => {
+                message.innerHTML = DOMPurify.sanitize(message.innerHTML); // Sanitize HTML content
+            });
+            MathJax.typeset();  
     </script>
     """
 
     return messages_html
 
-
+def escape_html(unsafe_text):
+    return (
+        unsafe_text.replace("&", "&amp;")
+        .replace("<", "&lt;")
+        .replace(">", "&gt;")
+        .replace('"', "&quot;")
+        .replace("'", "&#039;")
+    )
 js = """
-function processMath() {
-    if (typeof MathJax !== 'undefined') {
-        MathJax.typeset();
-    }
-}
+console.log("hello")
 """
 
 css="""
 .chatbox {height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex;  flex-direction: column-reverse;}
 """
 with gr.Blocks(css=css, js=js) as demo:
-
+     gr.HTML("""
+    <script type="text/javascript" id="MathJax-script" async
+        src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/3.2.2/es5/tex-mml-chtml.js">
+    </script>
+    """)
+    
     with gr.Column(visible=True) as auth_view:
         gr.Markdown("## P-MSQ Authorization")
         gr.Markdown("P-MSQ is in closed alpha test! The model, api and more are subject to change.")
@@ -180,6 +191,7 @@ with gr.Blocks(css=css, js=js) as demo:
         auth_status = gr.Textbox(label="Authorization Status", interactive=False)
 
     with gr.Column(visible=False) as chat_view:
+        
         gr.Markdown("## P-MSQ Chat Interface")
         chatbot_output = gr.HTML(elem_id="chatbox-container")