Update app.py

app.py

@@ -1,10 +1,14 @@
 import streamlit as st
 import pandas as pd
-import requests
+import os
 from groq import Groq
+import requests
 
 # Set the API key directly or from environment variable
-api_key = "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"  # Replace with your actual demo API key
+api_key = os.getenv("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")  # Ensure your API key is stored in the environment variable.
+
+# Initialize the Groq client with the API key
+client = Groq(api_key=api_key)
 
 # Function to load and preprocess data
 @st.cache_data
@@ -44,13 +48,14 @@ def get_health_articles(query):
     headers = {"Authorization": f"Bearer {api_key}"}  # Use the demo API key in the header
 
     try:
-        response = requests.get(url, headers=headers, timeout=10)  # Timeout after 10 seconds
+        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:
-            return [{"title": item["title"], "url": item["url"]} for item in data['results']]
+            articles = [{"title": item["title"], "url": item["url"]} for item in data['results']]
         else:
-            return []
+            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}")