Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+from dotenv import load_dotenv
 import streamlit as st
 from langchain_groq import ChatGroq
 from langchain.document_loaders import PyPDFLoader
@@ -7,10 +8,18 @@ from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.vectorstores import FAISS
 from langchain.chains import RetrievalQA
 
+# Load environment variables from .env file
+load_dotenv()
+
 def main():
-    # Directly set the API key
-    groq_api_key = "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"
+    # Retrieve API key from environment variables
+    groq_api_key = os.getenv("GROQ_API_KEY")
     
+    # Verify API key is loaded
+    if not groq_api_key:
+        st.error("GROQ API Key not found. Please check your .env file.")
+        return
+
     st.title("PDF Chat with Groq LLM")
     
     # File uploader
@@ -44,7 +53,7 @@ def main():
         llm = ChatGroq(
             temperature=0.7, 
             model_name='llama3-70b-8192',
-            api_key=groq_api_key  # Use api_key instead of groq_api_key
+            api_key=groq_api_key
         )
         
         # Create QA chain