Update app.py

app.py

@@ -69,22 +69,27 @@ def generate_art(mood):
     return image
 
 def mood_art_generator(image):
-    image_encoded = base64.b64encode(image).decode("utf-8")
-    image_encoded = f"data:image/jpeg;base64,{image_encoded}"
-    image = Image.open(BytesIO(base64.b64decode(image_encoded.split(",")[1])))
-    mood = get_mood_from_image(np.array(image))
+    image_encoded = image.decode().split(",")[1]
+    image_decoded = base64.b64decode(image_encoded)
+    image_bytes = BytesIO(image_decoded)
+    image = Image.open(image_bytes)
+
+    mood = get_mood_from_image(image)
     if mood:
         art = generate_art(mood)
-        return art
+        return np.array(art)
     else:
         return None
 
-image_input = gr.inputs.Image()
-outputs = gr.outputs.Image(type = "pil",label="Generated Artwork")
-
-interface = gr.Interface(
+iface = gr.Interface(
     fn=mood_art_generator,
-    inputs=image_input,
-    outputs=outputs,
+    inputs="image",
+    outputs="image",
     title="Mood-based Art Generator",
-    description="Upload an
+    description="Upload an image of yourself and let the AI generate artwork based on your mood.",
+    allow_flagging=False,
+    analytics_enabled=False,
+    share=True
+)
+
+iface.launch()