Update app.py

app.py

@@ -2,8 +2,12 @@ import os
 import streamlit as st
 from groq import Groq
 
-# Set your Groq API key (preferably via environment variable for security)
-GROQ_API_KEY = os.environ.get("GROQ_API_KEY") or "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"  # Fallback only for testing
+# Fetch the Groq API key from environment variables set in Hugging Face Space
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+
+if not GROQ_API_KEY:
+    st.error("Error: GROQ_API_KEY environment variable is not set.")
+    st.stop()  # Stop execution if API key is not found
 
 # Initialize Groq client
 try:
@@ -50,8 +54,8 @@ for chat in st.session_state.chat_history:
 
 # Sidebar information
 st.sidebar.title("Settings")
-st.sidebar.write("API Key Source: " + ("Environment Variable" if os.environ.get("GROQ_API_KEY") else "Directly in Code (Not Recommended)"))
-if os.environ.get("GROQ_API_KEY"):
+st.sidebar.write("API Key Source: Environment Variable (Recommended for security)")
+if GROQ_API_KEY:
     st.sidebar.success("Groq Client Initialized Successfully!")
 else:
-    st.sidebar.warning("Using API Key Directly in Code (Not Recommended). Set GROQ_API_KEY Environment Variable for Security.")
+    st.sidebar.warning("Using API Key Directly in Code (Not Recommended). Please set GROQ_API_KEY environment variable.")