Update app.py

app.py

@@ -51,23 +51,40 @@ def chatbot_interface():
         if len(st.session_state.messages) == 1:
             st.session_state.messages.append({"role": "bot", "content": "On a scale of 1 to 10, how would you rate your anxiety level?"})
         elif len(st.session_state.messages) == 2:
-            anxiety_level = int(user_input)
-            st.session_state.user_data = {'anxiety_level': anxiety_level}
-            st.session_state.messages.append({"role": "bot", "content": f"Got it! Your anxiety level is {anxiety_level}. Now, on a scale of 1 to 10, how would you rate your self-esteem?"})
+            # Validate anxiety level input
+            try:
+                anxiety_level = int(user_input)
+                if anxiety_level < 1 or anxiety_level > 10:
+                    raise ValueError
+                st.session_state.user_data = {'anxiety_level': anxiety_level}
+                st.session_state.messages.append({"role": "bot", "content": f"Got it! Your anxiety level is {anxiety_level}. Now, on a scale of 1 to 10, how would you rate your self-esteem?"})
+            except ValueError:
+                st.error("Please enter a valid number between 1 and 10 for anxiety level.")
         elif len(st.session_state.messages) == 3:
-            self_esteem = int(user_input)
-            st.session_state.user_data['self_esteem'] = self_esteem
-            st.session_state.messages.append({"role": "bot", "content": f"Got it! Your self-esteem level is {self_esteem}. Please select your mental health history from the options below."})
+            # Validate self-esteem input
+            try:
+                self_esteem = int(user_input)
+                if self_esteem < 1 or self_esteem > 10:
+                    raise ValueError
+                st.session_state.user_data['self_esteem'] = self_esteem
+                st.session_state.messages.append({"role": "bot", "content": f"Got it! Your self-esteem level is {self_esteem}. Please select your mental health history from the options below."})
+            except ValueError:
+                st.error("Please enter a valid number between 1 and 10 for self-esteem.")
         elif len(st.session_state.messages) == 4:
             st.session_state.user_data['mental_health_history'] = user_input
             st.session_state.messages.append({"role": "bot", "content": "Thanks! Now, on a scale of 0 to 2, how would you rate your stress level?"})
         elif len(st.session_state.messages) == 5:
-            stress_level = int(user_input)
-            st.session_state.user_data['stress_level'] = stress_level
-            st.session_state.messages.append({"role": "bot", "content": "Thank you for providing the information! Let me get your personalized advice..."})
-            # Get the advice from Groq API
-            advice = get_personalized_advice(st.session_state.user_data)
-            st.session_state.messages.append({"role": "bot", "content": advice or "I couldn't retrieve specific advice right now, but please check back later."})
+            try:
+                stress_level = int(user_input)
+                if stress_level < 0 or stress_level > 2:
+                    raise ValueError
+                st.session_state.user_data['stress_level'] = stress_level
+                st.session_state.messages.append({"role": "bot", "content": "Thank you for providing the information! Let me get your personalized advice..."})
+                # Get the advice from Groq API
+                advice = get_personalized_advice(st.session_state.user_data)
+                st.session_state.messages.append({"role": "bot", "content": advice or "I couldn't retrieve specific advice right now, but please check back later."})
+            except ValueError:
+                st.error("Please enter a valid number between 0 and 2 for stress level.")
         elif len(st.session_state.messages) == 6:
             st.session_state.messages.append({"role": "bot", "content": "Would you like to start over?"})