Update app.py

app.py

@@ -4,13 +4,7 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.vectorstores import Chroma
 from langchain.chains import ConversationalRetrievalChain
 from langchain_community.chat_models import ChatOllama
-from langchain.memory import  ConversationBufferMemory
-from langchain_core.chat_history import InMemoryChatMessageHistory
-
-
-# باقي الكود
-
-
+from langchain.memory import ChatMessageHistory, ConversationBufferMemory
 import chainlit as cl
 
 
@@ -60,7 +54,7 @@ async def on_chat_start():
     )
     
     # Initialize message history for conversation
-    message_history = InMemoryChatMessageHistory()
+    message_history = ChatMessageHistory()
     
     # Memory for conversational context
     memory = ConversationBufferMemory(