Update app.py

app.py

@@ -16,6 +16,21 @@ genai.configure(api_key="AIzaSyBO3-HG-WcITn58PdpK7mMyvFQitoH00qA")  # Replace wi
 # Load Gemini model
 gemini_model = genai.GenerativeModel('gemini-pro')
 
+
+import re
+
+def latex_to_plain_math(latex_expr):
+    # Replace LaTeX formatting with plain text math
+    latex_expr = latex_expr.strip()
+    latex_expr = re.sub(r"\\frac\{(.+?)\}\{(.+?)\}", r"(\1) / (\2)", latex_expr)
+    latex_expr = re.sub(r"\\sqrt\{(.+?)\}", r"√(\1)", latex_expr)
+    latex_expr = latex_expr.replace("^2", "²").replace("^3", "³")
+    latex_expr = re.sub(r"\^(\d)", r"^\1", latex_expr)  # other powers
+    latex_expr = latex_expr.replace("\\pm", "±")
+    latex_expr = latex_expr.replace("\\cdot", "⋅")
+    latex_expr = latex_expr.replace("{", "").replace("}", "")
+    return latex_expr
+
 # === Load Models ===
 print("Loading zero-shot classifier...")
 classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
@@ -104,9 +119,11 @@ Write only the formatted solution, as it would appear in a math textbook.
 
         try:
             model = genai.GenerativeModel('gemini-2.0-flash')  # or use 'gemini-1.5-flash'
+            
             response = model.generate_content(prompt)
-            formatted_answer = response.text
+            formatted_answer = latex_to_plain_math(response.text)
             print("Gemini Answer:", formatted_answer)
+
             return {"answer": formatted_answer, "retrieved_context": context}
         except Exception as e:
             print("Gemini generation error:", e)