Update app.py

app.py

@@ -58,6 +58,15 @@ def fix_code_bugs(buggy_code, model="gpt-4o-mini"):
     )
     return response['choices'][0]['message']['content']
 
+# Function to generate AI-based mathematical solutions
+def generate_math_solution(query):
+    prompt = f"Explain and solve the following mathematical problem step by step: {query}"
+    response = openai.ChatCompletion.create(
+        model="gpt-4o-mini",
+        messages=[{"role": "user", "content": prompt}]
+    )
+    return response['choices'][0]['message']['content']
+
 from PIL import Image  # Required for local image files
 
 # Streamlit app starts here
@@ -289,15 +298,17 @@ if openai_api_key:
         # User input for math questions
         math_query = st.text_input("Ask a mathematics-related question:")
 
-        if math_query:
-            with st.spinner("Getting answer..."):
-                prompt = f"Answer the following math question: {math_query}"
-                response = openai.ChatCompletion.create(
-                    model="gpt-4o-mini",
-                    messages=[{"role": "user", "content": prompt}]
-                )
-                answer = response['choices'][0]['message']['content']
-                st.write(f"### Answer: {answer}")
+        if st.button("Solve Problem"):
+            if math_query:
+                with st.spinner("Generating solution..."):
+                    # Generate the solution using GPT-4
+                    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}")
+            else:
+                st.error("Please enter a math problem to solve.")
 
     # **New Section: Biology Assistant**
     elif mode == "Biology Assistant":