Update app.py

app.py

@@ -92,20 +92,21 @@ 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 []
             # 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
+            # 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.
+            chain = get_chain(vectorstore)
             output = chain({"question": inp, "chat_history": history})["answer"]
             history.append((inp, output))
         except Exception as e:
@@ -136,8 +137,6 @@ with block:
     
     vectorstore = load_vectorstore(embeddings.value)
 
-    print(vectorstore)
-
     chatbot = gr.Chatbot()
 
     with gr.Row():