back button somewhat works

README.md

@@ -25,8 +25,8 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
 [x] Implement save and continue later button
 [x] Implement questions with proper buttons (with text)
 [ ] Back button
-[ ] Dataset linking HF
+[x] Dataset linking HF
 [x] prettify the context field with new lines and highlighting popularity etc. keywords in bold
-[ ] Doing it for two models - combine datasets
+[x] Doing it for two models - combine datasets
 [ ] Add check for ratings should not be 0 for Exit & Resume Later
 [ ] Check the firebase DB rules

app.py

@@ -30,6 +30,12 @@ def initialization():
         st.session_state.survey_continued = None
     if "start_new_survey" not in st.session_state:
         st.session_state.start_new_survey = False
+    if 'ratings' not in st.session_state:
+        st.session_state.ratings = {}
+    if 'previous_gemini_ratings' not in st.session_state:
+        st.session_state.previous_gemini_ratings = {}
+    if 'previous_llama_ratings' not in st.session_state:
+        st.session_state.previous_llama_ratings = {}
 
 
 def exit_screen():

views/continue_survey.py

@@ -35,7 +35,6 @@ def continue_survey_screen(data):
                 st.success("Resuming your survey!")
                 # Set survey_continued flag to True only when there's saved progress
                 st.session_state.survey_continued = True
-                print(st.session_state.current_index)
                 st.rerun()
                 # questions_screen(data)
 

views/nav_buttons.py

@@ -41,16 +41,24 @@ def navigation_buttons(data, response: Response):
 
     col1, col2, col3 = st.columns([1, 1, 2])
 
-    with col1:  # Back button #TODO fix
-        if st.button("Back") and current_index > 0:
-            st.session_state.current_index -= 1
-            st.rerun()
+    with col1:  # Back button #TODO fix: only gets ratings for the session, not from previous session
+        if st.button("Back"):
+            if current_index > 0:
+                st.session_state.current_index -= 1
+                previous_ratings = st.session_state.ratings.get(st.session_state.current_index, {})
+                st.session_state.previous_ratings = previous_ratings
+                # st.session_state.previous_gemini_ratings = previous_ratings.get("gemini", {})
+                # st.session_state.previous_llama_ratings = previous_ratings.get("llama", {})
+                st.rerun()
+            else:
+                st.warning("You are at the beginning of the survey, can't go back.")
+            # st.rerun()
 
     with col2:  # Next button
         if st.button("Next"):
             all_ratings = flatten_ratings(response)
             if any(rating == 0 for rating in all_ratings):
-                st.warning("Please provide a rating before proceeding.")
+                st.warning("Please provide all ratings before proceeding.")
             else:
                 if current_index < len(data) - 1:
                     st.session_state.current_index += 1
@@ -63,7 +71,9 @@ def navigation_buttons(data, response: Response):
             last_response = st.session_state.responses[-1]
             all_last_ratings = flatten_ratings(last_response)
             if not any(rating == 0 for rating in all_last_ratings):
+                print("Submitting feedback")
                 submit_feedback(current_index)
             else:
-                st.session_state.completed = True
-                st.rerun()
+                st.warning("Please provide ratings before exiting.")
+                # st.session_state.completed = True
+                # st.rerun()

views/questions_screen.py

@@ -48,6 +48,25 @@ def display_ratings_row(model_name, config, current_index):
 
 def render_query_ratings(model_name, query_label, config, query_key, current_index, has_persona_alignment=False):
     """Helper function to render ratings for a given query."""
+    # Get stored ratings if they exist
+    previous_ratings = st.session_state.get("previous_ratings", {}).get(model_name, {})
+    stored_query_ratings = {}
+    if previous_ratings:
+        if "query_v" in query_key:
+            stored_query_ratings = previous_ratings.query_v_ratings
+        elif "query_p0" in query_key:
+            stored_query_ratings = previous_ratings.query_p0_ratings
+        elif "query_p1" in query_key:
+            stored_query_ratings = previous_ratings.query_p1_ratings
+        else:
+            stored_query_ratings = {}
+    # Extract individual stored values, or default to None
+    # stored_query_ratings = previous_ratings.get(f"query_{query_key}_ratings", {})
+
+    stored_relevance = stored_query_ratings.get("relevance", 0)
+    stored_clarity = stored_query_ratings.get("clarity", 0)
+    stored_persona_alignment = stored_query_ratings.get("persona_alignment", 0) if has_persona_alignment else None
+
     if model_name == "gemini":
         bg_color = "#e0f7fa"
     else:
@@ -69,19 +88,25 @@ def render_query_ratings(model_name, query_label, config, query_key, current_ind
                     "Persona Alignment:", options=[0, 1, 2, 3, 4],
                     format_func=lambda x: ["N/A", "Not Aligned", "Partially Aligned", "Aligned", "Unclear"][x],
                     key=f"rating_{query_key}_persona_alignment_{current_index}",
+                    index=stored_persona_alignment if stored_persona_alignment is not None else 0
                 )
-
         with columns[1]:
-            relevance_rating = st.radio("Relevance:",
-                                        options, key=f"rating_{query_key}_relevance_{current_index}",
-                                        format_func=lambda x:
-                                        ["N/A", "Not Relevant", "Somewhat Relevant", "Relevant", "Unclear"][x], )
+            relevance_rating = st.radio(
+                "Relevance:",
+                options,
+                key=f"rating_{query_key}_relevance_{current_index}",
+                format_func=lambda x: ["N/A", "Not Relevant", "Somewhat Relevant", "Relevant", "Unclear"][x],
+                index=stored_relevance if stored_relevance is not None else 0
+            )
         with columns[2]:
-            clarity_rating = st.radio("Clarity:",
-                                      options=[0, 1, 2, 3],
-                                      key=f"rating_{query_key}_clarity_{current_index}",
-                                      format_func=lambda x: ["N/A", "Not Clear", "Somewhat Clear", "Very Clear"][x],
-                                      )
+            with columns[2]:
+                clarity_rating = st.radio(
+                    "Clarity:",
+                    options=[0, 1, 2, 3],
+                    key=f"rating_{query_key}_clarity_{current_index}",
+                    format_func=lambda x: ["N/A", "Not Clear", "Somewhat Clear", "Very Clear"][x],
+                    index=stored_clarity if stored_clarity is not None else 0
+                )
     return {
         "clarity": clarity_rating,
         "relevance": relevance_rating,
@@ -135,7 +160,7 @@ def questions_screen(data):
             comment=comment,
             timestamp=datetime.now().isoformat()
         )
-
+        st.session_state.ratings[current_index] = response.model_ratings
         if len(st.session_state.responses) > current_index:
             st.session_state.responses[current_index] = response
         else: