Update app.py

app.py

@@ -45,16 +45,29 @@ def health_advice_chatbot():
         ["Low", "Moderate", "High"]
     )
 
+    # Store user input in session state for persistent state across reruns
+    if 'user_data' not in st.session_state:
+        st.session_state.user_data = {}
+
+    st.session_state.user_data['anxiety_level'] = anxiety_level
+    st.session_state.user_data['self_esteem'] = self_esteem
+    st.session_state.user_data['stress_level'] = stress_level
+
     # Submit button to get health advice
     if st.button("Get Health Advice"):
         # Fetch health advice from Groq API
-        advice = get_health_advice_from_groq(anxiety_level, self_esteem, stress_level)
+        advice = get_health_advice_from_groq(
+            st.session_state.user_data['anxiety_level'],
+            st.session_state.user_data['self_esteem'],
+            st.session_state.user_data['stress_level']
+        )
         if advice:
             st.subheader("Here is your personalized health advice:")
             st.write(advice)
 
-    # Option to restart the conversation
+    # Option to reset form and restart
     if st.button("Start Over"):
+        st.session_state.user_data = {}
         st.experimental_rerun()
 
 # Main function to run the chatbot