Update app.py

app.py

@@ -16,10 +16,9 @@ def set_openai_api_key(api_key: str):
     """
     print(os.environ["TEST"])
     
-    if api_key:
-        os.environ["OPENAI_API_KEY"] = api_key
-        chain = get_chain(vectorstore)
-        os.environ["OPENAI_API_KEY"] = ""
+          
+        
+        
         return chain
 
 class ChatWrapper:
@@ -27,19 +26,21 @@ class ChatWrapper:
     def __init__(self):
         self.lock = Lock()
     def __call__(
-        self, api_key: str, inp: str, history: Optional[Tuple[str, str]], chain
+        self, api_key: str, inp: str, history: Optional[Tuple[str, str]]
     ):
         """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))
@@ -92,14 +93,10 @@ with block:
     state = gr.State()
     agent_state = gr.State()
 
-    submit.click(chat, inputs=[openai_api_key_textbox, message, state, agent_state], outputs=[chatbot, state],api_name="chat")
+    
     message.submit(chat, inputs=[openai_api_key_textbox, message, state, agent_state], outputs=[chatbot, state])
 
-    openai_api_key_textbox.change(
-        set_openai_api_key,
-        inputs=[openai_api_key_textbox],
-        outputs=[agent_state],
-    )
+    
 def echo(name, request: gr.Request):
     if request:
         print("Request headers dictionary:", request.headers)