output_file = open("RedBlack_tree.txt", "w+", encoding = 'utf-8') tree._write() output_file.close() while True: _inputuser = input("Please enter the depth of nodes you would like printed to file: ") # Checks if the input is valid for the tree. if int(_inputuser) >= int(tree._height()) or int(_inputuser) < 0: print("Depth is not valid, please choose another depth size" ) continue else: break #Depth of tree is read k=int(_inputuser) depth_file = open("RB_depth.txt", "w+", encoding="utf-8") tree._depth(k) depth_file.close() if _input is "1": #AVL Tree tree = AVLTree() read_file() print("Nodes in tree: "+ str(tree._size())) print("\n") print("Height: " + str(tree._height())) output_file = open("AVL_tree.txt", "w+", encoding = 'utf-8') tree._write() output_file.close() while True: