def read_script_config(path):
with open(path, 'r') as inf:
return yaml.load(inf)
if __name__ == '__main__':
print '\t--> Reading settings file: "config.yml"'
configs = read_script_config('config.yml')
print '\t--> Configuring script object'
script_configs = configs['script_configs']
movie_script = Script(**script_configs)
print '\t\t--> Loading script from file "{}"'.format(configs['run_configs']['script_path'])
movie_script.load_from_file(configs['run_configs']['script_path'])
print '\t\t--> Parsing script text'
movie_script.parse()
print '\t--> Setting up Markov generator for script notes'
note_generator = MarkovGenerator(movie_script.notes)
print '\t--> Setting up Markov generator for scene settings'
setting_generator = MarkovGenerator(movie_script.scenes)
print '\t--> Setting up Markov generators for character dialog'
character_generators = {}
for character, character_dialog in movie_script.character_dialog.items():
character_generators[character] = MarkovGenerator(character_dialog)
