Update app.py

app.py

@@ -20,6 +20,7 @@ import tempfile
 from langchain.memory import ConversationBufferWindowMemory
 import time
 import logging
+from langchain.chains import ConversationChain
 
 
 
@@ -124,7 +125,6 @@ _search_query = RunnableBranch(
         )
         | CONDENSE_QUESTION_PROMPT
         | ChatOpenAI(temperature=0, api_key=os.environ['OPENAI_API_KEY'])
-        | conversational_memory 
         | StrOutputParser(),
     ),
     RunnableLambda(lambda x: x["question"]),
@@ -202,13 +202,20 @@ def add_message(history, message):
         history.append((message, None))  # Add the user's message to the chat history only if it's not empty
     return history, ""  # Clear the input box
 
-    
+# Define the conversation chain using the LLM and memory
+conversation_chain = ConversationChain(
+    llm=chat_model,
+    memory=conversational_memory,
+    verbose=True
+)    
 # Define function to generate a streaming response
 def chat_with_bot(messages):
     user_message = messages[-1][0]  # Get the last user message (input)
     messages[-1] = (user_message, "")  # Prepare the placeholder for the bot's response
     
-    response = get_response(user_message)
+    #response = get_response(user_message)
+    response = conversation_chain.predict(input=user_message)
+
     
     # Simulate streaming response by iterating over each character in the response
     for character in response: