save

app.py

@@ -87,14 +87,31 @@ with st.spinner("AI is thinking..."):
         st.session_state.uploaded_pdf_path = get_local_pdf_path()
 
     filepath = st.session_state.uploaded_pdf_path
-    text_prompt = f"Use the provided document. Create a list of math problems found in the document. Format the output as a JSON string"
+    text_prompt = """Use the provided document. "Read the list of 5 quadratic equations. 
+Return your response in JSON format, as a list of strings. 
+Do not include any extra text, explanations, or backslashes.
+
+Example JSON output:
+[
+  "x^2 - 5x + 6 = 0",
+  "2x^2 + 3x - 1 = 0",
+  "x^2 - 9 = 0",
+  "3x^2 - 2x + 4 = 0",
+  "x^2 + 8x + 15 = 0"
+]
+"""
     response = multimodal_prompt(filepath, text_prompt)  # Use the local filepath
     st.markdown(response)
 
 try:
     problems = json.loads(response)
+    equations = []
     for problem in problems:
-        st.write(problem)
+        equation = problem.split(": ")[1] #Split the string at ": " and take the second part.
+        equations.append(equation)
+    st.write("Equations only:")
+    for equation in equations:
+        st.write(equation)
 except json.JSONDecodeError:
     st.write("Error: Invalid JSON format in the response.")
 except Exception as e: