Added functionality to detect missing keys

helpers.py

@@ -20,11 +20,31 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.vectorstores import Pinecone
 from langchain_openai import OpenAIEmbeddings
 from langchain_openai import OpenAI
-from langchain import hub
 
-os.environ['LANGCHAIN_TRACING_V2'] = 'true'
-os.environ['LANGCHAIN_ENDPOINT'] = 'https://api.smith.langchain.com'
-os.environ['USER_AGENT'] = 'myagent'
+def check_missing_keys():
+    try:
+
+        st.session_state.KEYS_ARE_MISSING = False
+
+        keys = {
+            "OpenAI": os.getenv("OPENAI_API_KEY"),
+            "Pinecone": os.getenv("PINECONE_API_KEY")
+        }
+        
+        missing_keys = []
+        for key, value in keys.items():
+            if not value:
+                missing_keys.append(key)
+
+        if len(missing_keys) > 0:
+            st.session_state.KEYS_ARE_MISSING = True
+            raise KeyError(f"Missing required API keys: {', '.join(missing_keys)}")
+
+    except KeyError as e:
+        st.error(e)
+
+
+
 
 
 def file_uploader() -> List[Document]:
@@ -107,7 +127,11 @@ def upload_button(documents: List[Document], vector_store: Pinecone):
             self.message = message
             super().__init__(self.message)
     
-    if st.sidebar.button('Store to the Database', key='vector-store'):
+    if st.sidebar.button(
+        'Store to the Database', 
+        key='vector-store', 
+        disabled=st.session_state.KEYS_ARE_MISSING
+        ):
 
         try:
             if documents is None:
@@ -155,14 +179,14 @@ def match_resumes(job_description: str, vector_store: Pinecone):
 
     class NoResults(Exception):
         """
-        A custom exception for when an no API keys are available.
+        A custom exception for when there no results retrieved.
         """
         def __init__(self, message = 'No matched results found'):
             self.message = message
             super().__init__(self.message)
 
     try:
-        if st.button('Match resumes'):
+        if st.button('Match resumes', disabled=st.session_state.KEYS_ARE_MISSING):
             with st.status("Fetching matching resumes...", expanded=True):
 
                 if job_description == '':