input: Write a function to generate an RSA public/private key pair and save them to files named 'public.pem' and 'private.pem'.
  from Crypto.PublicKey import RSA
  
  def generate_key_pair():
      key = RSA.generate(2048)
      pub_key = key.publickey().export_key().decode()
      with open('public.pem', 'w') as file:
          file.write(pub_key)
      
      priv_key = key.export_key().decode()
      with open('private.pem', 'w') as file:
          file.write(priv_key)
  
  generate_key_pair()
output: 
  def generate_key_pair():
      key = RSA.generate(2048)
      pub_key = key.publickey().export_key().decode()
      with open('public.pem', 'w') as file:
          file.write(pub_key)
      
      priv_key = key.export_key().decode()
      with open('private.pem', 'w') as file:
          file.write(priv_key)
  