Update app.py

app.py

@@ -281,23 +281,25 @@ if st.button("Generate Questions", key="generate_questions"):
     with st.spinner("Generating questions..."):
         questions = generate_questions(model_choice, role, question_type, num_questions, difficulty)
         if questions:
-            question_index = st.session_state.get('question_index', 0)
             st.session_state['questions'] = questions
-            st.session_state['question_index'] = question_index
+            st.session_state['question_index'] = 0  # Reset index to start
             st.session_state['questions_count'] = len(questions)
             
-            if question_index < len(questions):
-                st.markdown(style_output("Question:", "#4CAF50"), unsafe_allow_html=True)
-                st.write(st.session_state['questions'][question_index])
-                col1, col2 = st.columns([3, 1])
-                with col2:
-                    if st.button("Next", key="next_question"):
-                        if question_index + 1 < len(st.session_state['questions']):
-                            st.session_state['question_index'] += 1
-                        else:
-                            st.session_state['question_index'] = 0  # Loop back to first question
-                            st.success("You have reached the end of the questions.")
-        progress_data["questions_solved"][question_type] += num_questions
+            st.markdown(style_output("Question:", "#4CAF50"), unsafe_allow_html=True)
+            st.write(st.session_state['questions'][st.session_state['question_index']])
+            
+            col1, col2 = st.columns([3, 1])
+            with col2:
+                if st.button("Next", key="next_question"):
+                    if st.session_state['question_index'] + 1 < len(st.session_state['questions']):
+                        st.session_state['question_index'] += 1
+                    else:
+                        st.session_state['question_index'] = 0  # Loop back to first question
+                        st.success("You have reached the end of the questions.")
+                    st.markdown(style_output("Question:", "#4CAF50"), unsafe_allow_html=True)
+                    st.write(st.session_state['questions'][st.session_state['question_index']])
+            
+            progress_data["questions_solved"][question_type] += num_questions
 
 st.header("🗣️ Provide Feedback")
 answer = st.text_area("Submit Your Answer", key="text_area_answer")
@@ -336,4 +338,3 @@ st.markdown("""
     </div>
     """, unsafe_allow_html=True)
 
-