Spaces:
Sleeping
Sleeping
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 | |
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() | |