save

app.py

@@ -91,6 +91,27 @@ def multimodal_prompt(pdf_path, text_prompt, file_type="PDF"):
     except Exception as e:
         return f"An error occurred: {e}"
 
+def get_equation(response):
+    # Remove the ```json and ``` and extra spaces.
+    try:
+        json_string = response.replace('```json', '').replace('```', '').strip()
+
+        # Parse the JSON string into a Python list.
+        problems_list = json.loads(json_string)
+
+        # return the first item found
+        return problems_list[0] 
+
+    except json.JSONDecodeError:
+        st.error("Invalid JSON format in the response.")
+        return None
+    except Exception as e:
+        st.error(f"An unexpected error occurred: {e}")
+        return None
+
+if "problem_step" not in st.session_state:
+    st.session_state.problem_step = 0   
+
 # --- Main Page ---
 st.title("📚❓Problem Solving Tutor")
 about = """
@@ -108,21 +129,21 @@ with st.spinner("Loading the problem..."):
 # --- Display the image ---
 st.image(filepath, caption="Problem Image", use_container_width=True)
 
-# Remove the ```json and ``` and extra spaces.
-try:
-    json_string = response.replace('```json', '').replace('```', '').strip()
-
-    # Parse the JSON string into a Python list.
-    problems_list = json.loads(json_string)
-
-    # Iterate over the list and print each item using st.write().
-    for item in problems_list:
-        st.write(item)
-
-except json.JSONDecodeError:
-    st.write("Error: Invalid JSON format in the response.")
-except Exception as e:
-    st.write(f"An unexpected error occurred: {e}")
+equation = get_equation(response)
+st.write(f"**Equation:** {equation}")   
+
+problem_step = st.session_state.problem_step
+if problem_step == 0:
+    #Show instructions to submit the answer
+    st.write("Please write down your answer in a piece of paper.  Take a picture of the paper and submit it in the next step.")
+    st.write("Click the button below to proceed.")
+    image = st.camera_input("Take a picture of your answer.")
+
+    if image is not None:
+        # process the answer
+        st.write("Processing your answer...")    
+    else:
+        st.write("Please take a picture of your answer.")
 
 
 st.markdown("Visit our Hugging Face Space!")