Update app.py

app.py

@@ -7,6 +7,14 @@ import markdown
 # Initialize the Hugging Face Inference Client
 client = InferenceClient()
 
+
+def render_latex(latex_input):
+    try:
+        rendered_html = katex.render(latex_input, display_mode=True)
+        return f"{rendered_html}"
+    except Exception as e:
+        return f"{latex_input}"
+
 # Function to generate and format AI response
 def generate_response(prompt_template, **kwargs):
     # Simulate processing/loading
@@ -28,8 +36,7 @@ def generate_response(prompt_template, **kwargs):
             "markdown.extensions.attr_list",
         ]
     )
-    return f"""<script src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js"></script>
-    {html}"""
+    return f"""{render_latex(html)}"""
 
 # Gradio app interface
 with gr.Blocks() as app: