Update app.py

app.py

@@ -3,14 +3,15 @@ import PyPDF2
 import tiktoken
 import faiss
 import numpy as np
+import os
 from sentence_transformers import SentenceTransformer
 import requests
 
 # Load embedding model
 embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
 
-# GROQ API configuration
-GROQ_API_KEY = "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"  # ⛔ Replace this with your actual GROQ API key
+# GROQ API configuration (🔐 loaded securely from environment variable)
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 GROQ_URL = "https://api.groq.com/openai/v1/chat/completions"
 LLAMA3_MODEL = "llama3-8b-8192"
 
@@ -42,6 +43,9 @@ def search_index(index, query, chunks, top_k=3):
 
 # Generate answer using GROQ
 def generate_answer(prompt):
+    if not GROQ_API_KEY:
+        return "🚫 GROQ API key not found. Please set it in environment variables."
+
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
         "Content-Type": "application/json"
@@ -81,9 +85,12 @@ def main():
                 top_chunks = search_index(st.session_state.index, query, st.session_state.chunks)
                 context = "\n\n".join(top_chunks)
                 prompt = f"Use the following context to answer the question:\n\n{context}\n\nQuestion: {query}"
-                answer = generate_answer(prompt)
-                st.markdown("### 🧠 Answer:")
-                st.write(answer)
+                try:
+                    answer = generate_answer(prompt)
+                    st.markdown("### 🧠 Answer:")
+                    st.write(answer)
+                except requests.exceptions.HTTPError as e:
+                    st.error(f"❌ API Error: {e}")
 
 if __name__ == "__main__":
     main()