Update app.py

app.py

@@ -53,6 +53,10 @@ def generate_answer_with_groq(question, context):
     """Generates an answer using the Groq API."""
     url = "https://api.groq.com/openai/v1/chat/completions"
     api_key = os.environ.get("GROQ_API_KEY")
+    if not api_key:
+        st.error("GROQ_API_KEY environment variable not found. Please set it.")
+        return None  # Indicate failure
+
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
@@ -92,7 +96,7 @@ def perform_rag_groq(vectorstore, query):
     relevant_docs = retriever.get_relevant_documents(query)
     context = "\n\n".join([doc.page_content for doc in relevant_docs])
     answer = generate_answer_with_groq(query, context)
-    return {"answer": answer, "sources": [doc.metadata['source'] for doc in relevant_docs]} # You might need to adjust how sources are stored
+    return {"answer": answer, "sources": [doc.metadata['source'] for doc in relevant_docs] if relevant_docs else []}
 
 # --- 3. Streamlit UI ---
 
@@ -100,12 +104,6 @@ def main():
     st.title("PDF Q&A with Local Docs (Powered by Groq)")
     st.info("Make sure you have a 'docs' folder in the same directory as this script containing your PDF files.")
 
-    groq_api_key = st.text_input("Enter your Groq API Key:", type="password")
-    if not groq_api_key:
-        st.warning("Please enter your Groq API key to ask questions.")
-        return
-    os.environ["GROQ_API_KEY"] = groq_api_key
-
     with st.spinner("Loading and processing PDF(s)..."):
         all_text, all_tables = load_and_process_pdfs_from_folder()
 
@@ -121,11 +119,14 @@ def main():
         if query:
             with st.spinner("Searching for answer..."):
                 result = perform_rag_groq(vectorstore, query)
-                st.subheader("Answer:")
-                st.write(result["answer"])
-                if "sources" in result:
-                    st.subheader("Source:")
-                    st.write(", ".join(result["sources"])) # Display sources
+                if result and result.get("answer"):
+                    st.subheader("Answer:")
+                    st.write(result["answer"])
+                    if "sources" in result and result["sources"]:
+                        st.subheader("Source:")
+                        st.write(", ".join(result["sources"]))
+                else:
+                    st.warning("Could not generate an answer.")
 
     if all_tables:
         st.subheader("Extracted Tables:")