Update app.py

app.py

@@ -10,17 +10,6 @@ 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):
@@ -28,7 +17,11 @@ class ChatWrapper:
     def __call__(
         self, inp: str, history: Optional[Tuple[str, str]]
     ):
+
+
+        chain = get_chain(vectorstore)
         """Execute the chat functionality."""
+        os.environ["OPENAI_API_KEY"] = api_key
         self.lock.acquire()
         api_key = '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'
         chain = set_openai_api_key(api_key)