Update app.py

app.py

@@ -10,7 +10,7 @@ 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."""
-    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."
+    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:
         response = client.chat.completions.create(
@@ -22,7 +22,6 @@ def get_personalized_advice(user_data):
         st.error(f"Error fetching advice from Groq: {str(e)}")
         return None
 
-# Function to convert numerical stress level to string form
 def stress_level_to_string(stress_level):
     """Convert numerical stress level (0, 1, 2) to a string representation."""
     if stress_level == 0:
@@ -32,16 +31,11 @@ def stress_level_to_string(stress_level):
     else:
         return "High"
 
-# Function for chatbot interaction with professional styling
+# Chatbot interface for conversation-like experience
 def chatbot_interface():
-    st.set_page_config(page_title="🎓 Student Well-being Chatbot", layout="centered")
     st.title("🎓 Student Well-being Chatbot")
-    st.write("Welcome! I'm here to help you with well-being advice tailored to your current state.")
-
-    # Styling the sidebar for improved UX
-    st.sidebar.title("Student Well-being")
-    st.sidebar.write("The well-being questionnaire will guide you through assessing your mental health and providing personalized advice.")
-
+    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
@@ -56,62 +50,61 @@ def chatbot_interface():
 
     # Step 0: Welcome message and initial question
     if st.session_state.step == 0:
-        st.write("Bot: Hi there! I'm here to assist you with your well-being. Let's get started! 😊")
-        st.markdown("<hr>", unsafe_allow_html=True)
-        next_step_button = st.button("Start Chat", key="start_button", help="Click to start the well-being questionnaire.")
+        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")
         if next_step_button:
             next_step()
 
-    # Step 1: Ask for anxiety level with slider
+    # Step 1: Ask for anxiety level
     elif st.session_state.step == 1:
         st.write("Bot: On a scale of 1 to 10, how would you rate your anxiety level?")
-        anxiety_level = st.slider("Your Anxiety Level:", min_value=1, max_value=10, value=5, step=1, help="Rate your anxiety from 1 (low) to 10 (high).")
-        st.session_state.user_data['anxiety_level'] = anxiety_level
-        next_step()
+        anxiety_level = st.slider("Anxiety level (1-10)", 1, 10, 5)
+        if anxiety_level:
+            st.session_state.user_data['anxiety_level'] = anxiety_level
+            next_step()
 
-    # Step 2: Ask for self-esteem with slider
+    # Step 2: Ask for self-esteem
     elif st.session_state.step == 2:
         st.write("Bot: On a scale of 1 to 10, how would you rate your self-esteem?")
-        self_esteem = st.slider("Your Self-Esteem Level:", min_value=1, max_value=10, value=5, step=1, help="Rate your self-esteem from 1 (low) to 10 (high).")
-        st.session_state.user_data['self_esteem'] = self_esteem
-        next_step()
+        self_esteem = st.slider("Self-esteem level (1-10)", 1, 10, 5)
+        if self_esteem:
+            st.session_state.user_data['self_esteem'] = self_esteem
+            next_step()
 
     # Step 3: Ask for mental health history
     elif st.session_state.step == 3:
         st.write("Bot: How would you describe your mental health history?")
         mental_health_history = st.selectbox(
-            "Select your mental health history:",
-            ["None", "Minor Issues", "Moderate Issues", "Severe Issues"],
-            help="Choose an option that best describes your mental health background."
+            "Select an option:",
+            ["None", "Minor Issues", "Moderate Issues", "Severe Issues"]
         )
         st.session_state.user_data['mental_health_history'] = mental_health_history
         next_step()
 
-    # Step 4: Ask for stress level with slider
+    # Step 4: Ask for stress level
     elif st.session_state.step == 4:
         st.write("Bot: On a scale of 0 to 2, how would you rate your stress level?")
-        stress_level = st.slider("Your Stress Level:", min_value=0, max_value=2, value=1, step=1, help="Rate your stress from 0 (low) to 2 (high).")
-        stress_level_string = stress_level_to_string(stress_level)  # Convert to string
-        st.session_state.user_data['stress_level'] = stress_level_string
+        stress_level = st.radio("Stress level (0 = Low, 1 = Moderate, 2 = High)", [0, 1, 2])
+        st.session_state.user_data['stress_level'] = stress_level
         next_step()
 
     # Step 5: Show advice based on user input
     elif st.session_state.step == 5:
-        st.write("Bot: Thank you for providing the information! Here’s your personalized advice:")
+        st.write("Bot: Thank you for providing the information!")
         user_data = st.session_state.user_data
-        st.write(f"Your Data: {user_data}")
+        st.write(f"Bot: Here's a summary of your responses: {user_data}")
 
-        # Fetch personalized advice from Groq
+        # 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)
+            st.write(f"Bot: {advice}")
         else:
-            st.write("I couldn't retrieve specific advice right now, but please check back later.")
+            st.write("Bot: I couldn't retrieve specific advice right now, but please check back later.")
 
         # Option to restart the chat
-        st.markdown("<hr>", unsafe_allow_html=True)
-        restart_button = st.button("Start Over", key="restart_button", help="Click to restart the questionnaire.")
+        restart_button = st.button("Start Over")
         if restart_button:
             st.session_state.step = 0
             st.session_state.user_data = {}