Update rag.py

rag.py

@@ -4,14 +4,14 @@ from sentence_transformers import SentenceTransformer
 from pinecone import Pinecone, ServerlessSpec
 from groq import Groq
 from tqdm.auto import tqdm
-import streamlit as st
+import os
 
 # Constants (hardcoded)
 FILE_PATH = "anjibot_chunks.json"
 BATCH_SIZE = 384
 INDEX_NAME = "groq-llama-3-rag"
-PINECONE_API_KEY = st.secrets["PINECONE_API_KEY"]
-GROQ_API_KEY = st.secrets["GROQ_API_KEY"]
+PINECONE_API_KEY = os.getenv["PINECONE_API_KEY"]
+GROQ_API_KEY = os.getenv["GROQ_API_KEY"]
 DIMENSIONS = 768