Update QA_Bot.py

QA_Bot.py

@@ -3,7 +3,7 @@ from QnA import Q_A
 import re,time
 
 
-def QA_Bot(vectorstore):
+def QA_Bot(vectorstore,API_KEY=None):
     st.title("Q&A Bot")
     # Initialize chat history
     if "messages" not in st.session_state:
@@ -21,7 +21,7 @@ def QA_Bot(vectorstore):
         # Add user message to chat history
         st.session_state.messages.append({"role": "user", "content": prompt})
 
-        ai_response = Q_A(vectorstore,prompt)
+        ai_response = Q_A(vectorstore,prompt,API_KEY)
         response = f"Echo: {ai_response}"
         # Display assistant response in chat message container
         with st.chat_message("assistant"):