bug fix: added st.rerun() after next

views/nav_buttons.py

@@ -52,33 +52,34 @@ def navigation_buttons(data, response: Response):
                 st.warning("You are at the beginning of the survey, can't go back.")
             # st.rerun()
 
-    with col2:  # Next button TODO might be buggy
-        if st.button("Next", disabled=st.session_state.current_index == len(data) - 1):
+    with col2:  # Next button
+        if st.button("Next", disabled=current_index == len(data) - 1):
             all_ratings = flatten_ratings(response)
             if any(rating == 0 for rating in all_ratings):
                 st.warning("Please provide all ratings before proceeding.")
-            else:
-                if current_index < len(data) - 1:
-                    config_id = data.iloc[st.session_state.current_index]['config_id']
-                    if isinstance(response.model_ratings, dict):
-                        st.session_state.previous_ratings[config_id] = {
-                            model: {
-                                'query_v_ratings': ratings.query_v_ratings,
-                                'query_p0_ratings': ratings.query_p0_ratings,
-                                'query_p1_ratings': ratings.query_p1_ratings
-                            }
-                            for model, ratings in response.model_ratings.items()
+            elif current_index < len(data) - 1:
+                # Store the ratings with proper structure
+                config_id = data.iloc[current_index]['config_id']
+                if isinstance(response.model_ratings, dict):
+                    st.session_state.previous_ratings[config_id] = {
+                        model: {
+                            'query_v_ratings': ratings.query_v_ratings,
+                            'query_p0_ratings': ratings.query_p0_ratings,
+                            'query_p1_ratings': ratings.query_p1_ratings
                         }
-                    else:
-                        st.session_state.previous_ratings[config_id] = {
-                            model: {
-                                'query_v_ratings': getattr(ratings, 'query_v_ratings', {}),
-                                'query_p0_ratings': getattr(ratings, 'query_p0_ratings', {}),
-                                'query_p1_ratings': getattr(ratings, 'query_p1_ratings', {})
-                            }
-                            for model, ratings in response.model_ratings.items()
+                        for model, ratings in response.model_ratings.items()
+                    }
+                else:
+                    st.session_state.previous_ratings[config_id] = {
+                        model: {
+                            'query_v_ratings': getattr(ratings, 'query_v_ratings', {}),
+                            'query_p0_ratings': getattr(ratings, 'query_p0_ratings', {}),
+                            'query_p1_ratings': getattr(ratings, 'query_p1_ratings', {})
                         }
-                    st.session_state.current_index += 1
+                        for model, ratings in response.model_ratings.items()
+                    }
+                st.session_state.current_index += 1
+                st.rerun()
 
     with col3:  # Submit button
         if st.button("Submit & Resume Later"):