import openai
openai.api_key='YOUR API KEY'

response = openai.Image.create_edit(
  image=open("sunlit_lounge.png", "rb"),
  mask=open("mask.png", "rb"),
  prompt="A sunlit indoor lounge area with a pool containing a flamingo",
  n=2,
  size="1024x1024"
)
for i in range(2):    
  image_url = response['data'][i]['url']
  print(image_url)