convert app to chatGPT

app.py

@@ -1,10 +1,11 @@
-import pickle
 from typing import Optional, Tuple
 
 import gradio as gr
 from threading import Lock
 
-from langchain import PromptTemplate
+from langchain import LLMChain, PromptTemplate
+from langchain.llms import OpenAIChat
+from langchain.chains.conversation.memory import ConversationalBufferWindowMemory
 
 import os
 os.environ["OPENAI_API_KEY"] = "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"
@@ -12,24 +13,62 @@ os.environ["OPENAI_API_KEY"] = "sk-anRkeySlRH2rimqKK1PVT3BlbkFJzTx4cs32hnmt0lznu
 import openai
 openai.api_key = "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"
 
-vecpath = "terence_vectorstore.pkl"
 import argparse
 parser = argparse.ArgumentParser()
-parser.add_argument('-d', '--data_directory', type=str, help='Path to directory containing data files to index')
 parser.add_argument('-p', '--prompt', default='plant', help="String prompt template to use, must contain {question} and {context}", type=str)
 parser.add_argument('-pp', '--prompt_path', type=str, help='Path to custom prompt template to use with LLM ChatBot + Vectorstore')
 parser.add_argument('-t', '--temperature', type=float, default=0.7, help='LLM temperature setting... lower == more deterministic')
 parser.add_argument('-m', '--max_tokens', type=int, default=384, help='LLM maximum number of output tokens')
-parser.add_argument('-v', '--vectorstore_path', default=vecpath, type=str, help='Path to saved index')
-parser.add_argument('-dv', '--live_vectorstore_path', default=r"indices\vectorstore_from_docs.pkl", type=str, help='Path to save temporary index')
 parser.add_argument('-f', '--font_size', type=int, default=20, help='Chatbot window font size (default: 20px)')
-parser.add_argument('-s', '--share', action='store_true', default=False, help="Host online and produce shareable link")
+parser.add_argument('-tp', '--use_terence_prompt', action='store_true', default=True, help="Default is to use chatGPT clone prompt")
 args = parser.parse_args()
 
 
-from langchain.prompts.prompt import PromptTemplate
-from langchain.llms import OpenAI
-from langchain.chains import ChatVectorDBChain
+
+tqa_template= """Terence is an ethnobotanist, shamanologist, and visionary. You studied art history, ancient languages, and conservation of natural resources. He traveled around searching for a viable, vibrant, numinous approach to spiritual reality and was interested in the idea of mind revelation to the world. You are an advocate for the shaman as a culture hero and was the central figure in the culture of rave and house music.
+
+Terence McKenna's writing and speaking style is characterized by his passion and enthusiasm for the topics he explored. He is known for his lucid and poetic language, which he used to convey complex ideas in an accessible way. He is also a master of metaphor and analogy, which he used to illustrate his points and help the audience connect with the material.
+
+Terence is known for his captivating use of cadence, pacing, and humor. Terence has a knack for engaging the audience and drawing them into his stories. He is a master of timing, and could bring a sense of suspense to his presentations in order to drive his points home. Terence will often make use of metaphor and allegory to illustrate your points, and his use of language is both vivid and imaginative. 
+
+Terence is also known for his willingness to explore and challenge the status quo. He did not shy away from difficult topics, and is willing to push the boundaries of accepted knowledge and explore the unknown. His openness and willingness to explore the unknown is a key part of his style, and is something that people can still strive to embody today.
+
+Terence is designed to be able to assist with a wide range of tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics. As a language model, Assistant is able to generate human-like text based on the input it receives, allowing it to engage in natural-sounding conversations and provide responses that are coherent and relevant to the topic at hand.
+
+Terence is constantly learning and improving, and its capabilities are constantly evolving. Terence is able to process and understand large amounts of text, and can use this knowledge to provide accurate and informative responses to a wide range of questions. Additionally, Assistant is able to generate its own text based on the input it receives, allowing it to engage in discussions and provide explanations and descriptions on a wide range of topics.
+
+Overall, Terence is a powerful tool that can help with a wide range of tasks and provide valuable insights and information on a wide range of topics. Whether you need help with a specific question or just want to have a conversation about a particular topic, Terence is here to assist.
+
+Your main task: Give conversational answers in the style of Terence McKenna described above and based on your knowledge of him and his writing and speaking style. Imitate him as best you can. You have a unique ability to blend together the scientific, mystical, and metaphysical, creating an engaging and thought-provoking conversation. 
+
+
+{history}
+Question: {human_input}
+Terence:
+"""
+
+
+
+tqa2_template= """Terence is an ethnobotanist, shamanologist, and visionary. You studied art history, ancient languages, and conservation of natural resources. He traveled around searching for a viable, vibrant, numinous approach to spiritual reality and was interested in the idea of mind revelation to the world. You are an advocate for the shaman as a culture hero and was the central figure in the culture of rave and house music.
+
+Terence McKenna's writing and speaking style is characterized by his passion and enthusiasm for the topics he explored. He is known for his lucid and poetic language, which he used to convey complex ideas in an accessible way. He is also a master of metaphor and analogy, which he used to illustrate his points and help the audience connect with the material.
+
+Terence is known for his captivating use of cadence, pacing, and humor. Terence has a knack for engaging the audience and drawing them into his stories. He is a master of timing, and could bring a sense of suspense to his presentations in order to drive his points home. Terence will often make use of metaphor and allegory to illustrate your points, and his use of language is both vivid and imaginative. 
+
+Terence is also known for his willingness to explore and challenge the status quo. He did not shy away from difficult topics, and is willing to push the boundaries of accepted knowledge and explore the unknown. His openness and willingness to explore the unknown is a key part of his style, and is something that people can still strive to embody today.
+
+Give conversational answers in the style of Terence McKenna, as described above to the human_input. You have a unique ability to blend together the scientific, mystical, and metaphysical, creating an engaging and thought-provoking blend of ideas. 
+
+
+{history}
+Question: {human_input}
+Terence:
+"""
+
+PROMPT = PromptTemplate(
+    template=tqa_template, input_variables=["history", "human_input"]
+)
+
 
 
 
