Update app.py

app.py

@@ -48,21 +48,28 @@ def generate_art(mood):
         "inputs": prompt
     }
 
-    response = requests.post('https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5', headers=headers, json=json_data).json()
+    response = requests.post('https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5', headers=headers, json=json_data)
 
-    # Print the response for debugging
-    print("Response:", response)
+    # Check if the response status is not 200 (OK)
+    if response.status_code != 200:
+        print(f"Error: API response status code {response.status_code}")
+        print("Response content:")
+        print(response.content)
+        return None
+
+    response_json = response.json()
 
     # Extract the generated image URL from the response
-    image_url = response["generated_images"][0]["url"]
+    image_url = response_json["generated_images"][0]["url"]
 
     # Download the image and return it as a PIL Image
     image_response = requests.get(image_url)
     image = Image.open(BytesIO(image_response.content))
-    print("Art Image Mode:", image.mode)
+
     return image
 
 
+
 def mood_art_generator(image):
     mood = get_mood_from_image(image)
     print("Mood:", mood)