Update app.py

app.py

@@ -7,10 +7,12 @@ from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import DeepLake
 from langchain.chat_models import ChatOpenAI
 from langchain.chains import ConversationalRetrievalChain
+from langchain.llms import OpenAI
 
 def set_api_key(key):
     os.environ["OPENAI_API_KEY"] = key
-    return f"Key Successfully Set to: {key}"
+    time.sleep(600)
+    os.environ["OPENAI_API_KEY"] = ""
     
 def get_api_key():
     api_key = os.getenv("OPENAI_API_KEY")
@@ -25,13 +27,13 @@ def respond(message, chat_history):
     db = DeepLake(dataset_path="./documentation_db", embedding_function=embeddings, read_only=True)
     #Set retriever settings
     retriever = db.as_retriever(search_kwargs={"distance_metric":'cos',
-                                               "fetch_k":20,
+                                               "fetch_k":10,
                                                "maximal_marginal_relevance":True,
-                                               "k":20})
+                                               "k":10})
 
     # Create ChatOpenAI and ConversationalRetrievalChain
     model = ChatOpenAI(model='gpt-3.5-turbo')
-    qa = ConversationalRetrievalChain.from_llm(model, retriever)
+    qa = ConversationalRetrievalChain.from_llm(model, retriever, OpenAI(temperature=0))
 
     chat_history=[]
     bot_message = qa({"question": message, "chat_history": chat_history})
@@ -41,8 +43,11 @@ def respond(message, chat_history):
 
 with gr.Blocks() as demo:
     with gr.Tab("OpenAI API Key Submission"):
-        api_input = gr.Textbox(label = "API Key", placeholder = "Please provide your OpenAI API key here")
-        api_submission_conf = gr.Textbox(label = "Submission Confirmation")
+        api_input = gr.Textbox(label = "API Key - Your key will be active for 10 minutes. The bar below shows the time left.",
+                               placeholder = "Please provide your OpenAI API key here.")
+        api_submission_conf = gr.Textbox(label = "API Key Status",
+                                         placeholder = "Your API Key has been reset or has not be set yet. Please re-enter your key.",
+                                         interactive = False)
         api_submit_button = gr.Button("Submit")
     
     with gr.Tab("Coding Assistant"):