Update LegalBot2.py

LegalBot2.py

@@ -5,7 +5,7 @@ import streamlit as st
 # from langchain.chat_models import ChatOpenAI
 from langchain.chains import RetrievalQA
 # from langchain import PromptTemplate
-# from langchain.memory import ConversationBufferMemory
+from langchain.memory import ConversationBufferMemory
 # from langchain.agents import initialize_agent, Tool, AgentExecutor
 # from langchain.text_splitter import CharacterTextSplitter
 import openai
@@ -15,7 +15,7 @@ from langchain_openai import OpenAIEmbeddings, ChatOpenAI
 from langchain_community.vectorstores import Chroma
 from langchain_core.prompts import PromptTemplate
 # from langchain_community.chains import RetrievalQA
-from langchain_community.memory import ConversationBufferMemory
+# from langchain_community.memory import ConversationBufferMemory
 from langchain_community.agents import initialize_agent, Tool, AgentExecutor
 from langchain_core.text_splitters import CharacterTextSplitter