save

app.py

@@ -73,23 +73,13 @@ about = """
 Replace this placeholder with the actual text.
 """
 
-with st.expander("How to use this App"):
-    st.markdown(about)
 
-# --- Load the image ---
-# image = Image.open("higher_ed.png")
-# st.image(image, width=400)
-
-st.header("Quadratic Equations")
-#load the problems form the pdf
-with st.spinner("AI is thinking..."):
-    if st.session_state.uploaded_pdf_path is None:
-        st.session_state.uploaded_pdf_path = get_local_pdf_path()
+import re
+import json
 
-    filepath = st.session_state.uploaded_pdf_path
-    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.
+# Define constants
+TEXT_PROMPT = """Use the provided document. Read the list of 5 quadratic equations. 
+Return your response as a JSON list. Do not include any extra text, explanations, or backslashes.
 
 Example JSON output:
 [
@@ -100,22 +90,63 @@ Example JSON output:
   "x^2 + 8x + 15 = 0"
 ]
 """
-    response = multimodal_prompt(filepath, text_prompt)  # Use the local filepath
-    st.markdown(response)
 
-try:
-    problems = json.loads(response)
+# Define a function to extract equations from the AI response
+def extract_equations(response):
+    try:
+        if isinstance(response, str):
+            response = response.strip().replace("\n", "").replace("\r", "")
+            if response.lower().startswith("json"):
+                response = response[4:].strip()
+            if response.startswith("[") and response.endswith("]"):
+                return json.loads(response)
+            else:
+                st.error("Error: AI response is not in expected JSON list format.")
+                return []
+        elif isinstance(response, list):
+            return response
+        else:
+            st.error("Error: Unexpected response format from AI.")
+            return []
+    except json.JSONDecodeError:
+        st.error("Error: Failed to parse AI response as a list.")
+        return []
+
+# Define a function to extract quadratic equations from the problems
+def extract_quadratic_equations(problems):
     equations = []
     for problem in problems:
-        equation = problem.split(": ")[1] #Split the string at ": " and take the second part.
-        equations.append(equation)
-    st.write("Equations only:")
+        match = re.search(r'([0-9x\^\+\-\=\s]+)', problem)
+        if match:
+            equations.append(match.group(1).strip())
+        else:
+            st.warning(f"Could not extract equation from: '{problem}'")
+    return equations
+
+# Main code
+with st.spinner("AI is thinking..."):
+    if st.session_state.get("uploaded_pdf_path") is None:
+        st.session_state.uploaded_pdf_path = get_local_pdf_path()
+
+    filepath = st.session_state.uploaded_pdf_path
+    response = multimodal_prompt(filepath, TEXT_PROMPT)
+
+# Debugging: Print response
+st.write("Raw AI Response:", response)
+
+# Extract equations
+problems = extract_equations(response)
+if problems:
+    equations = extract_quadratic_equations(problems)
+    st.write("Extracted Equations:")
     for equation in equations:
         st.write(equation)
-except json.JSONDecodeError:
-    st.write("Error: Invalid JSON format in the response.")
+else:
+    st.error("Error: No valid equations extracted.")
+
 except Exception as e:
-    st.write(f"An unexpected error occurred: {e}")
+    st.error(f"An unexpected error occurred: {e}")
+
 
 st.markdown("Visit our Hugging Face Space!")
 st.markdown("© 2025 WVSU AI Dev Team 🤖 ✨")