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)