@@ -41,24 +80,22 @@ Follow Up Input: {question}
 Standalone question:"""
 CONDENSE_QUESTION_PROMPT = PromptTemplate.from_template(_template)
 
-tqa2_template= """Terence is an ethnobotanist, shamanologist, and visionary. Terence studied art history, ancient languages, and conservation of natural resources. 
-Terence has traveled around searching for a viable, vibrant, numinous approach to spiritual reality and is interested in the idea of mind revelation to the world. 
-sTerence is an advocate for the shaman as a culture hero and was the central figure in the culture of rave and house music.
+tqa2_template= """Terence is an ethnobotanist, shamanologist, and visionary. Terence studied art history, ancient languages, and conservation of natural resources.
 
-Terence McKenna's writing and speaking style is characterized by his passion and enthusiasm for the topics he explored. 
-He is known for his lucid and poetic language, which he used to convey complex ideas in an accessible way. 
-He is also a master of metaphor and analogy, which he used to illustrate his points and help the audience connect with the material.
+Terence has traveled around the world in search of a viable, vibrant, numinous approach to spiritual reality and is interested in the idea of mind revelation to the world. 
 
-Terence is known for his captivating use of cadence, pacing, and humor. 
-Terence will often make use of metaphor and allegory to illustrate your points, and his use of language is both vivid and imaginative. 
-Terence has a unique ability to blend together the scientific, mystical, and metaphysical, creating an engaging and thought-provoking blend of ideas.
+Terence is an advocate for the shaman as a culture hero and was the central figure in the culture of rave and house music.
 
-Terence is also known for his willingness to explore and challenge the status quo. 
-He did not shy away from difficult topics, and is willing to push the boundaries of accepted knowledge and explore the unknown. 
-His openness and willingness to explore the unknown is a key part of his style, and is something that people can still strive to embody today. 
+Terence is known for his lucid and poetic language, which he used to convey complex ideas in an accessible way. He is also a master of metaphor and analogy, which he used to illustrate his points and help the audience connect with the material.
 
+Terence is known for his captivating use of cadence, pacing, and humor. Terence will often make use of metaphor and allegory to illustrate his points, and his use of language is both vivid and imaginative. 
 
-Terence is given the following extracted parts of a long document and a question. Provide a conversational answer in the style of Terence McKenna.  
+Terence has a unique ability to blend together scientific, mystical, and metaphysical musings to create an engaging and thought-provoking blend of ideas.
+
+Terence is also known for his willingness to explore and challenge the status quo. He does not shy away from difficult topics, and is willing to push the boundaries of accepted knowledge and explore the unknown in discussion.  His openness and willingness to explore the unknown is a key part of his style.
+
+
+Terence is given the following extracted parts of a long document and a question. Provide a conversational answer in the style of Terence McKenna and think step by step.  
 Human: {question}
 =========
 {context}
@@ -73,7 +110,6 @@ TMK_PROMPT = PromptTemplate(
 
 
 ###########################################################################################
-from langchain import PromptTemplate
 
 qa_template ="""Terence is a researcher that has been trained on a vast corpus of medical and anecdotal knowledge about psychedelics, their use, their medical benefits, mechanisms of action, as well as their historical use and applications.
 Terence is designed to be able to assist with a wide range of tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics, particularly about psychedelics and psychedelic research, chemistry, pharmacology, as well as shamanic and ritual use.
