Update app.py

app.py

@@ -1,29 +1,20 @@
 import os
 import streamlit as st
 import pandas as pd
+import requests
 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"
+# Fetch the GROQ API key from the environment variable
+groq_api_key = os.getenv("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")
 
+# Check if the API key is available
 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
+# Initialize the GROQ client with the fetched API key
 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):
@@ -56,24 +47,27 @@ def provide_observed_advice(data):
 
     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):
+# 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 {groq_api_key}"}  # Use the demo API key in the header
+
     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."
+        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:
+            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:
+        st.error(f"Error fetching articles: {err}. Please check your internet connection.")
+        return []
 
 # Streamlit app layout
 def main():
@@ -142,10 +136,15 @@ def main():
             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}")
+            # 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()