Update app.py

app.py

@@ -5,6 +5,7 @@ Swami's Chatbot Alpha Version
 '''
 
 from langchain.vectorstores import FAISS
+from langchain.memory import ConversationBufferMemory
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.llms import OpenAI
 from langchain.chains import RetrievalQA
@@ -33,15 +34,20 @@ Answer:"""
 
 PROMPT = PromptTemplate(
     template= prompt_template,
-    input_variables=["context", "question"]
+    input_variables=["history","context", "question"]
 )
 
-chain = RetrievalQA.from_chain_type(llm= OpenAI(),
+chain = RetrievalQA.from_chain_type(llm= OpenAI(temperature= 0),
                                     chain_type="stuff",
                                     retriever= vectordb.as_retriever(),
-                                    chain_type_kwargs= {'prompt': PROMPT},
+                                    chain_type_kwargs= {'prompt': PROMPT,
+                                                        "verbose": True,
+                                                        "memory": ConversationBufferMemory(
+                                                                  memory_key="history",
+                                                                  input_key="question"),
+                                                        },
                                     return_source_documents= True,
-                                    verbose= False)
+                                    verbose= True)
 
 # --------------------------------------------------------------------------------