Update app.py

app.py

@@ -139,19 +139,25 @@ def render_message(history):
     messages_html += """
         </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>
-        // Trigger MathJax rendering
-        function processMath() {
+        document.addEventListener("DOMContentLoaded", function() {
             if (typeof MathJax !== 'undefined') {
                 MathJax.typeset();
             }
-        }
-        processMath();
+        });
     </script>
     """
 
     return messages_html
 
+
 js = """
 function processMath() {
     if (typeof MathJax !== 'undefined') {