Update app.py

app.py

@@ -21,6 +21,8 @@ import tempfile
 import os
 from langchain.llms import OpenAI  # Import the OpenAI class
 from langchain.chat_models import ChatOpenAI  # Import ChatOpenAI
+from langchain.agents import create_chat_conversational_react_description_agent
+from langchain.memory import ConversationBufferMemory
 
 
 
@@ -188,17 +190,26 @@ def get_embeddings_model():
 @st.cache_resource
 def initialize_qa_system(_vector_store):
     try:
-        # Use ChatOpenAI instead of OpenAI
         llm = ChatOpenAI(
             temperature=0,
             model_name="gpt-4",  # Or another OpenAI model like "gpt-3.5-turbo"
             api_key=os.environ.get('OPENAI_API_KEY'),
         )
 
-        qa_pipeline = RetrievalQA.from_chain_type(
+        memory = ConversationBufferMemory(memory_key="chat_history")  # Initialize memory
+
+        # Create the conversational agent
+        qa_pipeline = create_chat_conversational_react_description_agent(
             llm=llm,
-            retriever=_vector_store.as_retriever(),
-            chain_type="stuff"  # Use "stuff" chain type for chat models
+            tools=[
+                Tool(
+                    name="Search",
+                    func=_vector_store.as_retriever(search_kwargs={"k": 2}).get_relevant_documents,
+                    description="useful for when you need to answer questions about the documents you have been uploaded. Input should be a fully formed question.",
+                )
+            ],
+            verbose=True,
+            memory=memory,
         )
         return qa_pipeline
     except Exception as e: