Handle case when request is rejected

Author: nathan-fieldstack-com

src/dalle2/dalle2.py

@@ -82,6 +82,9 @@ def get_task_response(self, body):
             if data["status"] == "failed":
                 print(f"Task failed: {data['status_information']}")
                 return None
+            if data["status"] == "rejected":
+                print(f"Task rejected: {data['status_information']}")
+                return None
             if data["status"] == "succeeded":
                 print("🙌 Task completed!")
                 return data["generations"]["data"]