Update app.py

app.py

@@ -37,86 +37,57 @@ def chatbot_interface():
     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 = {}
-
-    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'm here to assist you with well-being advice based on your responses. Let's get started!")
-        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:
-        anxiety_level = st.text_input("On a scale of 1 to 10, how would you rate your anxiety level?")
-        if anxiety_level:
-            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
-                st.write(f"Got it! Your anxiety level is {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:
-            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
-                st.write(f"Got it! Your self-esteem level is {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"])
-        st.session_state.user_data['mental_health_history'] = mental_health_history
-        st.write(f"Got it! Your mental health history is: {mental_health_history}.")
-        next_step()
-
-    # Step 4: Ask for stress level
-    elif st.session_state.step == 4:
-        stress_level = st.radio("On a scale of 0 to 2, how would you rate your stress level?", [0, 1, 2])
-        st.session_state.user_data['stress_level'] = stress_level
-        st.write(f"Got it! Your stress level is: {stress_level_to_string(stress_level)}.")
-        next_step()
-
-    # Step 5: Show advice based on user input
-    elif st.session_state.step == 5:
-        st.write("Thank you for providing the information!")
-        user_data = st.session_state.user_data
-        st.write(f"Your data: {user_data}")
-
-        # 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)
+    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:
+            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?"})
+        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}. How would you describe your mental health history?"})
+        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."})
+        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.write("I couldn't retrieve specific advice right now, but please check back later.")
+            st.markdown(f"**You**: {message['content']}")
+
+    # User input field for chat interaction
+    user_input = st.text_input("Your message:", "")
+    if user_input:
+        next_step(user_input)
 
-        # Option to restart the chat
+    # Option to restart the conversation
+    if len(st.session_state.messages) >= 6:
         restart_button = st.button("Start Over")
         if restart_button:
-            st.session_state.step = 0
-            st.session_state.user_data = {}
+            st.session_state.messages = [{"role": "bot", "content": "Hi there! I'm here to help with your well-being. Let's start! 😊"}]
 
 # Main function to run the chatbot
 def main():