Fixed Previous Error

There was an error saying ChatbotTrainer was None, It's now fixed.

runCorpus.py

@@ -1,7 +1,8 @@
-# runMovieCorpus.py
+# self.runMovieCorpus.py
 import os
 from playsound3 import playsound
 import tensorflow
+from chatbotTrainer import ChatbotTrainer
 import time
 import random
 import pdb
@@ -10,12 +11,10 @@ import matplotlib.pyplot as plt
 
 # Personal files
 from preprocessed_dialogs import dialog_data
-from chatbotTrainer import ChatbotTrainer
 
 
 class CorpusTrainer:
 	def __init__(self):
-		self.chatbot_trainer = None
 		self.running_trouble = None
 		self.choices_yes = ["yes", "ya", "yeah", "yessir", "yesir", "y", "ye", "yah"]
 		self.exit_commands = ["exit", "quit", "stop", "x", "q", ""]
@@ -61,6 +60,7 @@ class CorpusTrainer:
 
 		# Initialize the corpus (Needed for convo-kit to initialize) Only needed when reading/creating the dialogs
 		self.corpus_path = '/root/.convokit/saved-corpora/movie-corpus'
+		self.chatbot_trainer = ChatbotTrainer()
 		self.chatbot_trainer.load_corpus(self.corpus_path)
 
 	def main(self, chatbot_trainer, user_choice, dialog_data, topConvo=0, top_num=0, play_notification=0):
@@ -340,12 +340,8 @@ class CorpusTrainer:
 
 if __name__ == "__main__":
 	while True:
-		# Initialize the corpus (Needed for convo-kit to initialize) Only needed when reading/creating the dialogs
-		corpus_path = '/root/.convokit/saved-corpora/movie-corpus'
-		chatbot_trainer = ChatbotTrainer()
-		chatbot_trainer.load_corpus(corpus_path)
 		app = CorpusTrainer()
-		user_choice = input(f"Run Supervised?({chatbot_trainer.model_filename})\n>")
+		user_choice = input(f"Run Supervised?({app.chatbot_trainer.model_filename})\n>")
 		play_notification = input(f"Would you like to play a notification after each training?\nHelps with manual stopping before max_vocabulary reached...  \n>")
-		app.main(chatbot_trainer=chatbot_trainer, user_choice=user_choice, dialog_data=dialog_data, play_notification=play_notification)
+		app.main(chatbot_trainer=app.chatbot_trainer, user_choice=user_choice, dialog_data=dialog_data, play_notification=play_notification)