Update app.py

app.py

@@ -91,6 +91,8 @@ class MathRetrievalQA(dspy.Program):
         context_items = retrieve_from_qdrant(question)
         context = "\n".join([item["solution"] for item in context_items if "solution" in item])
         print("Context for generation:", context)
+        f = latex_to_plain_math(context)
+        print(f)
 
         if not context:
             return {"answer": "", "retrieved_context": ""}
@@ -109,7 +111,7 @@ Problem:
 {question}
 
 Use the following context if needed:
-{context}
+{f}
 
 Write only the formatted solution, as it would appear in a math textbook.
 """
@@ -121,7 +123,7 @@ Write only the formatted solution, as it would appear in a math textbook.
             model = genai.GenerativeModel('gemini-2.0-flash')  # or use 'gemini-1.5-flash'
             
             response = model.generate_content(prompt)
-            formatted_answer = latex_to_plain_math(response.text)
+            formatted_answer = response.text
             print("Gemini Answer:", formatted_answer)
 
             return {"answer": formatted_answer, "retrieved_context": context}