main.py

import sys
from chatbot.chatbot import ChatBot
from trainer.train import TrainChatBot

#first parameter
input_command = sys.argv[1]

#second parameter (default = -1)
try:
 input_attribute = sys.argv[2]
except IndexError:
 input_attribute = -1

def run(command, attribute):
 chatbot = ChatBot('Chatbot')

 if command == 'reply':
 chatbot.respond(attribute)

 elif command == 'reply_instant':
 response = chatbot.respondInstant(attribute)
 print(response)

 elif command == 'train':
 trainer = TrainChatBot(chatbot)
 trainer.train()

 elif command == 'end':
 ChatBot.endChat(attribute)

 elif command == 'serve':
 server.run()

 else:
 print("Invalid Command")
 print("Available Commands are [reply, reply_instant, train, end]")


#run function
run(input_command, input_attribute)