Update app.py

app.py

@@ -9,15 +9,14 @@ client = Groq(api_key=groq_api_key)
 
 # 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 based on user data."""
     query = f"""
-    Based on the following user's responses:
+    Based on the following student's responses:
     - 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']}
-
-    Please provide professional, personalized mental health advice tailored to these responses.
+    Provide professional, personalized mental health advice.
     """
 
     try:
@@ -27,11 +26,11 @@ def get_personalized_advice(user_data):
         )
         return response.choices[0].message.content
     except Exception as e:
-        st.error(f"Error fetching advice from Groq: {str(e)}")
+        st.error(f"Error fetching advice: {str(e)}")
         return None
 
 def stress_level_to_string(stress_level):
-    """Convert numerical stress level (0, 1, 2) to a string representation."""
+    """Convert stress level to string representation."""
     if stress_level == 0:
         return "Low"
     elif stress_level == 1:
@@ -39,10 +38,10 @@ def stress_level_to_string(stress_level):
     else:
         return "High"
 
-# Chatbot interface for professional user experience
+# Chatbot interface
 def chatbot_interface():
-    st.title("🎓 Professional Student Well-being Chatbot")
-    st.write("Welcome to the Student Well-being Chatbot. I’m here to assist you in improving your mental well-being based on your responses. Please answer the following questions.")
+    st.title("🎓 Student Health and Stress Chatbot")
+    st.write("Hello! I’m here to assist you in assessing your well-being and provide helpful advice. Let's get started! 😊")
 
     # Initialize session state
     if 'step' not in st.session_state:
@@ -63,29 +62,30 @@ def chatbot_interface():
 
     # Step 0: Welcome message and initial question
     if st.session_state.step == 0:
-        st.write("Bot: Thank you for choosing to take this well-being assessment. We will gather some insights to provide personalized advice.")
-        st.write("Bot: Please answer the following questions as accurately as possible.")
-        next_step_button = st.button("Start Assessment")
+        st.write("Bot: Welcome to the Student Health and Stress Assessment. Let's start with some simple questions.")
+        start_button = st.button("Start Assessment")
         clear_button = st.button("Clear All")
-        if next_step_button:
+        if start_button:
             next_step()
         if clear_button:
             clear_data()
 
     # Step 1: Ask for anxiety level
     elif st.session_state.step == 1:
-        st.write("Bot: On a scale from 1 to 10, how would you rate your current anxiety level?")
+        st.write("Bot: On a scale of 1 to 10, how would you rate your current anxiety level?")
         anxiety_level = st.slider("Anxiety Level", 1, 10, 5)
-        if anxiety_level:
-            st.session_state.user_data['anxiety_level'] = anxiety_level
+        st.session_state.user_data['anxiety_level'] = anxiety_level
+        next_step_button = st.button("Next")
+        if next_step_button:
             next_step()
 
     # Step 2: Ask for self-esteem
     elif st.session_state.step == 2:
         st.write("Bot: On a scale from 1 to 10, how would you rate your current self-esteem?")
         self_esteem = st.slider("Self-esteem Level", 1, 10, 5)
-        if self_esteem:
-            st.session_state.user_data['self_esteem'] = self_esteem
+        st.session_state.user_data['self_esteem'] = self_esteem
+        next_step_button = st.button("Next")
+        if next_step_button:
             next_step()
 
     # Step 3: Ask for mental health history
@@ -96,30 +96,36 @@ def chatbot_interface():
             ["None", "Minor Issues", "Moderate Issues", "Severe Issues"]
         )
         st.session_state.user_data['mental_health_history'] = mental_health_history
-        next_step()
+        next_step_button = st.button("Next")
+        if next_step_button:
+            next_step()
 
     # Step 4: Ask for stress level
     elif st.session_state.step == 4:
         st.write("Bot: On a scale from 0 to 2, how would you rate your current stress level?")
-        stress_level = st.radio("Stress Level (0 = Low, 1 = Moderate, 2 = High)", [0, 1, 2])
+        stress_level = st.radio("Stress Level", [0, 1, 2])
         st.session_state.user_data['stress_level'] = stress_level
-        next_step()
+        next_step_button = st.button("Next")
+        if next_step_button:
+            next_step()
 
-    # Step 5: Show advice based on user input
+    # Step 5: Provide personalized advice based on responses
     elif st.session_state.step == 5:
-        st.write("Bot: Thank you for providing the information.")
+        st.write("Bot: Thank you for providing the information! Here's a summary of your responses:")
         user_data = st.session_state.user_data
-        st.write(f"Bot: Based on your inputs, here is a summary of your responses: {user_data}")
+        st.write(f"Bot: Your Anxiety Level: {user_data['anxiety_level']}")
+        st.write(f"Bot: Your Self-Esteem Level: {user_data['self_esteem']}")
+        st.write(f"Bot: Mental Health History: {user_data['mental_health_history']}")
+        st.write(f"Bot: Your Stress Level: {stress_level_to_string(user_data['stress_level'])}")
 
-        # Show professional, personalized advice
-        st.subheader("🔔 Professional Well-being Advice:")
+        st.subheader("🔔 Personalized Mental Health Advice:")
         advice = get_personalized_advice(user_data)
         if advice:
             st.write(f"Bot: {advice}")
         else:
-            st.write("Bot: Unfortunately, I couldn't retrieve specific advice at this moment, but feel free to try again later.")
-
-        # Option to restart the chat
+            st.write("Bot: I'm unable to retrieve personalized advice at the moment, but feel free to try again later.")
+        
+        # Option to restart or clear data
         restart_button = st.button("Start Over")
         clear_button = st.button("Clear All")
         if restart_button: