Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 import streamlit as st
 import pandas as pd
 from groq import Groq
@@ -9,13 +8,9 @@ groq_api_key = "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"
 # Initialize Groq client
 client = Groq(api_key=groq_api_key)
 
-@st.cache_data
-def load_data(file):
-    return pd.read_csv(file)
-
+# Define a 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"""
-    # Construct the query for the Groq API
     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:
@@ -37,92 +32,73 @@ def stress_level_to_string(stress_level):
     else:
         return "High"
 
+def chatbot_interface():
+    st.title("🎓 Student Well-being Chatbot")
+    st.write("I can help you with personalized well-being advice based on your responses. Let's get started!")
+
+    # Initializing 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
+
+    # Step 0: Welcome message and initial question
+    if st.session_state.step == 0:
+        st.write("Hi! I am your well-being assistant. I'll ask you some questions to understand how you're feeling.")
+        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:
+            st.session_state.user_data['anxiety_level'] = int(anxiety_level)
+            next_step()
+
+    # 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:
+            st.session_state.user_data['self_esteem'] = int(self_esteem)
+            next_step()
+
+    # 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
+        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
+        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}")
+
+        st.subheader("🔔 Personalized Health Advice")
+        advice = get_personalized_advice(user_data)
+        if advice:
+            st.write(advice)
+        else:
+            st.write("No advice available at the moment.")
+
+        # Option to restart the chat
+        restart_button = st.button("Restart Chat")
+        if restart_button:
+            st.session_state.step = 0
+            st.session_state.user_data = {}
+
+# Main function to run the chatbot
 def main():
-    st.set_page_config(
-        page_title="Student Well-being Advisor",
-        page_icon="📊",
-        layout="wide",
-        initial_sidebar_state="expanded",
-    )
-
-    st.sidebar.title("Navigation")
-    st.sidebar.write("Use the sidebar to navigate through the app.")
-    st.sidebar.markdown("### 📂 Upload Data")
-    st.sidebar.write("Start by uploading your dataset for analysis.")
-    st.sidebar.markdown("### 📊 Analysis & Advice")
-    st.sidebar.write("Get detailed insights and personalized advice.")
-
-    st.title("🎓 Student Well-being Advisor")
-    st.subheader("Analyze data and provide professional mental health recommendations.")
-    st.write("""
-        This app helps identify areas of concern in students' well-being and provides personalized advice based on their responses.
-    """)
-
-    # Dynamic Data Input
-    st.markdown("## 📋 Enter Your Data")
-    st.write("Please fill out the form below to receive personalized well-being advice.")
-
-    # Alternative form inputs for user data
-    anxiety_level = st.text_input("Anxiety Level (1 to 10)", "5")
-    self_esteem = st.text_input("Self-esteem (1 to 10)", "5")
-    mental_health_history = st.selectbox("Mental Health History", ["None", "Minor Issues", "Moderate Issues", "Severe Issues"])
-    depression = st.text_input("Depression Level (1 to 10)", "5")
-    headache = st.text_input("Headache Level (1 to 10)", "5")
-    blood_pressure = st.text_input("Blood Pressure (1 to 10)", "5")
-    sleep_quality = st.text_input("Sleep Quality (1 to 10)", "5")
-    breathing_problem = st.selectbox("Breathing Problem", ["None", "Mild", "Severe"])
-    noise_level = st.text_input("Noise Level (1 to 10)", "5")
-    living_conditions = st.text_input("Living Conditions (1 to 10)", "5")
-    safety = st.text_input("Safety (1 to 10)", "5")
-    basic_needs = st.text_input("Basic Needs Satisfaction (1 to 10)", "5")
-    academic_performance = st.text_input("Academic Performance (1 to 10)", "5")
-    study_load = st.text_input("Study Load (1 to 10)", "5")
-    teacher_student_relationship = st.text_input("Teacher-Student Relationship (1 to 10)", "5")
-    future_career_concerns = st.text_input("Future Career Concerns (1 to 10)", "5")
-    social_support = st.text_input("Social Support (1 to 10)", "5")
-    peer_pressure = st.text_input("Peer Pressure (1 to 10)", "5")
-    extracurricular_activities = st.text_input("Extracurricular Activities (1 to 10)", "5")
-    bullying = st.text_input("Bullying Level (1 to 10)", "5")
-    stress_level = st.radio("Stress Level", [0, 1, 2], index=1)
-
-    # Convert text inputs to integers
-    try:
-        user_data = {
-            "anxiety_level": int(anxiety_level),
-            "self_esteem": int(self_esteem),
-            "mental_health_history": mental_health_history,
-            "depression": int(depression),
-            "headache": int(headache),
-            "blood_pressure": int(blood_pressure),
-            "sleep_quality": int(sleep_quality),
-            "breathing_problem": breathing_problem,
-            "noise_level": int(noise_level),
-            "living_conditions": int(living_conditions),
-            "safety": int(safety),
-            "basic_needs": int(basic_needs),
-            "academic_performance": int(academic_performance),
-            "study_load": int(study_load),
-            "teacher_student_relationship": int(teacher_student_relationship),
-            "future_career_concerns": int(future_career_concerns),
-            "social_support": int(social_support),
-            "peer_pressure": int(peer_pressure),
-            "extracurricular_activities": int(extracurricular_activities),
-            "bullying": int(bullying),
-            "stress_level": stress_level
-        }
-    except ValueError:
-        st.error("Please enter valid numeric values.")
-        return
-
-    st.write(f"### Selected User Data: {user_data}")
-
-    # Provide personalized advice
-    st.subheader("🔔 Health Advice Based on Your Input")
-    advice = get_personalized_advice(user_data)
-    if advice:
-        st.write(advice)
-    else:
-        st.warning("No advice available at the moment.")
+    chatbot_interface()
 
 if __name__ == "__main__":
     main()