Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 from huggingface_hub import InferenceClient
 from datasets import load_dataset
 import random
+import re
 # Load math datasets for sample problems
 fw = load_dataset("HuggingFaceFW/fineweb-edu", name="sample-10BT", split="train", streaming=True)
 ds = load_dataset("HuggingFaceH4/ultrachat_200k", streaming=True)
@@ -62,6 +63,13 @@ def respond(
     temperature,
     top_p,
 ):
+    def render_latex(text):
+    # Wrap inline math properly
+    text = re.sub(r'\$(.*?)\$', r'$$\1$$', text)  # Display mode
+    return text
+
+    yield render_latex(response)
+    
     # Show initial thinking message
     yield "🤔 Thinking for the suitable answer..."