Added timeout request from api call to prevent infinite looping

app.py

@@ -47,13 +47,12 @@ def is_black_image(image: Image.Image) -> bool:
     img_array = np.array(image)
     return np.all(img_array == 0)
 
-def generate_art(mood, max_retries=10):
-    # Implement art generation logic using the Stable Diffusion API
+def generate_art(mood, max_retries=3, request_timeout=30):
     prompt = f"{mood} generative art with vibrant colors and intricate patterns ({str(np.random.randint(1, 10000))})"
     
     headers = {
         "Authorization": f"Bearer {STABLE_DIFFUSION_API_KEY}",
-        "Accept": "image/jpeg",  # Set the Accept header to receive an image directly
+        "Accept": "image/jpeg",
     }
 
     json_data = {
@@ -62,7 +61,12 @@ def generate_art(mood, max_retries=10):
 
     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)
+        try:
+            response = requests.post('https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5', headers=headers, json=json_data, timeout=request_timeout)
+        except requests.exceptions.Timeout:
+            print(f"Request timed out after {request_timeout} seconds. Retrying...")
+            retries += 1
+            continue
 
         if response.status_code == 503:
             print("Model is loading, waiting for 30 seconds before retrying...")
@@ -76,8 +80,7 @@ def generate_art(mood, max_retries=10):
             return None
 
         image = Image.open(BytesIO(response.content))
-        
-        # Check if the image is black
+
         if not is_black_image(image):
             break
 
@@ -88,8 +91,6 @@ def generate_art(mood, max_retries=10):
 
     return image
 
-
-
 def mood_art_generator(image):
     mood = get_mood_from_image(image)
     print("Mood:", mood)