Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 from googletrans import Translator
 import requests
 from dotenv import load_dotenv
-import numpy as np
+# import numpy as np
 from langchain_community.vectorstores import Chroma
 from langchain_community.document_loaders import UnstructuredPDFLoader, PyPDFLoader
 from langchain.text_splitter import CharacterTextSplitter
@@ -77,7 +77,13 @@ def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, pr
     if llm_name != default_llm:
         print(f"Using default LLM {default_llm} for {language}")
         llm_name = default_llm
-
+    
+    memory = ConversationBufferMemory(
+        memory_key="chat_history",
+        output_key='answer',
+        return_messages=True
+    )
+    
     qa_chain = ConversationalRetrievalChain.from_llm(
         llm=llm_name,
         retriever=vector_db.as_retriever(),