Update app.py

app.py

@@ -134,11 +134,21 @@ def render_message(history):
             <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>
-            </div>
-             <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>
-            """
+            </div>"""
+
+    messages_html += """
+        </div>
+    </div>
+    <script>
+        // Trigger MathJax rendering
+        function processMath() {
+            if (typeof MathJax !== 'undefined') {
+                MathJax.typeset();
+            }
+        }
+        processMath();
+    </script>
+    """
 
     return messages_html