Update app.py

app.py

@@ -1,137 +1,106 @@
 import streamlit as st
 from groq import Groq
 
-# Initialize Groq API Client
-groq_api_key = "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"  # Replace with your actual Groq API key
+# Replace with your Groq API key
+groq_api_key = "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"
+
+# Initialize Groq client
 groq_client = Groq(api_key=groq_api_key)
 
-# Function to get personalized advice from Groq API based on user data
+# Function to generate personalized advice based on user input
 def get_personalized_advice(user_data):
-    """Fetch personalized advice from Groq API."""
     query = f"""
-    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.
+    Based on the following information about the user:
+    Anxiety Level: {user_data['anxiety_level']}
+    Self-Esteem Level: {user_data['self_esteem']}
+    Mental Health History: {user_data['mental_health_history']}
+    Stress Level: {user_data['stress_level']}
+
+    Please provide personalized advice for the user's well-being.
     """
-    
     try:
         response = groq_client.chat.completions.create(
             messages=[{"role": "user", "content": query}],
-            model="llama-3.3-70b-versatile",  # You can use another model if needed
+            model="llama-3.3-70b-versatile"  # Use the appropriate model
         )
-        advice = response.choices[0].message.content
-        return advice
+        return response.choices[0].message.content
     except Exception as e:
-        st.error(f"Error fetching advice from Groq: {str(e)}")
-        return None
+        st.error(f"Error: {str(e)}")
+        return "Sorry, I couldn't retrieve personalized advice at the moment."
 
-# Chatbot interface with Streamlit
+# Function to display the chatbot interface
 def chatbot_interface():
-    """Create the chatbot interface."""
-    st.title("🎓 Student Health and Stress Chatbot")
-    st.write("Hello! I'm here to assist you with personalized well-being advice based on your responses.")
-    
+    st.title("🎓 Professional Student Well-being Chatbot")
+    st.markdown("#### Let's talk about how you're feeling today. I'm here to assist you with your mental well-being.")
+
     # Initialize session state
     if 'step' not in st.session_state:
         st.session_state.step = 0
         st.session_state.user_data = {}
+        st.session_state.chat_history = []
+
+    # Display chat history (simulating chatbot-style interaction)
+    for message in st.session_state.chat_history:
+        st.chat_message(message["role"]).markdown(message["content"])
 
     def next_step():
         st.session_state.step += 1
 
-    def restart_conversation():
+    def restart_chat():
         st.session_state.step = 0
         st.session_state.user_data = {}
-
-    # 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
+    # Step 1: Greet the user
     if st.session_state.step == 0:
-        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")
+        st.session_state.chat_history.append({"role": "assistant", "content": "Hello! How are you feeling today? 😊 I'm here to assist you with your mental well-being."})
+        user_input = st.text_input("You:", "")
         if user_input:
             st.session_state.chat_history.append({"role": "user", "content": user_input})
             next_step()
 
-    # Step 1: Ask for anxiety level
+    # Step 2: Form to collect data (Anxiety Level, Self-Esteem, Mental Health, Stress)
     elif st.session_state.step == 1:
-        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)
-        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 level
+        st.session_state.chat_history.append({"role": "assistant", "content": "Please fill out the form below to help me understand your current well-being."})
+        
+        with st.form("Well-being Form"):
+            anxiety_level = st.slider("Anxiety Level (1 to 10)", 1, 10, 5)
+            self_esteem = st.slider("Self-esteem Level (1 to 10)", 1, 10, 5)
+            mental_health_history = st.selectbox("Mental Health History", ["None", "Minor Issues", "Moderate Issues", "Severe Issues"])
+            stress_level = st.radio("Stress Level (0 to 2)", [0, 1, 2])
+            
+            submit_button = st.form_submit_button(label="Submit")
+            
+            if submit_button:
+                # Collecting user input in session state
+                st.session_state.user_data = {
+                    "anxiety_level": anxiety_level,
+                    "self_esteem": self_esteem,
+                    "mental_health_history": mental_health_history,
+                    "stress_level": stress_level
+                }
+                st.session_state.chat_history.append({"role": "user", "content": f"Form Submitted:\nAnxiety Level: {anxiety_level}\nSelf-esteem Level: {self_esteem}\nMental Health History: {mental_health_history}\nStress Level: {stress_level}"})
+                next_step()
+
+    # Step 3: Provide Personalized Advice
     elif st.session_state.step == 2:
-        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)
-        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.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"]
-        )
-        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.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])
-        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: Display personalized advice based on user input
-    elif st.session_state.step == 5:
-        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)
+        st.session_state.chat_history.append({"role": "assistant", "content": "Thank you for providing the information! Here's a summary of your well-being:"})
         user_data = st.session_state.user_data
         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:")
+        # Provide personalized advice
+        st.subheader("🔔 Personalized Well-being Advice:")
         advice = get_personalized_advice(user_data)
         if advice:
             st.write(f"Bot: {advice}")
         else:
-            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 Data")
-        if restart_button:
-            restart_conversation()
-        if clear_button:
-            restart_conversation()
+            st.write("Bot: I couldn't retrieve personalized advice at the moment.")
+
+        if st.button("Start Over"):
+            restart_chat()
 
 # Main function to run the chatbot
 def main():