Update app.py

app.py

@@ -8,9 +8,9 @@ groq_api_key = "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"
 # Initialize Groq client
 client = Groq(api_key=groq_api_key)
 
-# Define a function to fetch personalized advice using Groq API
+# Function to fetch personalized advice using Groq API
 def get_personalized_advice(user_data):
-    """Fetch personalized advice using the Groq API based on user data"""
+    """Fetch personalized advice using the Groq API based on user data."""
     query = f"Based on the user's data: anxiety level: {user_data['anxiety_level']}, self esteem: {user_data['self_esteem']}, mental health history: {user_data['mental_health_history']}, stress level: {user_data['stress_level']}, provide personalized mental health advice."
     
     try:
@@ -32,11 +32,12 @@ def stress_level_to_string(stress_level):
     else:
         return "High"
 
+# Chatbot interface for conversation-like experience
 def chatbot_interface():
     st.title("🎓 Student Well-being Chatbot")
-    st.write("I can help you with personalized well-being advice based on your responses. Let's get started!")
-
-    # Initializing session state
+    st.write("Hello! I’m here to assist you with well-being advice based on your responses.")
+    
+    # Initialize session state
     if 'step' not in st.session_state:
         st.session_state.step = 0
         st.session_state.user_data = {}
@@ -44,9 +45,13 @@ def chatbot_interface():
     def next_step():
         st.session_state.step += 1
 
+    def show_input_error(message="Please provide a valid input."):
+        st.error(message)
+        return False
+
     # Step 0: Welcome message and initial question
     if st.session_state.step == 0:
-        st.write("Hi! I am your well-being assistant. I'll ask you some questions to understand how you're feeling.")
+        st.write("I'm going to ask you a few questions to understand your well-being. Let's get started!")
         next_step_button = st.button("Start Chat")
         if next_step_button:
             next_step()
@@ -55,19 +60,32 @@ def chatbot_interface():
     elif st.session_state.step == 1:
         anxiety_level = st.text_input("On a scale of 1 to 10, how would you rate your anxiety level?", "")
         if anxiety_level:
-            st.session_state.user_data['anxiety_level'] = int(anxiety_level)
-            next_step()
+            try:
+                anxiety_level = int(anxiety_level)
+                if anxiety_level < 1 or anxiety_level > 10:
+                    raise ValueError
+                st.session_state.user_data['anxiety_level'] = anxiety_level
+                next_step()
+            except ValueError:
+                show_input_error("Please enter a number between 1 and 10 for anxiety level.")
 
     # Step 2: Ask for self-esteem
     elif st.session_state.step == 2:
         self_esteem = st.text_input("On a scale of 1 to 10, how would you rate your self-esteem?", "")
         if self_esteem:
-            st.session_state.user_data['self_esteem'] = int(self_esteem)
-            next_step()
+            try:
+                self_esteem = int(self_esteem)
+                if self_esteem < 1 or self_esteem > 10:
+                    raise ValueError
+                st.session_state.user_data['self_esteem'] = self_esteem
+                next_step()
+            except ValueError:
+                show_input_error("Please enter a number between 1 and 10 for self-esteem.")
 
     # Step 3: Ask for mental health history
     elif st.session_state.step == 3:
-        mental_health_history = st.selectbox("How would you describe your mental health history?", ["None", "Minor Issues", "Moderate Issues", "Severe Issues"])
+        mental_health_history = st.selectbox("How would you describe your mental health history?", 
+                                            ["None", "Minor Issues", "Moderate Issues", "Severe Issues"])
         st.session_state.user_data['mental_health_history'] = mental_health_history
         next_step()
 
@@ -83,15 +101,16 @@ def chatbot_interface():
         user_data = st.session_state.user_data
         st.write(f"Your data: {user_data}")
 
-        st.subheader("🔔 Personalized Health Advice")
+        # Show a conversational summary of the user's well-being status
+        st.subheader("🔔 Here's Your Personalized Well-being Advice:")
         advice = get_personalized_advice(user_data)
         if advice:
             st.write(advice)
         else:
-            st.write("No advice available at the moment.")
+            st.write("I couldn't retrieve specific advice right now, but please check back later.")
 
         # Option to restart the chat
-        restart_button = st.button("Restart Chat")
+        restart_button = st.button("Start Over")
         if restart_button:
             st.session_state.step = 0
             st.session_state.user_data = {}