nav buttons fixed the exit & save thing

views/nav_buttons.py

@@ -4,6 +4,7 @@ import streamlit as st
 from datetime import datetime
 import os
 from dotenv import load_dotenv
+
 load_dotenv()
 VALIDATION_CODE = os.getenv("VALIDATION_CODE")
 
@@ -24,6 +25,13 @@ def submit_feedback(current_index):
         st.error(f"An error occurred while submitting feedback: {e}")
 
 
+def any_value_zero(response):
+    return any(
+        any(value == 0 for value in query.values())
+        for query in [response.query_v, response.query_p0, response.query_p1]
+    )
+
+
 def navigation_buttons(data, ratings_v, ratings_p0, ratings_p1):
     """Display navigation buttons."""
     current_index = st.session_state.current_index
@@ -48,4 +56,10 @@ def navigation_buttons(data, ratings_v, ratings_p0, ratings_p1):
 
     with col3:  # Save & Resume Later button
         if st.button("Exit & Resume Later"):
-            submit_feedback(current_index)
+            last_response = st.session_state.responses[-1]
+            if not any_value_zero(last_response):
+                submit_feedback(current_index)
+            else:
+                st.session_state.completed = True
+                st.rerun()
+            # submit_feedback(current_index)