Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,6 +4,12 @@ from pinecone import Pinecone
 from groq import Groq
 import uuid
 import time
+import os
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+PINECONE_API_KEY = os.environ.get("PINECONE_API_KEY")
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+
 
 # ------------------------------- 
 # Page Configuration
@@ -185,20 +191,21 @@ def load_models():
     with st.spinner("🤖 Loading AI models..."):
         embed_model = SentenceTransformer(
             "google/embeddinggemma-300m",
-            token=st.secrets["HF_TOKEN"]
+            token=HF_TOKEN
         )
+
         reranker = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
     return embed_model, reranker
 
 @st.cache_resource(show_spinner=False)
 def initialize_pinecone():
-    pc = Pinecone(api_key=st.secrets["PINECONE_API_KEY"])
+    pc = Pinecone(api_key=PINECONE_API_KEY)
     index = pc.Index("rag")
     return index
 
 @st.cache_resource(show_spinner=False)
 def initialize_groq():
-    client = Groq(api_key=st.secrets["GROQ_API_KEY"])
+    client = Groq(api_key=GROQ_API_KEY)
     return client
 
 # Initialize models and services