added prompt again

app.py

@@ -7,7 +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
+from langchain.prompts.prompt import PromptTemplate
 
 
 BOOK_TOKEN = os.getenv("book")
@@ -58,17 +58,17 @@ with gr.Blocks() as demo:
             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)
+        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)