def main():
    text = input("Enter text: ")
    encrypt_answer = input("(E)ncrypt or (D)ecrypt?" )
    if encrypt_answer == "E":
        print("encrypt function goes here")



    if encrypt_answer == "D":
        print("decrypt function goes here")

main()