Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import streamlit as st
 from langchain_groq import ChatGroq
 from langchain.document_loaders import PyPDFLoader
@@ -7,6 +8,14 @@ from langchain.vectorstores import FAISS
 from langchain.chains import RetrievalQA
 
 def main():
+    # Retrieve Groq API key from environment variable
+    groq_api_key = os.getenv("GROQ_API_KEY")
+    
+    # Check if API key is available
+    if not groq_api_key:
+        st.error("GROQ_API_KEY not found. Please set the environment variable.")
+        return
+
     st.title("PDF Chat with Groq LLM")
     
     # File uploader
@@ -36,10 +45,11 @@ def main():
         # Create vector store
         vectorstore = FAISS.from_documents(texts, embeddings)
         
-        # Initialize Groq LLM
+        # Initialize Groq LLM with API key
         llm = ChatGroq(
             temperature=0.7, 
-            model_name='llama3-70b-8192'
+            model_name='llama3-70b-8192',
+            groq_api_key=groq_api_key  # Pass the API key explicitly
         )
         
         # Create QA chain