Had an issue of black images being generated

Idea is to keep looping it in clip to classify it as black image or not and if it is keep generating till it isnt

app.py

@@ -44,6 +44,10 @@ def get_mood_from_image(image: Image.Image):
 
     return selected_mood
 
+def is_black_image(image: Image.Image) -> bool:
+    img_array = np.array(image)
+    return np.all(img_array == 0)
+
 def generate_art(mood):
     # Implement art generation logic using the Stable Diffusion API
     prompt = f"{mood} generative art with vibrant colors and intricate patterns ({str(np.random.randint(1, 10000))})"
@@ -58,22 +62,25 @@ def generate_art(mood):
     }
 
     while True:
-        response = requests.post('https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5', headers=headers, json=json_data)
-
-        if response.status_code == 503:
-            print("Model is loading, waiting for 30 seconds before retrying...")
-            time.sleep(30)
-            continue
-
-        if response.status_code != 200:
-            print(f"Error: API response status code {response.status_code}")
-            print("Response content:")
-            print(response.content)
-            return None
-
-        break
-
-    image = Image.open(BytesIO(response.content))
+        while True:
+            response = requests.post('https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5', headers=headers, json=json_data)
+
+            if response.status_code == 503:
+                print("Model is loading, waiting for 30 seconds before retrying...")
+                time.sleep(30)
+                continue
+
+            if response.status_code != 200:
+                print(f"Error: API response status code {response.status_code}")
+                print("Response content:")
+                print(response.content)
+                return None
+
+            image = Image.open(BytesIO(response.content))
+            
+            # Check if the image is black
+            if not is_black_image(image):
+                break
 
     return image