Update app.py

app.py

@@ -22,99 +22,95 @@ def get_personalized_advice(user_data):
         st.error(f"Error fetching advice from Groq: {str(e)}")
         return None
 
-def stress_level_to_string(stress_level):
-    """Convert numerical stress level (0, 1, 2) to a string representation."""
-    if stress_level == 0:
-        return "Low"
-    elif stress_level == 1:
-        return "Moderate"
-    else:
-        return "High"
-
-# Chatbot interface for conversation-like experience
+# Function for chatbot interaction
 def chatbot_interface():
     st.title("🎓 Student Well-being Chatbot")
     st.write("Hello! I’m here to assist you with well-being advice based on your responses.")
-
-    # Initialize session state
-    if 'messages' not in st.session_state:
-        st.session_state.messages = [
-            {"role": "bot", "content": "Hi there! I'm here to help with your well-being. Let's start! 😊"}
-        ]
     
-    def next_step(user_input=None):
-        # Save the user's message to the chat history
-        if user_input:
-            st.session_state.messages.append({"role": "user", "content": user_input})
-        
-        # Proceed to next step in the conversation
-        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:
-            # Validate anxiety level input
+    # Initialize session state
+    if 'step' not in st.session_state:
+        st.session_state.step = 0
+        st.session_state.user_data = {}
+
+    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("Bot: Hi there! I'm here to help with your well-being. Let's start! 😊")
+        next_step_button = st.button("Start Chat")
+        if next_step_button:
+            next_step()
+
+    # 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.text_input("Your Anxiety Level:", "")
+        if anxiety_level:
             try:
-                anxiety_level = int(user_input)
+                anxiety_level = int(anxiety_level)
                 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?"})
+                st.session_state.user_data['anxiety_level'] = anxiety_level
+                next_step()
             except ValueError:
-                st.error("Please enter a valid number between 1 and 10 for anxiety level.")
-        elif len(st.session_state.messages) == 3:
-            # Validate self-esteem input
+                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:
+        st.write("Bot: On a scale of 1 to 10, how would you rate your self-esteem?")
+        self_esteem = st.text_input("Your Self-Esteem Level:", "")
+        if self_esteem:
             try:
-                self_esteem = int(user_input)
+                self_esteem = int(self_esteem)
                 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."})
+                next_step()
             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:
-            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?"})
-
-    # Display conversation history
-    for message in st.session_state.messages:
-        if message['role'] == 'bot':
-            st.markdown(f"**Bot**: {message['content']}")
-        else:
-            st.markdown(f"**You**: {message['content']}")
-
-    # User input field for chat interaction
-    user_input = ""
-    if len(st.session_state.messages) == 3:
-        # Mental health history (checkbox options)
-        mental_health_options = ["None", "Minor Issues", "Moderate Issues", "Severe Issues"]
-        mental_health_history = st.multiselect("Select your mental health history:", mental_health_options)
+                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:
+        st.write("Bot: How would you describe your mental health history?")
+        mental_health_history = st.multiselect(
+            "Select your mental health history:",
+            ["None", "Minor Issues", "Moderate Issues", "Severe Issues"]
+        )
         if mental_health_history:
-            user_input = ", ".join(mental_health_history)
-    elif len(st.session_state.messages) == 4:
-        # Stress level (radio button)
-        user_input = st.radio("On a scale of 0 to 2, how would you rate your stress level?", [0, 1, 2])
-    
-    if user_input:
-        next_step(user_input)
+            st.session_state.user_data['mental_health_history'] = ", ".join(mental_health_history)
+            next_step()
+
+    # 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.radio("Your Stress Level:", [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:")
+        user_data = st.session_state.user_data
+        st.write(f"Your data: {user_data}")
+
+        # Fetch personalized advice from Groq
+        st.subheader("🔔 Here's Your Personalized Well-being Advice:")
+        advice = get_personalized_advice(user_data)
+        if advice:
+            st.write(advice)
+        else:
+            st.write("I couldn't retrieve specific advice right now, but please check back later.")
 
-    # Option to restart the conversation
-    if len(st.session_state.messages) >= 6:
+        # Option to restart the chat
         restart_button = st.button("Start Over")
         if restart_button:
-            st.session_state.messages = [{"role": "bot", "content": "Hi there! I'm here to help with your well-being. Let's start! 😊"}]
+            st.session_state.step = 0
+            st.session_state.user_data = {}
 
 # Main function to run the chatbot
 def main():