added a prompt template

app.py

@@ -7,6 +7,7 @@ from langchain.chains import ConversationalRetrievalChain
 from langchain.chat_models import ChatOpenAI
 from langchain.memory import ConversationBufferMemory
 from langchain.vectorstores import Pinecone 
+from langchian.prompts.prompt import PromptTemplate
 
 
 BOOK_TOKEN = os.getenv("book")
@@ -56,9 +57,18 @@ with gr.Blocks() as demo:
             memory_key='chat_history',
             return_messages=False
         )
+
+        #PUT IT IN A PROMPT TEMPLATE
+        template = """The following is chat between a human and an AI assistant. The AI provides the answer along with the section it referred to for the answer.
+        Current Conversation:
+        {history}
+        Friend: {input}
+        AI:
+        """
+        PROMPT = PromptTemplate(input_variables=["history", "input"], template=template)
     
         #Initalize lanchain - Conversation Retrieval Chain
-        qa = ConversationalRetrievalChain.from_llm(ChatOpenAI(temperature=0), retriever=db.as_retriever(), memory=memory)
+        qa = ConversationalRetrievalChain.from_llm(ChatOpenAI(temperature=0), retriever=db.as_retriever(), memory=memory, prompt=PROMPT)