Trying to prevent black image

app.py

@@ -43,8 +43,11 @@ def get_mood_from_image(image: Image.Image):
     selected_mood = max(mood_scores, key=mood_scores.get)
 
     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):
+def generate_art(mood, max_retries=10):
     # 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))})"
     
@@ -57,7 +60,8 @@ def generate_art(mood):
         "inputs": prompt
     }
 
-    while True:
+    retries = 0
+    while retries < max_retries:
         response = requests.post('https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5', headers=headers, json=json_data)
 
         if response.status_code == 503:
@@ -71,13 +75,21 @@ def generate_art(mood):
             print(response.content)
             return None
 
-        break
+        image = Image.open(BytesIO(response.content))
+        
+        # Check if the image is black
+        if not is_black_image(image):
+            break
 
-    image = Image.open(BytesIO(response.content))
+        retries += 1
+
+    if retries == max_retries:
+        return None
 
     return image
 
 
+
 def mood_art_generator(image):
     mood = get_mood_from_image(image)
     print("Mood:", mood)