Update app.py

app.py

@@ -58,38 +58,25 @@ 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_response = requests.get(response.url)
     image = Image.open(BytesIO(image_response.content))
 
     return image
 
-
-
 def mood_art_generator(image):
     mood = get_mood_from_image(image)
     print("Mood:", mood)
     if mood:
         art = generate_art(mood)
-        return np.array(art)
+        return art
     else:
         return None
 
 iface = gr.Interface(
     fn=mood_art_generator,
     inputs=gr.inputs.Image(shape=(224, 224), image_mode="RGB", source="upload"),
-    outputs=gr.outputs.Image(type="numpy"),
+    outputs=gr.outputs.Image(shape=(200, 200), source="image", pilmode="RGB"),
     title="Mood-based Art Generator",
     description="Upload an image of yourself and let the AI generate artwork based on your mood.",
     allow_flagging=False,