Update app.py

app.py

@@ -2,10 +2,12 @@ import streamlit as st
 import pandas as pd
 import os
 from groq import Groq
-import requests
 
-# Set the API key directly or from environment variable
-api_key = os.getenv("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")  # Ensure your API key is stored in the environment variable.
+# Set the GROQ API key from environment variable
+groq_api_key = os.getenv("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")  # Make sure the GROQ API key is set in your environment
+
+# Initialize the GROQ client
+groq_client = Groq(api_key=groq_api_key)
 
 # Function to load and preprocess data
 @st.cache_data
@@ -39,27 +41,24 @@ def provide_observed_advice(data):
 
     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
-
+# Function to use the GROQ API for health queries (e.g., for symptoms or wellness queries)
+def get_health_advice_from_groq(query):
     try:
-        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 []
+        # 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():
@@ -128,15 +127,10 @@ def main():
             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.")
+            # 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()