potential bug fix

views/continue_survey.py

@@ -34,7 +34,8 @@ def continue_survey_screen(data):
                 # Set session state values based on saved progress
                 st.session_state.current_index = saved_state.get("current_index", 0)
                 st.session_state.responses = saved_state.get("responses", [])
-
+                if len(st.session_state.responses) == len(data):
+                    st.session_state.completed = True
                 # Find the last answered config_id in responses
                 last_config_id = None
                 if st.session_state.responses:

views/nav_buttons.py

@@ -68,4 +68,7 @@ def navigation_buttons(data, response: Response):
 
     with col3:  # Save & Resume Later button
         if st.button("Exit & Resume Later"):
-            submit_feedback(current_index)
+            submit_feedback(current_index)
+            if st.session_state.current_index == len(data) - 1:
+                st.session_state.completed = True
+                # st.rerun()

views/questions_screen.py

@@ -46,6 +46,7 @@ def get_previous_ratings(model_name, query_key, current_index):
         try:
             previous_ratings = response_from_session.model_ratings.get(model_name, {})
         except (AttributeError, KeyError):
+            print("Error in getting previous ratings from session state.")
             previous_ratings = response_from_session["model_ratings"].get(model_name, {})
 
     stored_query_ratings = {}