Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py
 import gradio as gr
 import os
 import requests
@@ -66,11 +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
+        # Do NOT remove ** or ## so that Markdown (including LaTeX) is preserved
         result = result.split('<|')[0]  # Remove any remaining special tokens
-        result = result.replace('**', '').replace('##', '')  # Remove markdown
-        result = result.strip()  # Remove leading/trailing whitespace
-        return result  # Return complete response
+        return result.strip()  # Return the cleaned response
     except Exception as e:
         return f"{model_name} error: {str(e)}"
 
@@ -141,6 +140,17 @@ def respond(message: str, history: List[List[str]], session_id: str) -> Generato
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("## Multi-LLM Collaboration Chat")
+    gr.Markdown(
+        """
+        **LaTeX Support:** You can include inline math like `$E = mc^2$` or block math like:
+        ```
+        $$
+        \\int_0^1 x^2 \\, dx
+        $$
+        ```
+        and it will be rendered correctly below.
+        """
+    )
     
     with gr.Row():
         session_id = gr.State(session_manager.create_session)