Update app.py

app.py

@@ -1,9 +1,11 @@
 import requests
 import json
+from io import BytesIO
 from PIL import Image
+import matplotlib.pyplot as plt
 
 # Replace YOUR_API_KEY with your OpenAI API key
-API_KEY = 'YOUR_API_KEY'
+API_KEY = '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'
 ENDPOINT = 'https://api.openai.com/v1/images/generations'
 
 def generate_image(prompt):
@@ -23,7 +25,8 @@ def generate_image(prompt):
     response = requests.post(url=ENDPOINT, headers=headers, data=json.dumps(data))
     if response.status_code == 200:
         result_url = response.json()['data'][0]['url']
-        image = Image.open(requests.get(result_url, stream=True).raw)
+        image_bytes = requests.get(result_url).content
+        image = Image.open(BytesIO(image_bytes))
         return image
     else:
         return None
@@ -32,29 +35,7 @@ def generate_image(prompt):
 prompt = "a cat sitting on a windowsill looking outside"
 image = generate_image(prompt)
 if image:
-    image.show()
-else:
-    print("Error generating image")
-# A red sports car driving on a winding road
-prompt = "a red sports car driving on a winding road"
-image = generate_image(prompt)
-if image:
-    image.show()
-else:
-    print("Error generating image")
-
-# A house in a forest with a river nearby
-prompt = "a house in a forest with a river nearby"
-image = generate_image(prompt)
-if image:
-    image.show()
-else:
-    print("Error generating image")
-
-# A slice of pepperoni pizza on a plate
-prompt = "a slice of pepperoni pizza on a plate"
-image = generate_image(prompt)
-if image:
-    image.show()
+    plt.imshow(image)
+    plt.show()
 else:
     print("Error generating image")