Update app.py

app.py

@@ -80,15 +80,27 @@ class MathRetrievalQA(dspy.Program):
         if not context:
             return {"answer": "", "retrieved_context": ""}
 
-        # Prompt Gemini to generate and format a math answer
-        prompt = f"""You are a math expert. Given the question and context below, generate a step-by-step solution in a clear, neat, and well-formatted way.
+        prompt = f"""
+You are a math expert writing solutions in a textbook. Provide a detailed, step-by-step solution for the following math problem.
 
-Question: {question}
+Follow these guidelines:
+- Use clear headings like "Step 1", "Step 2", etc.
+- Use proper mathematical notation with LaTeX.
+- Make the explanation educational and logically structured.
+- Format the final answer as: Final Answer: \\boxed{{...}}
 
-Context:
+---
+
+Problem:
+{question}
+
+Context (if needed for the solution):
 {context}
 
-Answer:"""
+Write the solution in full below:
+"""
+
+
 
         try:
             model = genai.GenerativeModel('gemini-2.0-flash')  # or use 'gemini-1.5-flash'