Update app.py

app.py

@@ -7,7 +7,7 @@ from langchain.indexes.vectorstore import VectorStoreIndexWrapper
 from langchain.embeddings import OpenAIEmbeddings
 from typing_extensions import Concatenate
 from datasets import load_dataset
-
+from langchain.memory import ConversationBufferWindowMemory
 import cassio
 
 from PyPDF2 import PdfReader
@@ -34,8 +34,11 @@ def main():
             raw_text += content
  
         cassio.init(token=ASTRA_DB_APPLICATION_TOKEN,database_id=ASTRA_DB_ID)
+
+        
+        memor=ConversationBufferWindowMemory(k=1)
+        llm=OpenAI(openai_api_key=OPENAI_API_KEY,memory=memor)
         
-        llm=OpenAI(openai_api_key=OPENAI_API_KEY)
         embedding=OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)
         
         astra_vector_store=Cassandra(embedding=embedding,