Update Classes/Helper_Class.py

Classes/Helper_Class.py

@@ -6,8 +6,7 @@ from langchain_community.vectorstores import FAISS
 import google.generativeai as genai
 import os
 
-os.environ["GOOGLE_API_KEY"] = "AIzaSyD5BAPI-PH4nbw4gY7BkcFFp1BTf2yR4bY"
-genai.configure(api_key="AIzaSyD5BAPI-PH4nbw4gY7BkcFFp1BTf2yR4bY")
+
 
 class GenerateFIASSDB:
     def __init__(self,pdf_docs : List[str], save_loc:str, model_embeddings: str = "models/embedding-001")-> None:
@@ -36,10 +35,13 @@ class GenerateFIASSDB:
         vector_store = FAISS.from_texts(text_chunks, embedding=embeddings)
         vector_store.save_local(self.save_loc)
 
-
 class DB_Retriever:
     def __init__(self, db_loc : str, model_embeddings : str = "models/embedding-001") -> None:
         self.db_loc = db_loc
+        try:
+            genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+        except Exception as e:
+            print(e)
         self.embeddings = GoogleGenerativeAIEmbeddings(model = model_embeddings)
         self.db = FAISS.load_local(self.db_loc, self.embeddings,allow_dangerous_deserialization  = True)