Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 from pathlib import Path
+import re
 
 # Disable Chroma telemetry (optional)
 os.environ["CHROMA_TELEMETRY_ENABLED"] = "false"
@@ -42,11 +43,30 @@ def chat_fn(message, history):
             sources = None
     if sources:
         answer = f"{answer}\n\n📚 Sources: {', '.join(sources)}"
+
+    def format_answer(answer):
+        # Wrap LaTeX formulas in a span so MathJax can render them
+        answer = re.sub(r"\$\$(.+?)\$\$", r'<span class="math">$$\1$$</span>', answer)
+        return f"<div>{answer}</div>"
+
+    answer = format_answer(answer)
     history.append((message, answer))
     return history, ""
 
-with gr.Blocks() as demo:
-    gr.Markdown("## 📚 Course Assistant — Chat with your course files")
+CSS = """
+* { direction: rtl; text-align: right; font-family: 'Vazir', sans-serif; }
+.gr-chat-message { white-space: pre-wrap; }
+.math { font-size: 1.2em; }
+"""
+
+with gr.Blocks(css=CSS) as demo:
+    demo.load(lambda: None, [], [], _js="""
+    const script = document.createElement('script');
+    script.src = "https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js";
+    document.head.appendChild(script);
+""")
+    
+    gr.Markdown("## 📚 SCR Course Assistant — Chat with course files")
     # chatbot = gr.Chatbot(elem_id="chatbot", type="messages")
     chatbot = gr.Chatbot(elem_id="chatbot", type="tuples")
     txt = gr.Textbox(show_label=False, placeholder="Ask about the course...")