Update app.py

app.py

@@ -3,9 +3,8 @@ import json
 from io import BytesIO
 from PIL import Image
 import matplotlib.pyplot as plt
-pip install requests json Pillow matplotlib
-# Replace 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 with your OpenAI API key
-API_KEY = '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'
+
+API_KEY = 'YOUR_API_KEY'
 ENDPOINT = 'https://api.openai.com/v1/images/generations'
 
 def generate_image(prompt):
@@ -22,14 +21,22 @@ def generate_image(prompt):
         'response_format': 'url'
     }
 
-    response = requests.post(url=ENDPOINT, headers=headers, data=json.dumps(data))
-    if response.status_code == 200:
+    try:
+        response = requests.post(url=ENDPOINT, headers=headers, data=json.dumps(data))
+        response.raise_for_status()
         result_url = response.json()['data'][0]['url']
         image_bytes = requests.get(result_url).content
         image = Image.open(BytesIO(image_bytes))
         return image
-    else:
-        return None
+    except requests.exceptions.HTTPError as e:
+        print(f"HTTP error {e.response.status_code}: {e.response.reason}")
+        print(f"Error details: {e.response.text}")
+    except requests.exceptions.RequestException as e:
+        print(f"Request error: {e}")
+    except Exception as e:
+        print(f"Unexpected error: {e}")
+
+    return None
 
 # Example usage
 prompt = "a cat sitting on a windowsill looking outside"
@@ -37,5 +44,4 @@ image = generate_image(prompt)
 if image:
     plt.imshow(image)
     plt.show()
-else:
-    print("Error generating image")
+