Update app.py

app.py

@@ -1,53 +1,67 @@
-import os
-import logging
-import pandas as pd
 import streamlit as st
+import pandas as pd
 import requests
-from transformers import pipeline
-
-# Set up logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+from groq import Groq
 
-# Load environment variables from .env file
-load_dotenv()
+# Set the API key directly or from environment variable
+api_key = "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"  # Replace with your actual demo API key
 
-# Get the Hugging Face API key from environment variables
-hf_api_key = os.getenv("HUGGINGFACE_API_KEY")
-if not hf_api_key:
-    raise ValueError("HUGGINGFACE_API_KEY is not set. Please provide a valid API key.")
-
-# Hugging Face API URL
-hf_api_url = "https://api-inference.huggingface.co/models/{model_name}"
+# Initialize the Groq client with the API key
+client = Groq(api_key=api_key)
 
 # Function to load and preprocess data
 @st.cache_data
 def load_data(file):
-    try:
-        df = pd.read_csv(file)
-        return df
-    except Exception as e:
-        logger.error(f"Error loading CSV file: {e}")
-        st.error("There was an issue loading the file. Please try again.")
-        return pd.DataFrame()  # Return an empty DataFrame in case of error
-
-# Function to call Hugging Face API for text generation
-def generate_text_from_model(model_name, text_input):
-    headers = {"Authorization": f"Bearer {hf_api_key}"}
-    data = {"inputs": text_input}
+    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 fetch health articles from the GROC API based on the query
+def get_health_articles(query):
+    url = f"https://api.groc.com/search?q={query}"
+    headers = {"Authorization": f"Bearer {api_key}"}  # Use the demo API key in the header
 
     try:
-        response = requests.post(hf_api_url.format(model_name=model_name), headers=headers, json=data)
-        response.raise_for_status()
-        result = response.json()
-        if 'generated_text' in result:
-            return result['generated_text']
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()  # This will raise an HTTPError for bad responses
+        data = response.json()  # Assuming the API returns JSON
+        if 'results' in data:
+            articles = [{"title": item["title"], "url": item["url"]} for item in data['results']]
         else:
-            return "No result from model. Please try again."
+            articles = []
+        return articles
+    except requests.exceptions.HTTPError as http_err:
+        st.error(f"HTTP error occurred: {http_err}. Please check your API key and the endpoint.")
+        st.error(f"Response content: {response.text}")
+        return []
     except requests.exceptions.RequestException as err:
-        logger.error(f"Error interacting with Hugging Face API: {err}")
-        st.error(f"Error interacting with Hugging Face API: {err}")
-        return ""
+        st.error(f"Error fetching articles: {err}. Please check your internet connection.")
+        return []
 
 # Streamlit app layout
 def main():
@@ -75,18 +89,56 @@ def main():
     )
 
     # Title and header
-    st.title("🌟 **Hugging Face Text Generation** 🌟")
-    st.markdown("### **Generate text using Hugging Face Models**")
-
-    # User input for text generation
-    model_name = st.selectbox("🔹 Select Hugging Face Model", ["gpt2", "distilgpt2", "t5-small"])
-    text_input = st.text_area("🔹 Input Text", "Once upon a time...")
-
-    # Generate text based on input
-    if st.button("🔍 Generate Text"):
-        st.subheader("🔔 **Generated Text** 🔔")
-        generated_text = generate_text_from_model(model_name, text_input)
-        st.write(f"📜 {generated_text}")
+    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 related health articles based on user input
+            st.subheader("📰 **Related Health Articles** 📰")
+            query = "mental health anxiety depression isolation stress relief"
+            articles = get_health_articles(query)
+            if articles:
+                for article in articles:
+                    st.write(f"🌐 [{article['title']}]({article['url']})")
+            else:
+                st.write("No articles found. Please check your API key or internet connection.")
 
 if __name__ == "__main__":
     main()