Update app.py

app.py

@@ -29,10 +29,6 @@ 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):
@@ -281,14 +277,6 @@ 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")
@@ -367,8 +355,6 @@ 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(
@@ -440,41 +426,38 @@ 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
             
-            # Quiz display logic - static until submitted
+            if "quiz_answers" not in st.session_state:
+                st.session_state.quiz_answers = {}
+                st.session_state.quiz_submitted = False
+            
             if not st.session_state.quiz_submitted:
-                # 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()
+                for i, question in enumerate(quiz_data):
+                    st.subheader(f"Question {i+1}")
+                    st.write(question["question"])
                     
-                    # Submit button inside the form
-                    submit_button = st.form_submit_button("Submit Quiz")
-                    if submit_button:
-                        st.session_state.quiz_submitted = True
+                    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}")
+                    
+                    st.session_state.quiz_answers[i] = answer
+                    st.divider()
+                
+                if st.button("Submit Quiz"):
+                    st.session_state.quiz_submitted = True
+                    st.experimental_rerun()
             else:
-                # Show results after submission
+                # Show results
                 correct_count = 0
                 
                 for i, question in enumerate(quiz_data):
@@ -505,48 +488,38 @@ GOOGLE_API_KEY=your_google_api_key
                 st.subheader(f"Your Score: {correct_count}/{len(quiz_data)}")
                 
                 if st.button("Retake Quiz"):
-                    reset_quiz()
+                    st.session_state.quiz_submitted = False
+                    st.session_state.quiz_answers = {}
+                    st.experimental_rerun()
         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
             
-            # 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}")
+            for i, question in enumerate(assignment_data):
+                with st.expander(f"Question {i+1}", expanded=True):
                     st.write(question["question"])
                     
-                    # Use a checkbox to toggle hints instead of a nested expander
                     if "hints" in question and question["hints"]:
-                        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}")
+                        st.subheader("Hints")
+                        for hint in question["hints"]:
+                            st.write(f"- {hint}")
                     
-                    # Input area for the answer
+                    # Input area for answers
                     st.text_area("Your Answer:", key=f"assignment_q{i}", height=150)
-                    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):
+                    
+                    # Reveal key points button
+                    if st.button("Show Key Points", key=f"key_points_btn_{i}"):
+                        st.subheader("Key Points to Include")
                         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()