Update app.py

app.py

@@ -9,21 +9,23 @@ client = InferenceClient()
 
 
 def preprocess_latex(content):
+    """Preprocess the AI response to format LaTeX properly."""
     lines = content.split("\n")  # Split response into lines
     processed_lines = []
     
     for line in lines:
-        # Identify and format block math
-        if "[" in line and "]" in line:  # Treat as block math
-            math_expr = line[line.index("[") + 1:line.index("]")]  # Extract the math inside brackets
-            processed_lines.append(f"$$ {math_expr.strip()} $$")
-        elif "$" in line:  # Inline math expressions
-            processed_lines.append(line)  # Inline math is already wrapped
-        else:
-            processed_lines.append(line)  # Plain text remains unchanged
+        if "[" in line and "]" in line:  # Detect math expressions in brackets
+            while "[" in line and "]" in line:
+                start = line.index("[")
+                end = line.index("]") + 1
+                math_expr = line[start:end].strip("[]").strip()
+                line = line[:start] + f"$$ {math_expr} $$" + line[end:]
+        processed_lines.append(line)
 
     return "\n".join(processed_lines)  # Join lines back into a single string
 
+
+
 # Function to generate and format AI response
 def generate_response(prompt_template, **kwargs):
     # Simulate processing/loading