Spaces:
Running
Running
| """ | |
| Image Enhancement Lab — Gradio (Spaces-ready, labeled sliders) | |
| """ | |
| import gradio as gr | |
| import numpy as np | |
| import tempfile | |
| from PIL import Image, ImageEnhance, ImageFilter | |
| def apply_enhancements(image, contrast, sharpness, brightness, saturation, upscale, blur_radius): | |
| if image is None: | |
| return None, None | |
| original = Image.fromarray(image).convert("RGB") | |
| # Prevent large image crashes | |
| max_size = 1024 | |
| if max(original.size) > max_size: | |
| original.thumbnail((max_size, max_size)) | |
| enhanced = original.copy() | |
| # Upscale | |
| if upscale != 1.0: | |
| w, h = enhanced.size | |
| enhanced = enhanced.resize((int(w * upscale), int(h * upscale)), Image.LANCZOS) | |
| # Enhancements | |
| enhanced = ImageEnhance.Contrast(enhanced).enhance(contrast) | |
| enhanced = ImageEnhance.Brightness(enhanced).enhance(brightness) | |
| enhanced = ImageEnhance.Color(enhanced).enhance(saturation) | |
| enhanced = ImageEnhance.Sharpness(enhanced).enhance(sharpness) | |
| # Blur | |
| if blur_radius > 0: | |
| enhanced = enhanced.filter(ImageFilter.GaussianBlur(radius=blur_radius)) | |
| return np.array(original), np.array(enhanced) | |
| def save_enhanced(image, contrast, sharpness, brightness, saturation, upscale, blur_radius): | |
| _, enh_np = apply_enhancements( | |
| image, contrast, sharpness, brightness, saturation, upscale, blur_radius | |
| ) | |
| if enh_np is None: | |
| return None | |
| tmp = tempfile.NamedTemporaryFile(suffix=".png", delete=False) | |
| tmp.close() | |
| Image.fromarray(enh_np).save(tmp.name) | |
| return tmp.name | |
| with gr.Blocks(title="Image Enhancement Lab") as demo: | |
| gr.Markdown("# 🖼️ Image Enhancement Lab") | |
| gr.Markdown("Upload an image and adjust sliders to enhance it.") | |
| with gr.Row(): | |
| with gr.Column(scale=1): | |
| image_input = gr.Image(label="Upload Image", type="numpy") | |
| gr.Markdown("### 🎨 Color Adjustments") | |
| contrast = gr.Slider(0.1, 3.0, 1.0, label="Contrast (1.0 = original)") | |
| brightness = gr.Slider(0.1, 3.0, 1.0, label="Brightness (1.0 = original)") | |
| saturation = gr.Slider(0.0, 3.0, 1.0, label="Saturation (0 = grayscale)") | |
| gr.Markdown("### 🔍 Detail Controls") | |
| sharpness = gr.Slider(0.0, 5.0, 1.0, label="Sharpness (0 = blurry)") | |
| blur = gr.Slider(0.0, 10.0, 0.0, label="Gaussian Blur Radius") | |
| gr.Markdown("### 📐 Size Control") | |
| upscale = gr.Slider(0.25, 4.0, 1.0, label="Upscale Factor") | |
| apply_btn = gr.Button("Apply Enhancements") | |
| save_btn = gr.Button("Download Enhanced Image") | |
| download = gr.File(label="Download File", visible=True) | |
| with gr.Column(scale=2): | |
| original_out = gr.Image(label="Original Image") | |
| enhanced_out = gr.Image(label="Enhanced Image") | |
| inputs = [image_input, contrast, sharpness, brightness, saturation, upscale, blur] | |
| apply_btn.click(apply_enhancements, inputs, [original_out, enhanced_out]) | |
| save_btn.click(save_enhanced, inputs, download) | |
| # Required for Hugging Face Spaces | |
| demo.queue() | |
| demo.launch() |