upload code and palettes

main.py

@@ -0,0 +1,91 @@
+import streamlit as st
+from PIL import Image, ImageEnhance
+from io import BytesIO
+from pathlib import Path
+
+def load_palette_file(path: Path):
+    """Load a .hex palette file into a list of RGB tuples."""
+    colors = []
+    with open(path, "r") as f:
+        for line in f:
+            hexcode = line.strip().lstrip("#")
+            if len(hexcode) == 6:  # valid hex
+                r = int(hexcode[0:2], 16)
+                g = int(hexcode[2:4], 16)
+                b = int(hexcode[4:6], 16)
+                colors.append((r, g, b))
+            else:
+                st.warning(f"Invalid hex code in palette file: {line}")
+    return colors
+
+def apply_custom_palette(image: Image.Image, colors, dithering: bool):
+    """Quantize an image to a custom palette with optional dithering."""
+    # Build palette (zero pad to 256 colors if needed)
+    palette_img = Image.new("P", (1, 1))
+    palette_data = []
+    for r, g, b in colors[:256]:
+        palette_data.extend([r, g, b])
+    palette_data.extend([0] * (768 - len(palette_data)))
+    palette_img.putpalette(palette_data)
+
+    # Quantize with the custom palette
+    dither_mode = Image.FLOYDSTEINBERG if dithering else Image.NONE
+    image = image.quantize(palette=palette_img, dither=dither_mode, method=Image.MEDIANCUT)
+    return image.convert("RGB")
+
+st.set_page_config(page_title="Pixelizer", layout="wide")
+st.sidebar.title("Pixelizer")
+st.sidebar.header("Image Controls")
+uploaded_file = st.sidebar.file_uploader("Upload image", type=["png", "jpg", "jpeg"])
+show_original = st.sidebar.toggle("Show original", value=False)
+dithering = st.sidebar.toggle("Enable dithering", value=True)
+pixel_size = st.sidebar.number_input("Pixel size", min_value=1, max_value=100, value=4, step=1)
+palette_dir = Path("palettes")
+palette_files = sorted(palette_dir.glob("*.hex"))
+palette_names = [f.stem for f in palette_files]
+palette = st.sidebar.selectbox("Palette", palette_names) 
+
+contrast = st.sidebar.slider("Contrast", 0.5, 2.0, 1.0, 0.1)
+brightness = st.sidebar.slider("Brightness", 0.5, 2.0, 1.0, 0.1)
+
+if uploaded_file:
+    original_image = Image.open(uploaded_file).convert("RGB")
+
+    # Contrast and brightness adjustment 
+    image = original_image.copy()
+    enhancer_contrast = ImageEnhance.Contrast(image)
+    image = enhancer_contrast.enhance(contrast)
+    enhancer_brightness = ImageEnhance.Brightness(image)
+    image = enhancer_brightness.enhance(brightness)
+
+    # Downscale for pixelization
+    processed_image = image.copy()
+    w, h = processed_image.size
+    processed_image = processed_image.resize(
+        (max(1, w // pixel_size), max(1, h // pixel_size)),
+        resample=Image.NEAREST
+    )
+
+    # Apply palette
+    palette_file = palette_dir / f"{palette}.hex"
+    processed_image = apply_custom_palette(processed_image, load_palette_file(palette_file), dithering) 
+    upscaled_processed_image = processed_image.resize((w, h), Image.NEAREST)
+
+    # Display
+    if show_original:
+        st.image(original_image, caption="Original", width="content")
+    else:
+        st.image(upscaled_processed_image, caption="Processed", width="content")
+
+    # Save 
+    buf = BytesIO()
+    processed_image.save(buf, format="PNG")
+    byte_im = buf.getvalue()
+
+    st.sidebar.download_button(
+        "Save image",
+        data=byte_im,
+        file_name="pixelized.png",
+        mime="image/png"
+    )
+

palettes/antiquity16.hex

@@ -0,0 +1,16 @@
+202020
+2d211e
+452923
+6d3d29
+b16b4a
+e89f6e
+e8be82
+5d7557
+8e9257
+707b88
+8aa7ac
+e55d4d
+f1866c
+d26730
+de9a28
+e8d8a5

palettes/blk-nx64.hex

@@ -0,0 +1,64 @@
+000000
+12173d
+293268
+464b8c
+6b74b2
+909edd
+c1d9f2
+ffffff
+a293c4
+7b6aa5
+53427f
+3c2c68
+431e66
+5d2f8c
+854cbf
+b483ef
+8cff9b
+42bc7f
+22896e
+14665b
+0f4a4c
+0a2a33
+1d1a59
+322d89
+354ab2
+3e83d1
+50b9eb
+8cdaff
+53a1ad
+3b768f
+21526b
+163755
+008782
+00aaa5
+27d3cb
+78fae6
+cdc599
+988f64
+5c5d41
+353f23
+919b45
+afd370
+ffe091
+ffaa6e
+ff695a
+b23c40
+ff6675
+dd3745
+a52639
+721c2f
+b22e69
+e54286
+ff6eaf
+ffa5d5
+ffd3ad
+cc817a
+895654
+61393b
+3f1f3c
+723352
+994c69
+c37289
+f29faa
+ffccd0

palettes/cc-29.hex

@@ -0,0 +1,29 @@
+f2f0e5
+b8b5b9
+868188
+646365
+45444f
+3a3858
+212123
+352b42
+43436a
+4b80ca
+68c2d3
+a2dcc7
+ede19e
+d3a068
+b45252
+6a536e
+4b4158
+80493a
+a77b5b
+e5ceb4
+c2d368
+8ab060
+567b79
+4e584a
+7b7243
+b2b47e
+edc8c4
+cf8acb
+5f556a

palettes/crystal-flames.hex

@@ -0,0 +1,24 @@
+913a46
+b36067
+ffaba1
+ffbaa1
+db7684
+eb91a2
+f7c3a3
+ffb259
+f0c060
+ffc773
+ff9354
+fa7550
+dea55b
+f7b068
+e8534d
+c93b36
+cc7349
+b8553d
+d92b25
+ad1025
+a64116
+822333
+99222c
+82012a

palettes/dustbyte.hex

@@ -0,0 +1,4 @@
+788374
+f5e9bf
+aa644d
+372a39

palettes/midnight-ablaze.hex

@@ -0,0 +1,7 @@
+ff8274
+d53c6a
+7c183c
+460e2b
+31051e
+1f0510
+130208

palettes/oil-6.hex

@@ -0,0 +1,6 @@
+fbf5ef
+f2d3ab
+c69fa5
+8b6d9c
+494d7e
+272744

palettes/retro-115.hex

@@ -0,0 +1,7 @@
+040026
+004bc0
+0097ec
+00f3fc
+ffbcff
+d569f6
+6c1fd3

palettes/snail-village.hex

@@ -0,0 +1,6 @@
+323c39
+524b24
+7d5563
+917a46
+a3ad54
+99e550