Update app.py

app.py

@@ -6,6 +6,10 @@ from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 from langchain_google_genai import ChatGoogleGenerativeAI
 
+key = os.environ.get("GOOGLE_API_KEY")
+if not key:
+    st.error("❌ GOOGLE_API_KEY not found. Please set it in your Hugging Face Secrets.")
+    st.stop()
 
 # Set page config
 st.set_page_config(page_title="Company Info Agent", layout="wide")
@@ -67,7 +71,7 @@ tools = [search_tool]
 
 # --- Initializing the LLM
 llm = ChatGoogleGenerativeAI(
-    api_key = GOOGLE_API_KEY,
+    api_key = key,
     model = "gemini-2.0-flash"
 )