Update app.py

app.py

@@ -126,20 +126,17 @@ if "profile_data" in st.session_state:
 
             if st.button("Submit Answer", key=f"submit{st.session_state.question_index}"):
                 if answer:
-                    st.warning("Data saved successfully. click again to proceed")
-                    # Save the answer and increment the question index
-                    st.session_state.question_index += 1
                     st.session_state.answers[question] = answer
-
-                    # No need to call a special function; the app will rerun automatically
+                    st.session_state.question_index += 1
+                    st.rerun()  # Trigger app rerun to show the next question
                 else:
                     st.warning("Please enter an answer before submitting.")
         else:
             st.success("All questions have been answered. Click below to generate your recommendations.")
             if st.button("Generate Response"):
-                st.warning("Data saved successfully. click again to proceed")
                 st.session_state.profile_data.update(st.session_state.answers)
                 st.session_state.ask_additional_questions = False
+                st.rerun()  # Trigger app rerun after completing all questions
 
     elif st.session_state.get("ask_additional_questions") is False:
         # Directly generate recommendations