import logging

from rasa_core import utils, train, run
from rasa_core.training import interactive
from rasa_core.interpreter import RasaNLUInterpreter

logger = logging.getLogger(__name__)


def train_agent(interpreter):
    return train.train_dialogue_model(domain_file="domain.yml",
                                      stories_file="data/stories.md",
                                      output_path="models/dialogue",
                                      interpreter=interpreter,
                                      policy_config='nlu_config.yml'
                                      )


if __name__ == '__main__':
    utils.configure_colored_logging(loglevel="INFO")
    interpreter = RasaNLUInterpreter('./models/default/healthbot')
    agent = train_agent(interpreter)
    interactive.run_interactive_learning(agent, stories='data/stories.md')