fix math render

app.py

@@ -7,6 +7,8 @@ import time
 import requests
 from PIL import Image
 import gradio as gr
+
+
 # =========================
 # Config
 # =========================
@@ -22,6 +24,8 @@ LATEX_DELIMS = [
 ]
 AUTH_HEADER = {"Authorization": f"bearer {TOKEN}"}
 JSON_HEADERS = {**AUTH_HEADER, "Content-Type": "application/json"}
+
+
 # =========================
 # Base64 and Example Loading Logic
 # =========================
@@ -38,6 +42,25 @@ def image_to_base64_data_url(filepath: str) -> str:
         print(f"Error encoding image to Base64: {e}")
         return ""
 
+def _escape_inequalities_in_math(md: str) -> str:
+    """
+    Finds math blocks in a Markdown string and replaces < and > with
+    their LaTeX equivalents, \lt and \gt, to prevent markdown parsing errors.
+    """
+    _MATH_PATTERNS = [
+        re.compile(r"\$\$([\s\S]+?)\$\$"),
+        re.compile(r"\$([^\$]+?)\$"),
+        re.compile(r"\\\[([\s\S]+?)\\\]"),
+        re.compile(r"\\\(([\s\S]+?)\\\)"),
+    ]
+    def fix(s: str) -> str:
+        s = s.replace("<=", r" \le ").replace(">=", r" \ge ")
+        s = s.replace("≤", r" \le ").replace("≥", r" \ge ")
+        s = s.replace("<",  r" \lt ").replace(">",  r" \gt ")
+        return s
+    for pat in _MATH_PATTERNS:
+        md = pat.sub(lambda m: m.group(0).replace(m.group(1), fix(m.group(1))), md)
+    return md
 
 def _get_examples_from_dir(dir_path: str) -> List[List[str]]:
     supported_exts = {".png", ".jpg", ".jpeg", ".bmp", ".webp"}
@@ -193,7 +216,8 @@ def _process_api_response_page(result: Dict[str, Any]) -> Tuple[str, str, str]:
         output_html = f'<img src="{output_image_url}" alt="Detection Visualization">'
     else:
         print("Warning: No visualization image URL found in the API response.")
-
+        
+    md_text = _escape_inequalities_in_math(md_text)
     return md_text or "(Empty result)", output_html, md_text
 
 # =========================