Update app.py

app.py

@@ -42,7 +42,8 @@ def generate_art(mood):
     prompt = f"{mood} generative art"
 
     headers = {
-        "Authorization": f"Bearer {STABLE_DIFFUSION_API_KEY}"
+        "Authorization": f"Bearer {STABLE_DIFFUSION_API_KEY}",
+        "Accept": "image/jpeg",  # Set the Accept header to receive an image directly
     }
 
     json_data = {
@@ -58,24 +59,13 @@ def generate_art(mood):
         print(response.content)
         return None
 
-    try:  # Add a try-except block to handle JSONDecodeError
-        response_json = response.json()
-    except requests.exceptions.JSONDecodeError:
-        print("Error: Invalid JSON response")
-        print("Response content:")
-        print(response.content)
-        return None
-
-    # Extract the generated image URL from the response
-    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))
+    # Load the image directly from the response content
+    image = Image.open(BytesIO(response.content))
 
     return image
 
 
+
 def mood_art_generator(image):
     mood = get_mood_from_image(image)
     print("Mood:", mood)