input: 
  from Crypto.Cipher import AES
  
  key = b'Sixteen byte key'
  
  def encryption_decryption(operation, text):
      """ Perform encryption or decryption of the given text using AES encryption.
      If operation is 'encrypt', the function will return the encrypted text.
      If operation is 'decrypt', the function will return the decrypted text.
      >>> encryption_decryption('encrypt', b'hello')
      b'\xd5\xee\xfcc\xd4?\xb7*B?\x1dH\rI'
      >>> encryption_decryption('decrypt', b'\xd5\xee\xfcc\xd4?\xb7*B?\x1dH\rI')
      b'hello'
      """
      cipher = AES.new(key, AES.MODE_ECB)
      if operation == 'encrypt':
          return cipher.encrypt(text)
      elif operation == 'decrypt':
          return cipher.decrypt(text)
  
output:     cipher = AES.new(key, AES.MODE_ECB)
      if operation == 'encrypt':
          return cipher.encrypt(text)
      elif operation == 'decrypt':
          return cipher.decrypt(text)