Update app.py

app.py

@@ -29,18 +29,26 @@ class ChatWrapper:
         api_key = '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'
         chain = self.set_openai_api_key(api_key)
         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 inp, history
-            # Set OpenAI key
-            #import openai
-            #openai.api_key = api_key
-            # Run chain and append input.
-            output = chain({"question": inp, "chat_history": history})["answer"]
-            history.append((inp, output))
-            chatResult = (output, history)
+          
+
+
+
+
+            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:")
+                question = input()
+                result = qa_chain({"question": question, "chat_history": chat_history})
+                chat_history.append((question, result["answer"]))
+                print("AI:")
+                print(result["answer"])
+                chatResult = (output, history)
+            
         except Exception as e:
             raise e
         finally: