Add conversations for training statements

chatterbot/trainers.py

@@ -14,8 +14,11 @@ def __init__(self, storage, **kwargs):
         self.chatbot = kwargs.get('chatbot')
         self.storage = storage
         self.logger = logging.getLogger(__name__)
+
         self.show_training_progress = kwargs.get('show_training_progress', True)
 
+        self.training_conversation_id = self.storage.create_conversation()
+
     def get_preprocessed_statement(self, input_statement):
         """
         Preprocess the input statement.
@@ -113,9 +116,13 @@ def train(self, conversation):
                 statement.add_response(
                     Response(previous_statement_text)
                 )
+                self.storage.add_to_conversation(
+                    self.training_conversation_id,
+                    statement,
+                    Statement(text=previous_statement_text)
+                )
 
             previous_statement_text = statement.text
-            self.storage.update(statement)
 
 
 class ChatterBotCorpusTrainer(Trainer):
@@ -163,9 +170,15 @@ def train(self, *corpus_paths):
                             statement.add_response(
                                 Response(previous_statement_text)
                             )
+                            self.storage.add_to_conversation(
+                                self.training_conversation_id,
+                                statement,
+                                Response(text=previous_statement_text)
+                            )
+                        else:
+                            self.storage.update(statement)
 
                         previous_statement_text = statement.text
-                        self.storage.update(statement)
 
 
 class TwitterTrainer(Trainer):
@@ -421,6 +434,4 @@ def train(self):
                             statement.add_response(
                                 Response(previous_statement_text)
                             )
-
-                        previous_statement_text = statement.text
                         self.storage.update(statement)