backup

app.py

@@ -78,15 +78,7 @@ def create_color_palette_image(colors):
         ax.set_xticks([])
         ax.set_yticks([])
 
-        # Save to bytes buffer
-        buf = io.BytesIO()
-        plt.savefig(buf, format="png", bbox_inches="tight", dpi=100)
-        plt.close("all")  # Close all figures to prevent memory leaks
-        plt.close(fig)  # Explicitly close the current figure
-
-        # Convert to base64 string
-        buf.seek(0)
-        return buf
+        return fig  # Return the matplotlib figure directly
     except Exception as e:
         print(f"Error creating color palette: {e}")
         return None
@@ -209,7 +201,7 @@ with gr.Blocks() as demo:
                     label="Lighting Details",
                     lines=2,
                 )
-                color_palette_output = gr.Image(
+                color_palette_output = gr.Plot(
                     label="Color Palette",
                     height=100,
                 )