app.py

import os
import streamlit as st
import pandas as pd
from groq import Groq
from dotenv import load_dotenv

# Load environment variables from .env file (optional)
load_dotenv()

# Get the API key from environment variable or set directly
groq_api_key = os.getenv("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")  # First try from environment variable
if not groq_api_key:
    # Optionally, set the API key directly for testing
    groq_api_key = "your_actual_groq_api_key_here"

if not groq_api_key:
    st.error("GROQ_API_KEY is not set. Please provide a valid API key.")
    st.stop()

# Initialize the GROQ client
groq_client = Groq(api_key=groq_api_key)

if not groq_client:
    st.error("Failed to initialize the GROQ client. Please check your API key.")
    st.stop()

# Function to load and preprocess data
@st.cache_data
def load_data(file):
    df = pd.read_csv(file)
    return df

# Function to provide detailed health advice based on user data
def provide_observed_advice(data):
    advice = []

    # High depression and anxiety with low stress-relief activities
    if data['depression'] > 7 and data['anxiety'] > 7:
        advice.append("You seem to be experiencing high levels of both depression and anxiety. It's important to consider professional mental health support. You might also benefit from engaging in calming activities like deep breathing, mindfulness, or yoga.")

    # Moderate depression or anxiety
    elif data['depression'] > 5 or data['anxiety'] > 5:
        advice.append("You are showing moderate levels of depression or anxiety. It would be helpful to develop healthy coping strategies like maintaining a regular sleep schedule, engaging in physical activity, and reaching out to friends or family for support.")

    # High isolation and low stress-relief activities
    if data['isolation'] > 7 and data['stress_relief_activities'] < 5:
        advice.append("It seems you are feeling isolated, and your engagement in stress-relief activities is low. It's important to connect with friends or join community groups. Incorporate activities that help alleviate stress, such as walking, journaling, or meditation.")

    # High future insecurity
    if data['future_insecurity'] > 7:
        advice.append("You are feeling a significant amount of insecurity about the future. It can be helpful to break down your larger goals into smaller, manageable tasks. Seeking career counseling or mentorship could provide valuable guidance and reduce anxiety about the future.")

    # Overall low engagement in stress-relief activities
    if data['stress_relief_activities'] < 5:
        advice.append("Your engagement in stress-relief activities is quite low. It's essential to engage in activities that reduce stress and promote mental wellness, such as hobbies, physical exercise, and relaxation techniques like deep breathing or yoga.")

    return advice

# Function to use the GROQ API for health queries (e.g., for symptoms or wellness queries)
def get_health_advice_from_groq(query):
    try:
        # Sending request to GROQ API
        chat_completion = groq_client.chat.completions.create(
            messages=[{
                "role": "user",
                "content": query
            }],
            model="llama-3.3-70b-versatile",
        )

        # Extracting and returning the response from GROQ API
        response_content = chat_completion.choices[0].message.content
        return response_content
    except Exception as e:
        st.error(f"Error with GROQ API: {str(e)}")
        return "Could not fetch advice from GROQ API."

# Streamlit app layout
def main():
    # Set a background color and style
    st.markdown(
        """
        <style>
        .stApp {
            background-color: #F4F4F9;
        }
        .stButton>button {
            background-color: #6200EE;
            color: white;
            font-size: 18px;
        }
        .stSlider>div>div>span {
            color: #6200EE;
        }
        .stTextInput>div>div>input {
            background-color: #E0E0E0;
        }
        </style>
        """,
        unsafe_allow_html=True
    )

    # Title and header
    st.title("🌟 **Student Health Advisory Assistant** 🌟")
    st.markdown("### **Analyze your well-being and get personalized advice**")

    # File upload
    uploaded_file = st.file_uploader("Upload your dataset (CSV)", type=["csv"])
    if uploaded_file:
        df = load_data(uploaded_file)
        st.write("### Dataset Preview:")
        st.dataframe(df.head())

        # User input for analysis
        st.markdown("### **Input Your Details**")
        gender = st.selectbox("🔹 Gender", ["Male", "Female"], help="Select your gender.")
        age = st.slider("🔹 Age", 18, 35, step=1)
        depression = st.slider("🔹 Depression Level (1-10)", 1, 10)
        anxiety = st.slider("🔹 Anxiety Level (1-10)", 1, 10)
        isolation = st.slider("🔹 Isolation Level (1-10)", 1, 10)
        future_insecurity = st.slider("🔹 Future Insecurity Level (1-10)", 1, 10)
        stress_relief_activities = st.slider("🔹 Stress Relief Activities Level (1-10)", 1, 10)

        # Data dictionary for advice
        user_data = {
            "gender": gender,
            "age": age,
            "depression": depression,
            "anxiety": anxiety,
            "isolation": isolation,
            "future_insecurity": future_insecurity,
            "stress_relief_activities": stress_relief_activities,
        }

        # Provide advice based on user inputs
        if st.button("🔍 Get Observed Advice", key="advice_btn"):
            st.subheader("🔔 **Health Advice Based on Observations** 🔔")
            advice = provide_observed_advice(user_data)
            if advice:
                for i, tip in enumerate(advice, 1):
                    st.write(f"📌 {i}. {tip}")
            else:
                st.warning("No advice available based on your inputs.")

            # Fetch additional health advice from GROQ API
            st.subheader("💬 **Additional Health Advice from GROQ** 💬")
            groq_advice = get_health_advice_from_groq("Provide wellness tips for managing anxiety and stress.")
            st.write(f"💡 {groq_advice}")

if __name__ == "__main__":
    main()