added submit & resume later button

views/nav_buttons.py

@@ -39,7 +39,7 @@ def navigation_buttons(data, response: Response):
 
     current_index = st.session_state.current_index
 
-    col1, col2, col3 = st.columns([1, 1, 2])
+    col1, col2, col3, col4 = st.columns([1, 1, 1, 2])
 
     with col1:  # Back button #TODO fix: only gets ratings for the session, not from previous session
         if st.button("Back", disabled=st.session_state.current_index == 0):
@@ -66,6 +66,13 @@ def navigation_buttons(data, response: Response):
                     if st.button("Finish"):
                         submit_feedback(current_index)
 
-    with col3:  # Save & Resume Later button
+    with col3:  # Submit button
+        if st.button("Submit & Resume Later"):
+            all_ratings = flatten_ratings(response)
+            if any(rating == 0 for rating in all_ratings):
+                st.warning("Please provide all ratings before submitting.")
+            else:
+                submit_feedback(current_index)
+    with col4:  # Save & Resume Later button
         if st.button("Exit & Resume Later"):
             submit_feedback(current_index)

views/questions_screen.py

@@ -227,7 +227,7 @@ def questions_screen(data):
 
     try:
         config = data.iloc[current_index]
-
+        st.markdown(f"## Hello {st.session_state.username.capitalize()} 👋")
         # Progress bar
         progress = (current_index + 1) / len(data)
         st.progress(progress)
@@ -247,7 +247,7 @@ def questions_screen(data):
             # st.write("**Cities:**", config["city"])
         # with st.expander("Full Context", expanded=False):
         #     st.text_area("", config["context"], height=300, disabled=False)
-
+        st.markdown("### Rate the following queries based on the above context.")
         g_ratings = display_ratings_row("gemini", config, current_index)
         l_ratings = display_ratings_row("llama", config, current_index)