Update app.py

app.py

@@ -29,6 +29,10 @@ if "is_initialized" not in st.session_state:
     st.session_state.is_initialized = False
 if "index_dimensions" not in st.session_state:
     st.session_state.index_dimensions = 1024  # Set this based on your Pinecone index
+if "quiz_submitted" not in st.session_state:
+    st.session_state.quiz_submitted = False
+if "quiz_answers" not in st.session_state:
+    st.session_state.quiz_answers = {}
 
 # Functions for PDF processing
 def extract_text_from_pdf(pdf_file):
@@ -277,6 +281,14 @@ def generate_assignment(pdf_content, assignment_type="short_answer", num_questio
         st.error(f"Error parsing assignment response: {str(e)}")
         return []
 
+# Callback functions for quiz submission and reset
+def submit_quiz():
+    st.session_state.quiz_submitted = True
+
+def reset_quiz():
+    st.session_state.quiz_submitted = False
+    st.session_state.quiz_answers = {}
+
 # Streamlit UI
 def main():
     st.title("📚 PDF Learning Assistant")
@@ -355,6 +367,8 @@ GOOGLE_API_KEY=your_google_api_key
                 with st.spinner("Generating quiz..."):
                     quiz_data = generate_quiz(st.session_state.current_pdf_content, num_questions=quiz_questions)
                     st.session_state.quiz_data = quiz_data
+                    # Reset quiz state when generating a new quiz
+                    reset_quiz()
             
             # Assignment generation
             assignment_type = st.selectbox(
@@ -426,38 +440,41 @@ GOOGLE_API_KEY=your_google_api_key
         if "quiz_data" in st.session_state and st.session_state.quiz_data:
             quiz_data = st.session_state.quiz_data
             
-            if "quiz_answers" not in st.session_state:
-                st.session_state.quiz_answers = {}
-                st.session_state.quiz_submitted = False
-            
+            # Quiz display logic - static until submitted
             if not st.session_state.quiz_submitted:
-                for i, question in enumerate(quiz_data):
-                    st.subheader(f"Question {i+1}")
-                    st.write(question["question"])
-                    
-                    options = question["options"]
-                    option_labels = ["A", "B", "C", "D"]
-                    
-                    # Create radio buttons for options
-                    answer = st.radio(
-                        "Select your answer:",
-                        options=option_labels[:len(options)],
-                        key=f"q{i}",
-                        index=None
-                    )
-                    
-                    # Display options
-                    for j, option in enumerate(options):
-                        st.write(f"{option_labels[j]}: {option}")
+                # Quiz form
+                with st.form(key="quiz_form"):
+                    for i, question in enumerate(quiz_data):
+                        st.subheader(f"Question {i+1}")
+                        st.write(question["question"])
+                        
+                        options = question["options"]
+                        option_labels = ["A", "B", "C", "D"]
+                        
+                        # Create radio buttons for options
+                        answer = st.radio(
+                            "Select your answer:",
+                            options=option_labels[:len(options)],
+                            key=f"q{i}",
+                            index=None
+                        )
+                        
+                        # Display options
+                        for j, option in enumerate(options):
+                            st.write(f"{option_labels[j]}: {option}")
+                        
+                        # Store answer in session state
+                        if answer:
+                            st.session_state.quiz_answers[i] = answer
+                        
+                        st.divider()
                     
-                    st.session_state.quiz_answers[i] = answer
-                    st.divider()
-                
-                if st.button("Submit Quiz"):
-                    st.session_state.quiz_submitted = True
-                    st.experimental_rerun()
+                    # Submit button inside the form
+                    submit_button = st.form_submit_button("Submit Quiz")
+                    if submit_button:
+                        st.session_state.quiz_submitted = True
             else:
-                # Show results
+                # Show results after submission
                 correct_count = 0
                 
                 for i, question in enumerate(quiz_data):
@@ -488,38 +505,48 @@ GOOGLE_API_KEY=your_google_api_key
                 st.subheader(f"Your Score: {correct_count}/{len(quiz_data)}")
                 
                 if st.button("Retake Quiz"):
-                    st.session_state.quiz_submitted = False
-                    st.session_state.quiz_answers = {}
-                    st.experimental_rerun()
+                    reset_quiz()
         else:
             st.info("Generate a quiz from the sidebar to see it here")
     
     # Tab 3: Assignment
+# Tab 3: Assignment
     with tab3:
         st.header("Assignment")
         
         if "assignment_data" in st.session_state and st.session_state.assignment_data:
             assignment_data = st.session_state.assignment_data
             
-            for i, question in enumerate(assignment_data):
-                with st.expander(f"Question {i+1}", expanded=True):
+            # Create a form for the assignment
+            with st.form(key="assignment_form"):
+                for i, question in enumerate(assignment_data):
+                    st.subheader(f"Question {i+1}")
                     st.write(question["question"])
                     
+                    # Use a checkbox to toggle hints instead of a nested expander
                     if "hints" in question and question["hints"]:
-                        st.subheader("Hints")
-                        for hint in question["hints"]:
-                            st.write(f"- {hint}")
+                        show_hints = st.checkbox(f"Show hints for Question {i+1}", key=f"hint_checkbox_{i}")
+                        if show_hints:
+                            for hint in question["hints"]:
+                                st.write(f"- {hint}")
                     
-                    # Input area for answers
+                    # Input area for the answer
                     st.text_area("Your Answer:", key=f"assignment_q{i}", height=150)
-                    
-                    # Reveal key points button
-                    if st.button("Show Key Points", key=f"key_points_btn_{i}"):
-                        st.subheader("Key Points to Include")
+                    st.divider()
+                
+                # Add the submit button as a direct child of the form
+                submit_assignment = st.form_submit_button("Submit Assignment")
+            
+            # Process form submission outside the form block
+            if submit_assignment:
+                st.success("Assignment submitted! Here are the key points for each question:")
+                for i, question in enumerate(assignment_data):
+                    with st.expander(f"Key Points for Question {i+1}", expanded=True):
                         for point in question["key_points"]:
                             st.write(f"- {point}")
         else:
             st.info("Generate an assignment from the sidebar to see it here")
+    
 
 if __name__ == "__main__":
-    main()
+    main()