Update app.py

app.py

@@ -6,6 +6,25 @@ import time  # To simulate processing time if needed
 # Initialize the Hugging Face Inference Client
 client = InferenceClient()
 
+
+
+def preprocess_latex(content):
+    # Split content into lines for better formatting
+    lines = content.split("\n")
+    formatted_lines = []
+    
+    for line in lines:
+        # If a line contains equations or math expressions, wrap them properly
+        if "Simplify" in line or "Solve" in line or "boxed" in line or "frac" in line:
+            formatted_lines.append(f"$$ {line.strip()} $$")  # Block math
+        elif "(" in line and ")" in line:  # Inline math for variables
+            formatted_lines.append(line.replace("(", "$").replace(")", "$"))
+        else:
+            formatted_lines.append(line)  # Plain text
+
+    # Join lines back into a single string
+    return "\n".join(formatted_lines)
+
 # Function to generate and format AI response
 def generate_response(prompt_template, **kwargs):
     # Simulate processing/loading
@@ -19,7 +38,7 @@ def generate_response(prompt_template, **kwargs):
         top_p=0.8
     )
     response_content = response.choices[0].message["content"]
-    formatted_response = response_content.replace("[", "$$").replace("]", "$$")
+    formatted_response = preprocess_latex(response_content)
     return gr.update(value=f"{formatted_response}")
 
 # Gradio app interface