Update app.py

app.py

@@ -2,8 +2,14 @@
 import streamlit as st
 import numpy as np
 import faiss
+import os
 from sentence_transformers import SentenceTransformer
 import requests
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
 st.set_page_config(page_title="Meraj Graphics Assistant")
 
@@ -25,10 +31,10 @@ def search(query, top_k=3):
     return "\n".join(results)
 
 # Call Groq API
-def query_groq(context, question, api_key):
+def query_groq(context, question):
     url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {
-        "Authorization": f"Bearer {api_key}",
+        "Authorization": f"Bearer {GROQ_API_KEY}",
         "Content-Type": "application/json"
     }
     data = {
@@ -45,14 +51,15 @@ def query_groq(context, question, api_key):
 st.title("📋 Meraj Graphics Chat Assistant")
 
 question = st.text_input("Ask something about our services:")
-groq_key = st.text_input("Groq API Key", type="password")
 
 if st.button("Get Answer"):
-    if not question or not groq_key:
-        st.warning("Please provide both question and API key.")
+    if not question:
+        st.warning("Please enter a question.")
+    elif not GROQ_API_KEY:
+        st.error("API key not found. Please set GROQ_API_KEY in your environment.")
     else:
         with st.spinner("Searching..."):
             context = search(question)
-            answer = query_groq(context, question, groq_key)
+            answer = query_groq(context, question)
             st.success("Answer:")
             st.write(answer)