Update app.py

app.py

@@ -305,8 +305,16 @@ if openai_api_key:
                     solution = generate_math_solution(math_query)
 
                     # Render the solution with LaTeX for mathematical notations
-                    st.write("### Solution and Explanation:")
-                    st.markdown(f"**Solution:**\n\n{solution}")
+                    formatted_solution = f"""
+                    ### Solution to the Problem
+                    **Problem:** {math_query}
+
+                    **Solution:**
+
+                    {solution}
+                    """
+
+                    st.markdown(formatted_solution)
             else:
                 st.error("Please enter a math problem to solve.")