Update app.py

app.py

@@ -16,29 +16,32 @@ def respond(user_message, history):
     if not user_message:
         return "", history
 
-    assistant_reply = client.chat.completions.create(
-        model="gpt-4o",
-        messages=[
-            {"role": "system", "content": MAIN_PROMPT},
-            *[
-                {"role": "user", "content": u} if i % 2 == 0 else {"role": "assistant", "content": a}
-                for i, (u, a) in enumerate(history)
+    try:
+        assistant_reply = client.chat.completions.create(
+            model="gpt-4o",
+            messages=[
+                {"role": "system", "content": MAIN_PROMPT},
+                *[
+                    {"role": "user", "content": u} if i % 2 == 0 else {"role": "assistant", "content": a}
+                    for i, (u, a) in enumerate(history)
+                ],
+                {"role": "user", "content": user_message}
             ],
-            {"role": "user", "content": user_message}
-        ],
-        max_tokens=512,
-        temperature=0.7,
-    ).choices[0].message.content
+            max_tokens=512,
+            temperature=0.7,
+        ).choices[0].message.content
+    except Exception as e:
+        assistant_reply = f"Error: {str(e)}"
 
     history.append((user_message, assistant_reply))
 
     return "", history
 
-# ✅ **Explicitly enable MathJax for LaTeX rendering**
+# ✅ **Ensure MathJax is Enabled for LaTeX Rendering**
 with gr.Blocks() as demo:
-    gr.Markdown("## AI-Guided Math PD Chatbot")
-    chatbot = gr.Chatbot(height=500)
+    gr.Markdown("# AI-Guided Math PD Chatbot")
 
+    chatbot = gr.Chatbot(height=500)
     state_history = gr.State([("", MAIN_PROMPT)])
 
     user_input = gr.Textbox(placeholder="Type your message here...", label="Your Input")
@@ -54,10 +57,10 @@ with gr.Blocks() as demo:
     )
 
     gr.Markdown(
-        r"""
-        <script type="text/javascript">
+        """
+        <script>
             MathJax = {
-                tex: { inlineMath: [['$', '$'], ['\\(', '\\)']] }
+                tex: { inlineMath: [['$', '$'], ['\\(', '\\)']], displayMath: [['$$', '$$']] }
             };
         </script>
         """