maybe fixed bug

app.py

@@ -6,9 +6,10 @@ from dotenv import load_dotenv
 from views.intro_screen import welcome_screen
 from views.questions_screen import questions_screen
 from views.continue_survey import continue_survey_screen
+from views.ui_helpers import exit_screen
 from css.layout import custom_css
-st.set_page_config(layout="wide")
 
+st.set_page_config(layout="wide")
 
 load_dotenv()
 VALIDATION_CODE = os.getenv("VALIDATION_CODE")
@@ -36,28 +37,6 @@ def initialization():
         st.session_state.previous_ratings = {}
 
 
-def exit_screen():
-    """Display exit screen"""
-    st.markdown("""
-        <div class='exit-container'>
-            <h1>Thank you for participating!</h1>
-            <p>Your responses have been saved successfully.</p>
-            <p>You can safely close this window or start a new survey.</p>
-        </div>
-    """, unsafe_allow_html=True)
-
-    if st.button("Resume Survey"):
-        reset_survey()
-        st.rerun()
-
-
-def reset_survey():
-    """Reset the survey state to start over."""
-    st.session_state.responses = []
-    st.session_state.completed = True
-    st.session_state.start_new_survey = True
-
-
 def ui():
     """Main function to control the survey flow."""
     custom_css()

views/intro_screen.py

@@ -1,8 +1,6 @@
 import streamlit as st
 import os
 from dotenv import load_dotenv
-from views.continue_survey import continue_survey_screen
-from db.crud import read
 
 load_dotenv()
 VALIDATION_CODE = os.getenv("VALIDATION_CODE")

views/nav_buttons.py

@@ -5,6 +5,8 @@ from datetime import datetime
 import os
 from dotenv import load_dotenv
 
+from views.ui_helpers import display_completion_message
+
 load_dotenv()
 VALIDATION_CODE = os.getenv("VALIDATION_CODE")
 
@@ -71,4 +73,5 @@ def navigation_buttons(data, response: Response):
             submit_feedback(current_index)
             if st.session_state.current_index == len(data) - 1:
                 st.session_state.completed = True
+                display_completion_message()
                 # st.rerun()

views/questions_screen.py

@@ -3,28 +3,10 @@ import streamlit as st
 from datetime import datetime
 from dotenv import load_dotenv
 from views.nav_buttons import navigation_buttons
-import random
 
 load_dotenv()
 
 
-def display_completion_message():
-    """Display a standardized survey completion message."""
-    st.markdown(
-        """
-        <div class='exit-container'>
-            <h1>You have already completed the survey! Thank you for participating!</h1>
-            <p>Your responses have been saved successfully.</p>
-            <p>You can safely close this window or start a new survey.</p>
-        </div>
-        """,
-        unsafe_allow_html=True,
-    )
-    st.session_state.show_questions = False
-    st.session_state.completed = True
-    st.session_state.start_new_survey = True
-
-
 def get_previous_ratings(model_name, query_key, current_index):
     """Retrieve previous ratings from session state."""
     previous_ratings = {}

views/ui_helpers.py

@@ -0,0 +1,40 @@
+import streamlit as st
+
+
+def exit_screen():
+    """Display exit screen"""
+    st.markdown("""
+        <div class='exit-container'>
+            <h1>Thank you for participating!</h1>
+            <p>Your responses have been saved successfully.</p>
+            <p>You can safely close this window or start a new survey.</p>
+        </div>
+    """, unsafe_allow_html=True)
+
+    if st.button("Resume Survey"):
+        reset_survey()
+        st.rerun()
+
+
+def reset_survey():
+    """Reset the survey state to start over."""
+    st.session_state.responses = []
+    st.session_state.completed = True
+    st.session_state.start_new_survey = True
+
+
+def display_completion_message():
+    """Display a standardized survey completion message."""
+    st.markdown(
+        """
+        <div class='exit-container'>
+            <h1>You have already completed the survey! Thank you for participating!</h1>
+            <p>Your responses have been saved successfully.</p>
+            <p>You can safely close this window or start a new survey.</p>
+        </div>
+        """,
+        unsafe_allow_html=True,
+    )
+    st.session_state.show_questions = False
+    st.session_state.completed = True
+    st.session_state.start_new_survey = True