Update app.py

app.py

@@ -67,6 +67,10 @@ st.caption('Never share your OpenAI API key to anyone. Note that usage of your O
 # Initialize messages object
 messages = []
 
+# Retrieve the OpenAI API Key from secrets
+# load_dotenv()
+# api_key = os.getenv("OPENAI_API_KEY")
+
 
 # new line space
 st.write("")
@@ -78,9 +82,10 @@ uploaded_files = st.file_uploader("", help='Up to five images only.', type=["jpg
 
 if api_key:
     # Initialize the OpenAI client with the API key
-    client = OpenAI(api_key=api_key)    
+    client = OpenAI(api_key=api_key)   
+
     # load from disk
-    docsearch = Chroma(persist_directory="./knowledge_base", embedding_function=OpenAIEmbeddings())
+    docsearch = Chroma(persist_directory="./knowledge_base", embedding_function=OpenAIEmbeddings(api_key=api_key))
 
     metadata_field_info = [
         AttributeInfo(