Update app.py

app.py

@@ -36,6 +36,7 @@ def get_mood_from_image(image: Image.Image):
     selected_mood = max(mood_scores, key=mood_scores.get)
 
     return selected_mood
+
 def generate_art(mood):
     # Implement art generation logic using the Stable Diffusion API
     prompt = f"{mood} generative art"
@@ -55,9 +56,15 @@ def generate_art(mood):
         print(f"Error: API response status code {response.status_code}")
         print("Response content:")
         print(response.content)
-        return None  # Add this line to return early in case of an error
+        return None
 
-    response_json = response.json()
+    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"]
@@ -69,6 +76,7 @@ def generate_art(mood):
     return image
 
 
+
 def mood_art_generator(image):
     mood = get_mood_from_image(image)
     print("Mood:", mood)