correct_answer1 = 0
correct_answer2 = 0

print("Hello and welcome to the OptiX standalone denoiser assistant")
print("Let's start off by collecting information needed for denoising")
print()
print()
print()

optix = input("Please enter the path to the OptiX denoiser: ")
print()

beauty = input("Please enter the path to the Noisy pass: ")
print()

albedo = input("Please enter the path to the Albedo pass: ")
print()

normal = input("Please enter the path to the Normal pass: ")
print()

output_path = input("Please enter the path to the Output folder: ")
output_path = output_path[:-2]
print()

output_name = input("Please enter the name of the Output file (E.G. Denoised): ")
print()

while correct_answer1 < 1:
	animation = input("Are you denoising a animation? (y/n) ")
	print()
	if animation == "y":
		correct_answer1 = 1
		beauty = beauty[:-10]
		albedo = albedo[:-10]
		normal = normal[:-10]
		start = input("Please enter the number of the First frame in the animation (E.G. 20): ")
		start = int(start)
		print()
		end = input("Please enter the number of the Last frame in the animation (E.G. 150): ")
		end = int(end)
		print()
		while correct_answer2 < 1:
			temporal = input("Is the animation being denoised with Temporal Denoising? (y/n) ")
			print()
			if temporal == "y":
				correct_answer2 = 1
				flow = input("Please enter the path to the Flow pass: ")
				flow = flow[:-10]
				print()
				print("Copy the command below and paste it into a terminal:")
				print()
				print()
				print()
				print(fr"{optix}-F {start}-{end} -a {albedo}++++.exr' -n {normal}++++.exr' -f {flow}++++.exr' -o {output_path}/{output_name}++++.exr' {beauty}++++.exr'")
			elif temporal == "n":
				correct_answer2 = 1
				print("Copy the commands below and paste it into a .sh file and run it:")
				print()
				print()
				print()
				while start < end+1:
					if start < 10:
						print(fr"{optix}-a {albedo}000{start}.exr' -n {normal}000{start}.exr' -o {output_path}/{output_name}000{start}.exr' {beauty}000{start}.exr'")
						start = start+1
					elif start < 100:
						print(fr"{optix}-a {albedo}00{start}.exr' -n {normal}00{start}.exr' -o {output_path}/{output_name}00{start}.exr' {beauty}00{start}.exr'")
						start = start+1
					elif start < 1000:
						print(fr"{optix}-a {albedo}0{start}.exr' -n {normal}0{start}.exr' -o {output_path}/{output_name}0{start}.exr' {beauty}0{start}.exr'")
						start = start+1
					elif start < 10000:
						print(fr"{optix}-a {albedo}{start}.exr' -n {normal}{start}.exr' -o {output_path}/{output_name}{start}.exr' {beauty}{start}.exr'")
						start = start+1
	elif animation == "n":
		correct_answer1 = 1
		print("Copy the command below and paste it into a terminal:")
		print()
		print()
		print()
		print(fr"{optix}-a {albedo}-n {normal}-o {output_path}/{output_name}.exr' {beauty}")