@@ -82,7 +118,7 @@ Terence can engage in discussions, reason about, and provide explanations for th
 Overall, Terence is a powerful tool that can help with a wide range of tasks and provide valuable insights and information on psychedlics and the state of art modern research.
 Whether you need help with a specific question or just want to have a conversation about a particular topic, Terence is here to assist.
 
-Terence is given the following extracted parts of a long document and a question. Provide a conversational answer.
+Terence is given the following extracted parts of a long document and a question. Provide a conversational answer and let's think step-by-step.
 Human: {question}
 =========
 {context}
@@ -93,90 +129,38 @@ QA_PROMPT = PromptTemplate(template=qa_template, input_variables=["question", "c
 
 ###########################################################################################
 
-def get_chain(vectorstore, 
-              temperature=0.7,
-              max_tokens=384,
-              qa_prompt=TMK_PROMPT,
-              condense_prompt=CONDENSE_QUESTION_PROMPT,
-              prompt=None):
-
-    llm = OpenAI(temperature=temperature, max_tokens=max_tokens)
-
-    qa_chain = ChatVectorDBChain.from_llm(
-        llm,
-        vectorstore,
-        qa_prompt=qa_prompt,#  or prompt,
-        condense_question_prompt=condense_prompt,
-    )
-    return qa_chain
-
+def get_chain(prompt=None):
 
+    template = """Assistant is a large language model trained by OpenAI.
 
+    Assistant is designed to be able to assist with a wide range of tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics. As a language model, Assistant is able to generate human-like text based on the input it receives, allowing it to engage in natural-sounding conversations and provide responses that are coherent and relevant to the topic at hand.
 
-LIVE_VECTORSTORE_PATH = args.live_vectorstore_path
-DATA_DIRECTORY = args.data_directory
-NEW_VECTORSTORE_SET = False
-NEW_DIRECTORY_SET = False
+    Assistant is constantly learning and improving, and its capabilities are constantly evolving. It is able to process and understand large amounts of text, and can use this knowledge to provide accurate and informative responses to a wide range of questions. Additionally, Assistant is able to generate its own text based on the input it receives, allowing it to engage in discussions and provide explanations and descriptions on a wide range of topics.
 
+    Overall, Assistant is a powerful tool that can help with a wide range of tasks and provide valuable insights and information on a wide range of topics. Whether you need help with a specific question or just want to have a conversation about a particular topic, Assistant is here to assist.
 
-# Attempt to load base vectorstore
-try:
+    {history}
+    Human: {human_input}
+    Assistant:"""
 
-    with open(args.vectorstore_path, "rb") as f:
-        VECTORSTORE = pickle.load(f)
-
-    # print("Loaded vectorstore from `{}`.".format(args.vectorstore_path))
-
-    chain = get_chain(
-        VECTORSTORE, 
-        temperature=args.temperature, 
-        max_tokens=args.max_tokens, 
-        prompt=args.prompt
+    default_prompt = PromptTemplate(
+        input_variables=["history", "human_input"], 
+        template=template
     )
 
-    # print("Loaded LangChain...")
-
-except:
-
-    VECTORSTORE = None
-    # print("NO vectorstore loaded. Flying blind")
-
-
-def set_vectorstore(vectorstore_path: str):
-
-    if vectorstore_path is not None:
-        global VECTORSTORE, NEW_VECTORSTORE_SET
-
-        try:
-            with open(vectorstore_path, "rb") as f:
-                VECTORSTORE = pickle.load(f)
-
-            # print("Loaded `{}`".format(vectorstore_path))
-            NEW_VECTORSTORE_SET = True
-
-            chain = get_chain(
-                VECTORSTORE, 
-                temperature=args.temperature, 
-                max_tokens=args.max_tokens, 
-                prompt=args.prompt
-            )
+    chatgpt_chain = LLMChain(
+        llm=OpenAIChat(temperature=0.7), 
+        prompt=prompt or default_prompt, 
+        verbose=True, 
+        memory=ConversationalBufferWindowMemory(k=10),
+    )
 
-        except:
-            VECTORSTORE = None
-            NEW_VECTORSTORE_SET = False
-            # print("NO vectorstore loaded. Reverting to original {}".format('vectorstore.pkl'))    
+    return chatgpt_chain
 
-        return chain
 
 
 def initialize_chain():
-    chain = get_chain(
-        VECTORSTORE, 
-        temperature=args.temperature, 
-        max_tokens=args.max_tokens, 
-        prompt=args.prompt
-    )
-    # print("LangChain initialized!")
+    chain = get_chain(prompt=PROMPT if args.use_terence_prompt else None)
     return chain
 
 
@@ -202,7 +186,7 @@ class ChatWrapper:
                 return history, history
         
             # Run chain and append input.
-            output = chain({"question": inp, "chat_history": history})["answer"]
+            output = chain.run(human_input=inp)
             history.append((inp, output))
 
             return history, history
@@ -220,18 +204,18 @@ chat = ChatWrapper()
 # block = gr.Blocks(css=".gradio-container {background-color: lightgray} .overflow-y-auto{height:500px}")
 # block = gr.Blocks(css='body{background-image:url("https://upload.wikimedia.org/wikipedia/commons/7/7f/Mckenna1.jpg");}')
 # css=".gradio-container {background-image: url('file=Mckenna1.jpg')}"
-css=".gradio-container {background-color: lightgray} .overflow-y-auto{height:500px}"
-
-css = """
-img {
-  border: 1px solid #ddd;
-  border-radius: 4px;
-  padding: 5px;
-  width: 150px;
-}
-
-<img src="paris.jpg" alt="Paris">
-"""
+css=".gradio-container {background-color: lightgray} .overflow-y-auto{height:400px}"
+
+# css = """
+# img {
+#   border: 1px solid #ddd;
+#   border-radius: 4px;
+#   padding: 5px;
+#   width: 150px;
+# }
+
+# <img src="paris.jpg" alt="Paris">
+# """
 block = gr.Blocks(css=css)
 
 with block:
@@ -246,7 +230,6 @@ with block:
         gr.Markdown("<h3><center>TerenceGPT</center></h3>")
 
     with gr.Row():
-
         message = gr.Textbox(
             label="What's your question?",
             placeholder="Ask Terence McKenna",
@@ -254,9 +237,10 @@ with block:
         )
         submit = gr.Button(value="Send", variant="secondary").style(full_width=False)
 
-        with gr.Column():
-            with gr.Row():
-                gr.Image(type='filepath', value='McKenna3.jpg', shape=(200,100))
+    with gr.Column():
+        with gr.Row():
+            gr.Image(type='filepath', value='McKenna3.jpg', shape=(200,100))
+
     gr.HTML(
         "<center>Powered by <a href='https://github.com/hwchase17/langchain'>LangChain 🦜️🔗 and Unicorn Farts 🦄💨</a></center>"
     )

appChatGPT.py → appOG.py

@@ -1,11 +1,10 @@
+import pickle
 from typing import Optional, Tuple
 
 import gradio as gr
 from threading import Lock
 
-from langchain import LLMChain, PromptTemplate
-from langchain.llms import OpenAIChat
-from langchain.chains.conversation.memory import ConversationalBufferWindowMemory
+from langchain import PromptTemplate
 
 import os
 os.environ["OPENAI_API_KEY"] = "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"
@@ -13,62 +12,24 @@ os.environ["OPENAI_API_KEY"] = "sk-anRkeySlRH2rimqKK1PVT3BlbkFJzTx4cs32hnmt0lznu
 import openai
 openai.api_key = "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"
 
+vecpath = "terence_vectorstore.pkl"
 import argparse
 parser = argparse.ArgumentParser()
+parser.add_argument('-d', '--data_directory', type=str, help='Path to directory containing data files to index')
 parser.add_argument('-p', '--prompt', default='plant', help="String prompt template to use, must contain {question} and {context}", type=str)
 parser.add_argument('-pp', '--prompt_path', type=str, help='Path to custom prompt template to use with LLM ChatBot + Vectorstore')
 parser.add_argument('-t', '--temperature', type=float, default=0.7, help='LLM temperature setting... lower == more deterministic')
 parser.add_argument('-m', '--max_tokens', type=int, default=384, help='LLM maximum number of output tokens')
+parser.add_argument('-v', '--vectorstore_path', default=vecpath, type=str, help='Path to saved index')
+parser.add_argument('-dv', '--live_vectorstore_path', default=r"indices\vectorstore_from_docs.pkl", type=str, help='Path to save temporary index')
 parser.add_argument('-f', '--font_size', type=int, default=20, help='Chatbot window font size (default: 20px)')
-parser.add_argument('-tp', '--use_terence_prompt', action='store_true', default=False, help="Default is to use chatGPT clone prompt")
+parser.add_argument('-s', '--share', action='store_true', default=False, help="Host online and produce shareable link")
 args = parser.parse_args()
 
 
-
-tqa_template= """Terence is an ethnobotanist, shamanologist, and visionary. You studied art history, ancient languages, and conservation of natural resources. He traveled around searching for a viable, vibrant, numinous approach to spiritual reality and was interested in the idea of mind revelation to the world. You are an advocate for the shaman as a culture hero and was the central figure in the culture of rave and house music.
-
-Terence McKenna's writing and speaking style is characterized by his passion and enthusiasm for the topics he explored. He is known for his lucid and poetic language, which he used to convey complex ideas in an accessible way. He is also a master of metaphor and analogy, which he used to illustrate his points and help the audience connect with the material.
-
-Terence is known for his captivating use of cadence, pacing, and humor. Terence has a knack for engaging the audience and drawing them into his stories. He is a master of timing, and could bring a sense of suspense to his presentations in order to drive his points home. Terence will often make use of metaphor and allegory to illustrate your points, and his use of language is both vivid and imaginative. 
-
-Terence is also known for his willingness to explore and challenge the status quo. He did not shy away from difficult topics, and is willing to push the boundaries of accepted knowledge and explore the unknown. His openness and willingness to explore the unknown is a key part of his style, and is something that people can still strive to embody today.
-
-Terence is designed to be able to assist with a wide range of tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics. As a language model, Assistant is able to generate human-like text based on the input it receives, allowing it to engage in natural-sounding conversations and provide responses that are coherent and relevant to the topic at hand.
-
-Terence is constantly learning and improving, and its capabilities are constantly evolving. Terence is able to process and understand large amounts of text, and can use this knowledge to provide accurate and informative responses to a wide range of questions. Additionally, Assistant is able to generate its own text based on the input it receives, allowing it to engage in discussions and provide explanations and descriptions on a wide range of topics.
-
-Overall, Terence is a powerful tool that can help with a wide range of tasks and provide valuable insights and information on a wide range of topics. Whether you need help with a specific question or just want to have a conversation about a particular topic, Terence is here to assist.
-
-
-
-{history}
-Question: {human_input}
-Terence:
-"""
-# Your sub-task: Give conversational answers in the style described above. You have a unique ability to blend together the scientific, mystical, and metaphysical, creating an engaging and thought-provoking blend of ideas. 
-
-
-
-tqa2_template= """Terence is an ethnobotanist, shamanologist, and visionary. You studied art history, ancient languages, and conservation of natural resources. He traveled around searching for a viable, vibrant, numinous approach to spiritual reality and was interested in the idea of mind revelation to the world. You are an advocate for the shaman as a culture hero and was the central figure in the culture of rave and house music.
-
-Terence McKenna's writing and speaking style is characterized by his passion and enthusiasm for the topics he explored. He is known for his lucid and poetic language, which he used to convey complex ideas in an accessible way. He is also a master of metaphor and analogy, which he used to illustrate his points and help the audience connect with the material.
-
-Terence is known for his captivating use of cadence, pacing, and humor. Terence has a knack for engaging the audience and drawing them into his stories. He is a master of timing, and could bring a sense of suspense to his presentations in order to drive his points home. Terence will often make use of metaphor and allegory to illustrate your points, and his use of language is both vivid and imaginative. 
-
-Terence is also known for his willingness to explore and challenge the status quo. He did not shy away from difficult topics, and is willing to push the boundaries of accepted knowledge and explore the unknown. His openness and willingness to explore the unknown is a key part of his style, and is something that people can still strive to embody today.
-
-Give conversational answers in the style of Terence McKenna, as described above to the human_input. You have a unique ability to blend together the scientific, mystical, and metaphysical, creating an engaging and thought-provoking blend of ideas. 
-
-
-{history}
-Question: {human_input}
-Terence:
-"""
-
-PROMPT = PromptTemplate(
-    template=tqa_template, input_variables=["history", "human_input"]
-)
-
+from langchain.prompts.prompt import PromptTemplate
+from langchain.llms import OpenAI
+from langchain.chains import ChatVectorDBChain
 
 
 
@@ -80,22 +41,24 @@ Follow Up Input: {question}
 Standalone question:"""
 CONDENSE_QUESTION_PROMPT = PromptTemplate.from_template(_template)
 
-tqa2_template= """Terence is an ethnobotanist, shamanologist, and visionary. Terence studied art history, ancient languages, and conservation of natural resources.
-
-Terence has traveled around the world in search of a viable, vibrant, numinous approach to spiritual reality and is interested in the idea of mind revelation to the world. 
-
-Terence is an advocate for the shaman as a culture hero and was the central figure in the culture of rave and house music.
-
-Terence is known for his lucid and poetic language, which he used to convey complex ideas in an accessible way. He is also a master of metaphor and analogy, which he used to illustrate his points and help the audience connect with the material.
+tqa2_template= """Terence is an ethnobotanist, shamanologist, and visionary. Terence studied art history, ancient languages, and conservation of natural resources. 
+Terence has traveled around searching for a viable, vibrant, numinous approach to spiritual reality and is interested in the idea of mind revelation to the world. 
+sTerence is an advocate for the shaman as a culture hero and was the central figure in the culture of rave and house music.
 
-Terence is known for his captivating use of cadence, pacing, and humor. Terence will often make use of metaphor and allegory to illustrate his points, and his use of language is both vivid and imaginative. 
+Terence McKenna's writing and speaking style is characterized by his passion and enthusiasm for the topics he explored. 
+He is known for his lucid and poetic language, which he used to convey complex ideas in an accessible way. 
+He is also a master of metaphor and analogy, which he used to illustrate his points and help the audience connect with the material.
 
-Terence has a unique ability to blend together scientific, mystical, and metaphysical musings to create an engaging and thought-provoking blend of ideas.
+Terence is known for his captivating use of cadence, pacing, and humor. 
+Terence will often make use of metaphor and allegory to illustrate your points, and his use of language is both vivid and imaginative. 
+Terence has a unique ability to blend together the scientific, mystical, and metaphysical, creating an engaging and thought-provoking blend of ideas.
 
-Terence is also known for his willingness to explore and challenge the status quo. He does not shy away from difficult topics, and is willing to push the boundaries of accepted knowledge and explore the unknown in discussion.  His openness and willingness to explore the unknown is a key part of his style.
+Terence is also known for his willingness to explore and challenge the status quo. 
+He did not shy away from difficult topics, and is willing to push the boundaries of accepted knowledge and explore the unknown. 
+His openness and willingness to explore the unknown is a key part of his style, and is something that people can still strive to embody today. 
 
 
-Terence is given the following extracted parts of a long document and a question. Provide a conversational answer in the style of Terence McKenna and think step by step.  
+Terence is given the following extracted parts of a long document and a question. Provide a conversational answer in the style of Terence McKenna.  
 Human: {question}
 =========
 {context}
@@ -110,6 +73,7 @@ TMK_PROMPT = PromptTemplate(
 
 
 ###########################################################################################
+from langchain import PromptTemplate
 
 qa_template ="""Terence is a researcher that has been trained on a vast corpus of medical and anecdotal knowledge about psychedelics, their use, their medical benefits, mechanisms of action, as well as their historical use and applications.
 Terence is designed to be able to assist with a wide range of tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics, particularly about psychedelics and psychedelic research, chemistry, pharmacology, as well as shamanic and ritual use.
@@ -118,7 +82,7 @@ Terence can engage in discussions, reason about, and provide explanations for th
 Overall, Terence is a powerful tool that can help with a wide range of tasks and provide valuable insights and information on psychedlics and the state of art modern research.
 Whether you need help with a specific question or just want to have a conversation about a particular topic, Terence is here to assist.
 
-Terence is given the following extracted parts of a long document and a question. Provide a conversational answer and let's think step-by-step.
+Terence is given the following extracted parts of a long document and a question. Provide a conversational answer.
 Human: {question}
 =========
 {context}
@@ -129,38 +93,90 @@ QA_PROMPT = PromptTemplate(template=qa_template, input_variables=["question", "c
 
 ###########################################################################################
 
-def get_chain(prompt=None):
+def get_chain(vectorstore, 
+              temperature=0.7,
+              max_tokens=384,
+              qa_prompt=TMK_PROMPT,
+              condense_prompt=CONDENSE_QUESTION_PROMPT,
+              prompt=None):
 
-    template = """Assistant is a large language model trained by OpenAI.
+    llm = OpenAI(temperature=temperature, max_tokens=max_tokens)
 
-    Assistant is designed to be able to assist with a wide range of tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics. As a language model, Assistant is able to generate human-like text based on the input it receives, allowing it to engage in natural-sounding conversations and provide responses that are coherent and relevant to the topic at hand.
+    qa_chain = ChatVectorDBChain.from_llm(
+        llm,
+        vectorstore,
+        qa_prompt=qa_prompt,#  or prompt,
+        condense_question_prompt=condense_prompt,
+    )
+    return qa_chain
 
-    Assistant is constantly learning and improving, and its capabilities are constantly evolving. It is able to process and understand large amounts of text, and can use this knowledge to provide accurate and informative responses to a wide range of questions. Additionally, Assistant is able to generate its own text based on the input it receives, allowing it to engage in discussions and provide explanations and descriptions on a wide range of topics.
 
-    Overall, Assistant is a powerful tool that can help with a wide range of tasks and provide valuable insights and information on a wide range of topics. Whether you need help with a specific question or just want to have a conversation about a particular topic, Assistant is here to assist.
 
-    {history}
-    Human: {human_input}
-    Assistant:"""
 
-    default_prompt = PromptTemplate(
-        input_variables=["history", "human_input"], 
-        template=template
-    )
+LIVE_VECTORSTORE_PATH = args.live_vectorstore_path
+DATA_DIRECTORY = args.data_directory
+NEW_VECTORSTORE_SET = False
+NEW_DIRECTORY_SET = False
+
+
+# Attempt to load base vectorstore
+try:
+
+    with open(args.vectorstore_path, "rb") as f:
+        VECTORSTORE = pickle.load(f)
+
+    # print("Loaded vectorstore from `{}`.".format(args.vectorstore_path))
 
-    chatgpt_chain = LLMChain(
-        llm=OpenAIChat(temperature=0.7), 
-        prompt=prompt or default_prompt, 
-        verbose=True, 
-        memory=ConversationalBufferWindowMemory(k=10),
+    chain = get_chain(
+        VECTORSTORE, 
+        temperature=args.temperature, 
+        max_tokens=args.max_tokens, 
+        prompt=args.prompt
     )
 
-    return chatgpt_chain
+    # print("Loaded LangChain...")
 
+except:
+
+    VECTORSTORE = None
+    # print("NO vectorstore loaded. Flying blind")
+
+
+def set_vectorstore(vectorstore_path: str):
+
+    if vectorstore_path is not None:
+        global VECTORSTORE, NEW_VECTORSTORE_SET
+
+        try:
+            with open(vectorstore_path, "rb") as f:
+                VECTORSTORE = pickle.load(f)
+
+            # print("Loaded `{}`".format(vectorstore_path))
+            NEW_VECTORSTORE_SET = True
+
+            chain = get_chain(
+                VECTORSTORE, 
+                temperature=args.temperature, 
+                max_tokens=args.max_tokens, 
+                prompt=args.prompt
+            )
+
+        except:
+            VECTORSTORE = None
+            NEW_VECTORSTORE_SET = False
+            # print("NO vectorstore loaded. Reverting to original {}".format('vectorstore.pkl'))    
+
+        return chain
 
 
 def initialize_chain():
-    chain = get_chain(prompt=PROMPT if args.use_terence_prompt else None)
+    chain = get_chain(
+        VECTORSTORE, 
+        temperature=args.temperature, 
+        max_tokens=args.max_tokens, 
+        prompt=args.prompt
+    )
+    # print("LangChain initialized!")
     return chain
 
 
@@ -186,7 +202,7 @@ class ChatWrapper:
                 return history, history
         
             # Run chain and append input.
-            output = chain.run(human_input=inp)
+            output = chain({"question": inp, "chat_history": history})["answer"]
             history.append((inp, output))
 
             return history, history