Update app.py

app.py

@@ -5,7 +5,7 @@ import threading
 from datetime import datetime
 from typing import List, Dict, Any, Generator
 from session_manager import SessionManager
-import katex
+from sympy import latex, sympify
 
 # Initialize session manager and get HF API key
 session_manager = SessionManager()
@@ -72,11 +72,13 @@ def query_model(model_name: str, messages: List[Dict[str, str]]) -> str:
         result = result.replace('**', '').replace('##', '')  # Remove markdown
         
         # Detect and format LaTeX content
-        import re
-        # Handle inline LaTeX
-        result = re.sub(r'\\\((.*?)\\\)', r'<span class="katex">\1</span>', result)
-        # Handle display LaTeX
-        result = re.sub(r'\\\[(.*?)\\\]', r'<div class="katex">\1</div>', result)
+        try:
+            # Convert mathematical expressions to LaTeX
+            result = latex(sympify(result, evaluate=False))
+            result = f'<div class="math">{result}</div>'
+        except:
+            # If conversion fails, return the original text
+            pass
         result = result.strip()  # Remove leading/trailing whitespace
         return result  # Return complete response
     except Exception as e:
@@ -149,18 +151,19 @@ def respond(message: str, history: List[List[str]], session_id: str) -> Generato
 # Create the Gradio interface with LaTeX support
 with gr.Blocks() as demo:
     gr.Markdown("## Multi-LLM Collaboration Chat")
-    # Add KaTeX CSS and JS
+    # Add MathJax configuration for LaTeX rendering
     demo.css = """
-    .katex { font-size: 1.1em; }
+    .math { font-size: 1.1em; }
     """
     demo.js = """
-    function renderLatex() {
-        document.querySelectorAll('.katex').forEach(el => {
-            katex.render(el.textContent, el, {
-                throwOnError: false
-            });
-        });
-    }
+    MathJax = {
+        tex: {
+            inlineMath: [['$', '$'], ['\\(', '\\)']]
+        },
+        svg: {
+            fontCache: 'global'
+        }
+    };
     """
     
     with gr.Row():