Update app.py

app.py

@@ -66,7 +66,7 @@ prompt = PromptTemplate(
 # Define the ChatOpenAI model
 llm = ChatOpenAI(temperature=0.5, model_name="gpt-3.5-turbo")
 
-memory = ConversationBufferMemory()
+memory = ConversationBufferMemory(memory_key="chat_history")
 # Create a RunnableSequence
 runnable_sequence = RunnableSequence(prompt | llm)
 
@@ -75,13 +75,13 @@ runnable_sequence = RunnableSequence(prompt | llm)
 # Define your get_text_response function
 def get_text_response(user_message, history):
     # Add user message to the memory (for example purposes, assume `add_user_message` is the correct method)
-    memory.save_context({"role": "user", "content": user_message}, None)
+    memory.save_context({"user_message": user_message}, None)
     
     # Use the RunnableSequence to generate a response
     response = runnable_sequence.run(user_message=user_message)
     
     # Add LLM response to the memory (for example purposes, assume `add_ai_message` is the correct method)
-    memory.save_context(None, {"role": "assistant", "content": response})
+    memory.save_context(None, {"chat_history": response})
     
     return response