Добавить with или обработку исключения

text_generator

	with open(model, 'w') as file: # writing model to file
	for bigram in d:
	w1, w2 = bigram
	file.write('{} {} {}\n'.format(w1, w2, d[bigram]))

	parse = ap.ArgumentParser(description='Train on some text')
	parse.add_argument('-i', '--input-dir', help='Input directory',
	default='', required=True)
	parse.add_argument('-m', '--model',
	help='Path to the file where model will be saved',
	required=True)
	parse.add_argument('--lc', action='store_true', help='Make text lowercase')
	args = parse.parse_args()
	train(args.input_dir, args.model, args.lc)

	def get_next_word(model, current_word):
	"""
	:param model: dict containing model
	:param current_word: last outputted word in text
	"""

	for i in range(length):
	output.write('{} '.format(current_word))
	current_word = get_next_word(model, current_word)

	output.write('{}\n'.format(current_word))
	else:
	output.write('{} '.format(current_word))

	total = sum(w for w in model[current_word].values())
	r = random.uniform(0, total)
	upto = 0
	for c, w in model[current_word].items():
	if upto + w >= r:
	return c
	upto += w

	if seed == '':
	return random.choice(list(model.keys()))
	else:
	if seed not in model.keys():
	raise ValueError('Invalid seed')
	return seed