Update app.py

app.py

@@ -48,13 +48,21 @@ def chatbot_interface():
         st.error(message)
         return False
 
+    def clear_data():
+        """Clear session data and reset to initial step."""
+        st.session_state.step = 0
+        st.session_state.user_data = {}
+
     # Step 0: Welcome message and initial question
     if st.session_state.step == 0:
         st.write("Bot: Hi there! I'm here to help with your well-being. Let's start! 😊")
         st.write("Bot: I'll ask you a few questions about your well-being to give you personalized advice.")
         next_step_button = st.button("Start Chat")
+        clear_button = st.button("Clear")
         if next_step_button:
             next_step()
+        if clear_button:
+            clear_data()
 
     # Step 1: Ask for anxiety level
     elif st.session_state.step == 1:
@@ -105,9 +113,11 @@ def chatbot_interface():
 
         # Option to restart the chat
         restart_button = st.button("Start Over")
+        clear_button = st.button("Clear")
         if restart_button:
-            st.session_state.step = 0
-            st.session_state.user_data = {}
+            clear_data()
+        if clear_button:
+            clear_data()
 
 # Main function to run the chatbot
 def main():