Update app.py

app.py

@@ -93,8 +93,8 @@ if user_input := st.chat_input("User Input"):
                 
                 # Save context in memory
                 memory.save_context(
-                    {"role": "user", "content": user_input},
-                    {"role": "assistant", "content": response}
+                    {"input": user_input},
+                    {"output": response}
                 )
 
                 # Append messages to session state for display