Update app.py

app.py

@@ -10,36 +10,27 @@ with open("vectorstore.pkl", "rb") as f:
     vectorstore = pickle.load(f)
 
 
-def set_openai_api_key(api_key: str):
-    """Set the api key and return chain.
-    If no api_key, then None is returned.
-    """
-    if api_key:
-        os.environ["OPENAI_API_KEY"] = api_key
-        chain = get_chain(vectorstore)
-        os.environ["OPENAI_API_KEY"] = ""
-        return chain
-
 
 class ChatWrapper:
 
     def __init__(self):
         self.lock = Lock()
-
     def __call__(
-        self, api_key: str, inp: str, history: Optional[Tuple[str, str]], chain
+        self, inp: str, history: Optional[Tuple[str, str]],chain
     ):
         """Execute the chat functionality."""
         self.lock.acquire()
         try:
             history = history or []
+            chain = get_chain(vectorstore)
             # If chain is None, that is because no API key was provided.
             if chain is None:
                 history.append((inp, "Please paste your OpenAI key to use"))
                 return history, history
             # Set OpenAI key
             import openai
-            openai.api_key = api_key
+            
+            openai.api_key = os.environ["OPENAI_API_KEY"]
             # Run chain and append input.
             output = chain({"question": inp, "chat_history": history})["answer"]
             history.append((inp, output))
@@ -64,13 +55,6 @@ with block:
         gr.Markdown(
             "<h2><center>The Bot Forge</h2></center><h3> <center></center></h3>")
 
-        openai_api_key_textbox = gr.Textbox(
-            placeholder="Paste your OpenAI API key (sk-...)",
-            show_label=False,
-            lines=1,
-            type="password",
-        )
-
     chatbot = gr.Chatbot()
 
     with gr.Row():