omdena-mental-health / cli_app.py
patti-j's picture
Create cli_app.py
7e168a8
raw
history blame contribute delete
509 Bytes
import pickle
from query_data import get_chain
if __name__ == "__main__":
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"])