Update app.py

app.py

@@ -58,12 +58,24 @@ 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(response.url)
+    image_response = requests.get(image_url)
     image = Image.open(BytesIO(image_response.content))
 
     return image
 
+
 def mood_art_generator(image):
     mood = get_mood_from_image(image)
     print("Mood:", mood)