Update app.py

app.py

@@ -121,7 +121,7 @@ def chat(message, history):
     history_useful = add_note_to_history(response, history_useful)
     list_history = history_useful[0].split('</s> <s>')
     history.append((list_history[-2], list_history[-1]))
-    store_message((list_history[-2], list_history[-1])) # Save to dataset
+    store_message(message, response) # Save to dataset
     return history, history
 
 gr.Interface(