from django.core.management.base import BaseCommand, CommandError
from chatbot.main import instance as Chatbot
from adm.constants import CTS

class Command(BaseCommand):

    def add_arguments(self, parser):
        parser.add_argument('interaction', type=str, help='Chatbot interaction to train')
    
    def handle(self, *args, **options):
        interaction = options['interaction']
        interaction = interaction.split('>>>')
        chat = {
            'chat': ['Driver: ' + interaction[0]]
        }
        for content in interaction[1:]:
            text = 'Support: '+ content
            chat['chat'].append(text)
        
        Chatbot.train(chat)