import argparse, json, sys, time, random
import utterance.voice
import utterance.utils

UTTERANCE_LEN = 64				#<--------------- these should be in config
				
def main() -> int:

	p = argparse.ArgumentParser()
	p.add_argument("-c", "--config", type=str, default="voice.config.json", help="configuratin file")
	p.add_argument("-i", "--iterations", type=int, default=10, help="number of iterations")
	args = p.parse_args()

	print(args)

	with open(args.config) as f:
		conf = json.load(f)

	voices = []
	for v in conf['voices']:
		model = v['model']
		voice = utterance.voice.Voice(name=v["name"].upper(), model=model['model_dir'], tokenizer=model['tokeniser_file'], temp=float(model["temperature"]), length=UTTERANCE_LEN)
		voices.append(voice)

	current_voice_name = ""
	for i in range(args.iterations):
		v = random.sample(voices, 1)[0]
		if v.name != current_voice_name:
			current_voice_name = v.name
			print("==========")
			print(v.name + ":")
			t = v.utter_one()
			t = utterance.utils.clean(t)
			t = utterance.utils.format(t)
			print(t)
		
		# time.sleep(1)


if __name__ == '__main__':
    sys.exit(main())