Update app.py

app.py

@@ -17,11 +17,17 @@ class ChatWrapper:
     def __call__(
         self, inp: str, history: Optional[Tuple[str, str]]
     ):
+    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
 
-
-        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)