Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import os
 import requests
+import re
 import threading
 from datetime import datetime
 from typing import List, Dict, Any, Generator
 from session_manager import SessionManager
-from sympy import latex, sympify
 
 # Initialize session manager and get HF API key
 session_manager = SessionManager()
@@ -67,18 +67,9 @@ def query_model(model_name: str, messages: List[Dict[str, str]]) -> str:
         response = requests.post(endpoint, json=payload, headers=headers)
         response.raise_for_status()
         result = response.json()[0]['generated_text']
-        # Clean up response formatting and handle LaTeX
+        # Clean up response formatting
         result = result.split('<|')[0]  # Remove any remaining special tokens
-        result = result.replace('**', '').replace('##', '')  # Remove markdown
-        
-        # Detect and format LaTeX content
-        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 = re.sub(r'\*\*|##', '', result)  # Remove markdown bold/headers but preserve LaTeX
         result = result.strip()  # Remove leading/trailing whitespace
         return result  # Return complete response
     except Exception as e:
@@ -148,29 +139,22 @@ def respond(message: str, history: List[List[str]], session_id: str) -> Generato
     # Return final combined response
     yield f"🔵 **Qwen2.5-Coder-32B-Instruct**\n{response1}\n\n🟣 **Qwen2.5-72B-Instruct**\n{response2}\n\n🟡 **Llama3.3-70B-Instruct**\n{response3}"
 
-# Create the Gradio interface with LaTeX support
+# Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("## Multi-LLM Collaboration Chat")
-    # Add MathJax configuration for LaTeX rendering
-    demo.css = """
-    .math { font-size: 1.1em; }
-    """
-    demo.js = """
-    MathJax = {
-        tex: {
-            inlineMath: [['$', '$'], ['\\(', '\\)']]
-        },
-        svg: {
-            fontCache: 'global'
-        }
-    };
-    """
+    gr.HTML("<script src='https://cdnjs.cloudflare.com/ajax/libs/mathjax/3.2.0/es5/tex-mml-chtml.js' id='MathJax-script'></script>")
     
     with gr.Row():
         session_id = gr.State(session_manager.create_session)
         new_session = gr.Button("🔄 New Session")
     
-    chatbot = gr.Chatbot(height=600)
+    chatbot = gr.Chatbot(
+        height=600,
+        latex_delimiters=[
+            {"left": "$$", "right": "$$", "display": True},
+            {"left": "$", "right": "$", "display": False}
+        ]
+    )
     msg = gr.Textbox(label="Message")
     
     def on_new_session():