update: add env variables support and secrets management

pages/01_Resume Screen.py

@@ -31,7 +31,7 @@ def load_lottiefile(filepath: str):
 st_lottie(load_lottiefile("images/welcome.json"), speed=1, reverse=False, loop=True, quality="high", height=300)
 
 if "VOYAGE_API_KEY" not in st.session_state:
-    st.session_state = st.text_input("PLEASE ENTER YOUR VOYAGE API KEY. Head over to this [link](https://dash.voyageai.com/)",
+    st.session_state["VOYAGE_API_KEY"] = st.text_input("PLEASE ENTER YOUR VOYAGE API KEY. Head over to this [link](https://dash.voyageai.com/)",
                                                  placeholder="Enter your Key here...",
                                                  type="password")
     st.stop()

pages/02_Professional Screen.py

@@ -30,7 +30,7 @@ def load_lottiefile(filepath: str):
 st_lottie(load_lottiefile("images/welcome.json"), speed=1, reverse=False, loop=True, quality="high", height=300)
 
 if "VOYAGE_API_KEY" not in st.session_state:
-    st.session_state = st.text_input("PLEASE ENTER YOUR VOYAGE API KEY. Head over to this [link](https://dash.voyageai.com/)",
+    st.session_state["VOYAGE_API_KEY"] = st.text_input("PLEASE ENTER YOUR VOYAGE API KEY. Head over to this [link](https://dash.voyageai.com/)",
                                                  placeholder="Enter your Key here...",
                                                  type="password")
     st.stop()

pages/03_Behavioral Screen.py

@@ -31,7 +31,7 @@ def load_lottiefile(filepath: str):
 st_lottie(load_lottiefile("images/welcome.json"), speed=1, reverse=False, loop=True, quality="high", height=300)
 
 if "VOYAGE_API_KEY" not in st.session_state:
-    st.session_state = st.text_input("PLEASE ENTER YOUR VOYAGE API KEY. Head over to this [link](https://dash.voyageai.com/)",
+    st.session_state["VOYAGE_API_KEY"] = st.text_input("PLEASE ENTER YOUR VOYAGE API KEY. Head over to this [link](https://dash.voyageai.com/)",
                                                  placeholder="Enter your Key here...",
                                                  type="password")
     st.stop()

pages/05_Search_Candidates.py

@@ -12,19 +12,28 @@ from dotenv import load_dotenv
 load_dotenv()
 
 st.title("Search the right candidates!")
-
-if "OPENAI_API_KEY" not in st.session_state:
-    st.session_state["OPENAI_API_KEY"] = st.text_input("PLEASE ENTER YOUR OPEN API KEY. Head over to this [link](https://dash.voyageai.com/)",
-                                                 placeholder="Enter your Key here...",
-                                                 type="password")
-    st.stop()
-
-if "COHERE_API_KEY" not in st.session_state:
-    st.session_state["COHERE_API_KEY"] = st.text_input("PLEASE ENTER YOUR COHERE API KEY. Head over to this [link](https://dash.voyageai.com/)",
-                                                 placeholder="Enter your Key here...",
-                                                 type="password")
-    st.stop()
-
+# st.write( "OPENAI_API_KEY" not in os.environ.keys())
+# if "OPENAI_API_KEY" not in os.environ.keys():
+#     os.environ["OPENAI_API_KEY"] = st.text_input(
+#         "PLEASE ENTER YOUR OPEN API KEY. Head over to this [link](https://platform.openai.com/api-keys)",
+#         placeholder="Enter your Key here...",
+#         type="password")
+#     st.write("You can also set the OPENAI_API_KEY environment variable.")
+#     st.rerun()
+#
+# if "COHERE_API_KEY" not in os.environ.keys():
+#     os.environ["COHERE_API_KEY"] = st.text_input(
+#         "PLEASE ENTER YOUR COHERE API KEY. Head over to this [link](https://dashboard.cohere.com/welcome/login?redirect_uri=%2F)",
+#         placeholder="Enter your Key here...",
+#         type="password")
+#     st.rerun()
+#
+# if "VOYAGE_API_KEY" not in os.environ.keys():
+#     os.environ["VOYAGE_API_KEY"] = st.text_input(
+#         "PLEASE ENTER YOUR VOYAGE API KEY. Head over to this [link](https://dash.voyageai.com/)",
+#         placeholder="Enter your Key here...",
+#         type="password")
+#     st.rerun()
 
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -40,7 +49,6 @@ if query := st.chat_input("Whom are you looking for today?"):
     st.session_state.messages.append({"role": "user", "content": query})
     with st.chat_message("user"):
         st.markdown(query)
-
     with st.chat_message("assistant"):
         message_placeholder = st.empty()
         with st.spinner('Pulling out amazing candidates behind the bushes...'):

utils/llm.py

@@ -12,7 +12,6 @@ from dotenv import load_dotenv
 from operator import itemgetter
 from langchain_core.runnables import RunnableParallel, RunnablePassthrough, RunnableLambda
 from langchain.output_parsers import StructuredOutputParser, ResponseSchema
-
 load_dotenv()
 
 response_schemas = [
@@ -43,8 +42,10 @@ human_message = ("You are provided with a list of resumes. "
 
 
 @st.cache_resource
-def init_retriever(dataset_path="hub://p1utoze/default", embeddings_model="voyage-lite-01"):
-    embeddings = VoyageEmbeddings(voyage_api_key=st.session_state["VOYAGE_API_KEY"], model=embeddings_model, show_progress_bar=True)
+def init_retriever(dataset_path="hub://p1utoze/default",
+                   embeddings_model="voyage-lite-01",
+):
+    embeddings = VoyageEmbeddings(model=embeddings_model, show_progress_bar=True)
     db = DeepLake(token=st.secrets["deep_lake_token"], dataset_path=dataset_path, read_only=True, embedding=embeddings)
     retriever = db.as_retriever()
     retriever.search_kwargs['top_k'] = "5"
@@ -58,7 +59,7 @@ def init_retriever(dataset_path="hub://p1utoze/default", embeddings_model="voyag
 
 @st.cache_resource
 def load_model():
-    chat = ChatOpenAI(openai_api_key=st.session_state["OPENAI_API_KEY"], model_name="gpt-3.5-turbo", temperature=0.4)
+    chat = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.4)
     return chat
 
 
@@ -81,7 +82,10 @@ def typewriter(text: str, title: str, speed: int):
         time.sleep(1 / speed)
 
 def model_pipeline(memory):
-    compression_retriever = init_retriever("hub://p1utoze/resumes", embeddings_model="voyage-lite-01")
+    compression_retriever = init_retriever(
+        "hub://p1utoze/resumes",
+        embeddings_model="voyage-lite-01",
+    )
     chat = load_model()
     system_message_prompt = SystemMessagePromptTemplate.from_template(system_message)
     human_message_prompt = HumanMessagePromptTemplate.from_template(