Update app.py

app.py

@@ -57,7 +57,7 @@ def chatbot_interface():
         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?"})
+            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."})
         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?"})
@@ -79,7 +79,17 @@ def chatbot_interface():
             st.markdown(f"**You**: {message['content']}")
 
     # User input field for chat interaction
-    user_input = st.text_input("Your message:", "")
+    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)
+        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)