Spaces:
Sleeping
Sleeping
| # app.py | |
| import streamlit as st | |
| from PIL import Image, ImageOps, ImageFilter, ImageEnhance | |
| import io | |
| import os | |
| st.set_page_config(page_title="Image Filter App", layout="wide") | |
| st.title("🖼️ Fun Image Filter App") | |
| # 3-column layout | |
| col1, col2, col3 = st.columns([1, 1, 1]) # equal width | |
| with col1: | |
| st.header("📤 Upload Image") | |
| st.caption("Max 5MB • Only PNG, JPG") | |
| uploaded_file = st.file_uploader("Choose an image", type=["png", "jpg", "jpeg"]) | |
| image = None | |
| filtered_image = None | |
| filter_option = "None" | |
| if uploaded_file: | |
| # Size check | |
| uploaded_file.seek(0, io.SEEK_END) | |
| file_size = uploaded_file.tell() | |
| uploaded_file.seek(0) | |
| if file_size > 5 * 1024 * 1024: | |
| with col1: | |
| st.error("❌ File too large! Must be under 5MB.") | |
| else: | |
| image = Image.open(uploaded_file) | |
| with col2: | |
| st.header("🎛️ Select Filter") | |
| filter_option = st.selectbox("Choose a filter:", [ | |
| "None", "Grayscale", "Invert Colors", "Sepia", "Blur", "Edge Detection", | |
| "Posterize", "Emboss", "Solarize", "Brightness Boost", "Contrast Boost", | |
| "Flip Horizontal", "Flip Vertical", "Rotate 90°" | |
| ]) | |
| with col3: | |
| st.header("🔍 Preview Output") | |
| if filter_option == "Grayscale": | |
| filtered_image = ImageOps.grayscale(image) | |
| st.image(filtered_image, caption="Grayscale", use_container_width=True) | |
| elif filter_option == "Invert Colors": | |
| if image.mode != "RGB": | |
| image = image.convert("RGB") | |
| filtered_image = ImageOps.invert(image) | |
| st.image(filtered_image, caption="Inverted", use_container_width=True) | |
| elif filter_option == "Sepia": | |
| # Sepia filter | |
| width, height = image.size | |
| pixels = image.load() # create the pixel map | |
| for py in range(height): | |
| for px in range(width): | |
| r, g, b = image.getpixel((px, py)) | |
| tr = int(0.393 * r + 0.769 * g + 0.189 * b) | |
| tg = int(0.349 * r + 0.686 * g + 0.168 * b) | |
| tb = int(0.272 * r + 0.534 * g + 0.131 * b) | |
| # Apply filter | |
| if tr > 255: | |
| tr = 255 | |
| if tg > 255: | |
| tg = 255 | |
| if tb > 255: | |
| tb = 255 | |
| pixels[px, py] = (tr, tg, tb) | |
| filtered_image = image | |
| st.image(filtered_image, caption="Sepia", use_container_width=True) | |
| elif filter_option == "Blur": | |
| filtered_image = image.filter(ImageFilter.GaussianBlur(radius=5)) | |
| st.image(filtered_image, caption="Blurred", use_container_width=True) | |
| elif filter_option == "Edge Detection": | |
| filtered_image = image.filter(ImageFilter.FIND_EDGES) | |
| st.image(filtered_image, caption="Edge Detection", use_container_width=True) | |
| elif filter_option == "Posterize": | |
| filtered_image = ImageOps.posterize(image, 4) | |
| st.image(filtered_image, caption="Posterized", use_container_width=True) | |
| elif filter_option == "Emboss": | |
| filtered_image = image.filter(ImageFilter.EMBOSS) | |
| st.image(filtered_image, caption="Emboss", use_container_width=True) | |
| elif filter_option == "Solarize": | |
| filtered_image = ImageOps.solarize(image, threshold=128) | |
| st.image(filtered_image, caption="Solarized", use_container_width=True) | |
| elif filter_option == "Brightness Boost": | |
| enhancer = ImageEnhance.Brightness(image) | |
| filtered_image = enhancer.enhance(2) # increase brightness | |
| st.image(filtered_image, caption="Brightness Boosted", use_container_width=True) | |
| elif filter_option == "Contrast Boost": | |
| enhancer = ImageEnhance.Contrast(image) | |
| filtered_image = enhancer.enhance(2) # increase contrast | |
| st.image(filtered_image, caption="Contrast Boosted", use_container_width=True) | |
| elif filter_option == "Flip Horizontal": | |
| filtered_image = image.transpose(Image.FLIP_LEFT_RIGHT) | |
| st.image(filtered_image, caption="Flipped Horizontal", use_container_width=True) | |
| elif filter_option == "Flip Vertical": | |
| filtered_image = image.transpose(Image.FLIP_TOP_BOTTOM) | |
| st.image(filtered_image, caption="Flipped Vertical", use_container_width=True) | |
| elif filter_option == "Rotate 90°": | |
| filtered_image = image.rotate(90) | |
| st.image(filtered_image, caption="Rotated 90°", use_container_width=True) | |
| else: | |
| st.info("Select a filter to see output.") | |
| with col1: | |
| if filtered_image: | |
| # Prepend "DEEZ_" to the original file name | |
| file_name = uploaded_file.name | |
| file_name_with_prefix = "DEEZ_" + file_name | |
| # Allow downloading the filtered image with the modified name | |
| st.download_button( | |
| label="Download Filtered Image", | |
| data=filtered_image.tobytes(), | |
| file_name=file_name_with_prefix, | |
| mime="image/png" | |
| ) | |