Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from PIL import Image
-import numpy as np
 import random
 from io import BytesIO
 
@@ -63,7 +62,11 @@ def generate_palette_image(colors):
         img = Image.new('RGB', (color_width, img_height), color)
         palette_img.paste(img, (i * color_width, 0))
     
-    return palette_img  # Return PIL image directly
+    # Save the image to a file in memory
+    buf = BytesIO()
+    palette_img.save(buf, format='PNG')
+    buf.seek(0)
+    return buf
 
 # Function to generate the CSS code for the color palette
 def generate_css_code(colors):
@@ -99,10 +102,7 @@ def generate_palette(image_path):
 def gradio_interface(image_path):
     palette_html, palette_img, css_code = generate_palette(image_path)
 
-    # Convert PIL image to NumPy array (Gradio expects a NumPy array or filepath)
-    palette_img_np = np.array(palette_img)
-
-    return palette_html, palette_img_np, css_code
+    return palette_html, palette_img, css_code
 
 # Create the Gradio interface
 with gr.Blocks() as interface:
@@ -124,5 +124,5 @@ with gr.Blocks() as interface:
     clear_btn.click(lambda: [None, None, None, None], inputs=[], outputs=[
         image_input, palette_output, palette_image_output, css_code_output])
 
-# Launch the interface (without `share=True`)
+# Launch the interface
 interface.launch()