Update app.py

app.py

@@ -12,17 +12,6 @@ class ChatWrapper:
     def __init__(self):
         self.lock = Lock()
 
-    def set_openai_api_key(self, api_key: str):
-        with open("vectorstore.pkl", "rb") as f:
-            vectorstore = pickle.load(f)
-        """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
 
     def __call__(self, inp: str, history: Optional[Tuple[str, str]]):
         self.lock.acquire()
@@ -41,14 +30,30 @@ class ChatWrapper:
             qa_chain = get_chain(vectorstore)
             
             print("Chat with your docs!")
+
+    with open("vectorstore.pkl", "rb") as f:
+        vectorstore = pickle.load(f)
+            qa_chain = get_chain(vectorstore)
+            chat_history = []
+            print("Chat with your docs!")
             while True:
                 print("Human:")
-                history = history or []
-                output = qa_chain({"question": inp, "chat_history": history})["answer"]
-                history.append((inp, output))
+                question = inp
+                output = qa_chain({"question": question, "chat_history": chat_history})
+                chat_history.append((question, output["answer"]))
                 print("AI:")
-                print(output["answer"])
-                chatResult = (output, history)
+                print(result["answer"])
+
+
+            
+           # while True:
+            #    print("Human:")
+             #   history = history or []
+              #  output = qa_chain({"question": inp, "chat_history": history})["answer"]
+                #history.append((inp, output))
+               # print("AI:")
+                #print(output["answer"])
+                chatResult = (output, chat_history)
             
         except Exception as e:
             raise e