Update app.py

app.py

@@ -1,48 +1,39 @@
 import streamlit as st
 from groq import Groq
 
-# Groq API Key (replace with your actual API key)
-groq_api_key = "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"
+# Initialize Groq API Client
+groq_api_key = "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"  # Replace with your actual Groq API key
+groq_client = Groq(api_key=groq_api_key)
 
-# Initialize Groq client
-client = Groq(api_key=groq_api_key)
-
-# Function to fetch personalized advice using Groq API
+# Function to get personalized advice from Groq API based on user data
 def get_personalized_advice(user_data):
-    """Fetch personalized advice based on user data."""
+    """Fetch personalized advice from Groq API."""
     query = f"""
-    Based on the following student's responses:
-    - 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 professional, personalized mental health advice.
+    Based on the following user 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 tailored advice to the user.
     """
-
+    
     try:
-        response = client.chat.completions.create(
+        response = groq_client.chat.completions.create(
             messages=[{"role": "user", "content": query}],
-            model="llama-3.3-70b-versatile",
+            model="llama-3.3-70b-versatile",  # You can use another model if needed
         )
-        return response.choices[0].message.content
+        advice = response.choices[0].message.content
+        return advice
     except Exception as e:
-        st.error(f"Error fetching advice: {str(e)}")
+        st.error(f"Error fetching advice from Groq: {str(e)}")
         return None
 
-def stress_level_to_string(stress_level):
-    """Convert stress level to string representation."""
-    if stress_level == 0:
-        return "Low"
-    elif stress_level == 1:
-        return "Moderate"
-    else:
-        return "High"
-
-# Chatbot interface
+# Chatbot interface with Streamlit
 def chatbot_interface():
+    """Create the chatbot interface."""
     st.title("🎓 Student Health and Stress Chatbot")
-    st.write("Hello! I’m here to assist you in assessing your well-being and provide helpful advice. Let's get started! 😊")
-
+    st.write("Hello! I'm here to assist you with personalized well-being advice based on your responses.")
+    
     # Initialize session state
     if 'step' not in st.session_state:
         st.session_state.step = 0
@@ -51,87 +42,96 @@ def chatbot_interface():
     def next_step():
         st.session_state.step += 1
 
-    def show_input_error(message="Please provide a valid input."):
-        st.error(message)
-        return False
-
-    def clear_data():
-        """Clear session data and reset to initial step."""
+    def restart_conversation():
         st.session_state.step = 0
         st.session_state.user_data = {}
 
-    # Step 0: Welcome message and initial question
+    # Chat history
+    if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
+
+    # Function to display chat messages
+    def display_chat(messages):
+        for message in messages:
+            st.chat_message(message['role']).markdown(message['content'])
+
+    # Step 0: Welcome message and initial prompt
     if st.session_state.step == 0:
-        st.write("Bot: Welcome to the Student Health and Stress Assessment. Let's start with some simple questions.")
-        start_button = st.button("Start Assessment")
-        clear_button = st.button("Clear All")
-        if start_button:
+        st.session_state.chat_history.append({"role": "assistant", "content": "Hi! I'm here to help you with your well-being. Let's start by discussing how you're feeling."})
+        display_chat(st.session_state.chat_history)
+        user_input = st.text_input("You:", key="input_1")
+        if user_input:
+            st.session_state.chat_history.append({"role": "user", "content": user_input})
             next_step()
-        if clear_button:
-            clear_data()
 
     # 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 current anxiety level?")
+        st.session_state.chat_history.append({"role": "assistant", "content": "On a scale of 1 to 10, how would you rate your anxiety level?"})
+        display_chat(st.session_state.chat_history)
         anxiety_level = st.slider("Anxiety Level", 1, 10, 5)
-        st.session_state.user_data['anxiety_level'] = anxiety_level
-        next_step_button = st.button("Next")
-        if next_step_button:
+        if st.button("Submit Anxiety Level"):
+            st.session_state.user_data['anxiety_level'] = anxiety_level
+            st.session_state.chat_history.append({"role": "user", "content": f"Anxiety Level: {anxiety_level}"})
             next_step()
 
-    # Step 2: Ask for self-esteem
+    # Step 2: Ask for self-esteem level
     elif st.session_state.step == 2:
-        st.write("Bot: On a scale from 1 to 10, how would you rate your current self-esteem?")
+        st.session_state.chat_history.append({"role": "assistant", "content": "On a scale of 1 to 10, how would you rate your self-esteem?"})
+        display_chat(st.session_state.chat_history)
         self_esteem = st.slider("Self-esteem Level", 1, 10, 5)
-        st.session_state.user_data['self_esteem'] = self_esteem
-        next_step_button = st.button("Next")
-        if next_step_button:
+        if st.button("Submit Self-Esteem Level"):
+            st.session_state.user_data['self_esteem'] = self_esteem
+            st.session_state.chat_history.append({"role": "user", "content": f"Self-Esteem Level: {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?")
+        st.session_state.chat_history.append({"role": "assistant", "content": "How would you describe your mental health history?"})
+        display_chat(st.session_state.chat_history)
         mental_health_history = st.selectbox(
             "Please select the option that best represents your mental health history:",
             ["None", "Minor Issues", "Moderate Issues", "Severe Issues"]
         )
-        st.session_state.user_data['mental_health_history'] = mental_health_history
-        next_step_button = st.button("Next")
-        if next_step_button:
+        if st.button("Submit Mental Health History"):
+            st.session_state.user_data['mental_health_history'] = mental_health_history
+            st.session_state.chat_history.append({"role": "user", "content": f"Mental Health History: {mental_health_history}"})
             next_step()
 
     # Step 4: Ask for stress level
     elif st.session_state.step == 4:
-        st.write("Bot: On a scale from 0 to 2, how would you rate your current stress level?")
+        st.session_state.chat_history.append({"role": "assistant", "content": "On a scale from 0 to 2, how would you rate your current stress level?"})
+        display_chat(st.session_state.chat_history)
         stress_level = st.radio("Stress Level", [0, 1, 2])
-        st.session_state.user_data['stress_level'] = stress_level
-        next_step_button = st.button("Next")
-        if next_step_button:
+        if st.button("Submit Stress Level"):
+            st.session_state.user_data['stress_level'] = stress_level
+            st.session_state.chat_history.append({"role": "user", "content": f"Stress Level: {stress_level}"})
             next_step()
 
-    # Step 5: Provide personalized advice based on responses
+    # Step 5: Display personalized advice based on user input
     elif st.session_state.step == 5:
-        st.write("Bot: Thank you for providing the information! Here's a summary of your responses:")
+        st.session_state.chat_history.append({"role": "assistant", "content": "Thank you for providing your responses! Here's a summary of your well-being:"})
+        display_chat(st.session_state.chat_history)
         user_data = st.session_state.user_data
-        st.write(f"Bot: Your Anxiety Level: {user_data['anxiety_level']}")
-        st.write(f"Bot: Your Self-Esteem Level: {user_data['self_esteem']}")
-        st.write(f"Bot: Mental Health History: {user_data['mental_health_history']}")
-        st.write(f"Bot: Your Stress Level: {stress_level_to_string(user_data['stress_level'])}")
+        st.write(f"Anxiety Level: {user_data['anxiety_level']}")
+        st.write(f"Self-Esteem Level: {user_data['self_esteem']}")
+        st.write(f"Mental Health History: {user_data['mental_health_history']}")
+        st.write(f"Stress Level: {user_data['stress_level']}")
 
+        # Generate personalized advice
         st.subheader("🔔 Personalized Mental Health Advice:")
         advice = get_personalized_advice(user_data)
         if advice:
             st.write(f"Bot: {advice}")
         else:
-            st.write("Bot: I'm unable to retrieve personalized advice at the moment, but feel free to try again later.")
-        
-        # Option to restart or clear data
+            st.write("Bot: I'm unable to provide personalized advice at the moment.")
+
+        # Option to restart the chat or clear all data
         restart_button = st.button("Start Over")
-        clear_button = st.button("Clear All")
+        clear_button = st.button("Clear All Data")
         if restart_button:
-            clear_data()
+            restart_conversation()
         if clear_button:
-            clear_data()
+            restart_conversation()
 
 # Main function to run the chatbot
 def main():