input: Write a function to generate an RSA public/private key pair and save them as files named 'public.pem' and 'private.pem'. If the function is called, it should generate the keys and store them accordingly.
  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)
  