Update app.py

app.py

@@ -1,13 +1,3 @@
-"""
-Image Enhancement Lab — Gradio Interface
------------------------------------------
-Install dependencies:
-    pip install gradio pillow numpy
-
-Run:
-    python image_enhancer.py
-"""
-
 import gradio as gr
 import numpy as np
 import tempfile
@@ -17,73 +7,76 @@ 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()
+
     if upscale != 1.0:
         w, h = enhanced.size
         enhanced = enhanced.resize((int(w * upscale), int(h * upscale)), Image.LANCZOS)
+
     enhanced = ImageEnhance.Contrast(enhanced).enhance(contrast)
     enhanced = ImageEnhance.Brightness(enhanced).enhance(brightness)
     enhanced = ImageEnhance.Color(enhanced).enhance(saturation)
     enhanced = ImageEnhance.Sharpness(enhanced).enhance(sharpness)
+
     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)
+    _, 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
 
 
-def build_app():
-    with gr.Blocks(title="Image Enhancement Lab") as demo:
-        gr.Markdown("# Image Enhancement Lab")
-        gr.Markdown("Upload an image and adjust sliders — preview updates in real time.")
-
-        with gr.Row():
-            with gr.Column(scale=1, min_width=280):
-                image_input = gr.Image(label="Upload Image", type="numpy", sources=["upload", "webcam", "clipboard"])
-                contrast_slider   = gr.Slider(0.1, 3.0,  value=1.0, step=0.05, label="Contrast  (1.0 = original)")
-                brightness_slider = gr.Slider(0.1, 3.0,  value=1.0, step=0.05, label="Brightness  (1.0 = original)")
-                saturation_slider = gr.Slider(0.0, 3.0,  value=1.0, step=0.05, label="Saturation  (0 = grayscale)")
-                sharpness_slider  = gr.Slider(0.0, 5.0,  value=1.0, step=0.1,  label="Sharpness  (0 = blurry)")
-                upscale_slider    = gr.Slider(0.25, 4.0, value=1.0, step=0.25, label="Upscale Factor")
-                blur_slider       = gr.Slider(0.0, 10.0, value=0.0, step=0.25, label="Gaussian Blur Radius  (0 = off)")
-
-                with gr.Row():
-                    reset_btn = gr.Button("Reset",  variant="secondary")
-                    apply_btn = gr.Button("Apply",  variant="primary")
-
-                save_btn      = gr.Button("Download Enhanced Image", variant="secondary")
-                download_file = gr.File(label="Enhanced image", visible=False)
-
-            with gr.Column(scale=2):
-                reference_output = gr.Image(label="Reference (original)", type="numpy", interactive=False)
-                enhanced_output  = gr.Image(label="Enhanced Preview",     type="numpy", interactive=False)
+with gr.Blocks(title="Image Enhancement Lab") as demo:
+    gr.Markdown("# Image Enhancement Lab")
+    gr.Markdown("Upload an image and adjust sliders.")
 
-        all_inputs  = [image_input, contrast_slider, sharpness_slider, brightness_slider, saturation_slider, upscale_slider, blur_slider]
-        all_outputs = [reference_output, enhanced_output]
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="numpy")
 
-        for ctrl in [contrast_slider, sharpness_slider, brightness_slider, saturation_slider, upscale_slider, blur_slider, image_input]:
-            ctrl.change(fn=apply_enhancements, inputs=all_inputs, outputs=all_outputs)
+            contrast = gr.Slider(0.1, 3.0, 1.0)
+            brightness = gr.Slider(0.1, 3.0, 1.0)
+            saturation = gr.Slider(0.0, 3.0, 1.0)
+            sharpness = gr.Slider(0.0, 5.0, 1.0)
+            upscale = gr.Slider(0.25, 4.0, 1.0)
+            blur = gr.Slider(0.0, 10.0, 0.0)
 
-        apply_btn.click(fn=apply_enhancements, inputs=all_inputs, outputs=all_outputs)
+            apply_btn = gr.Button("Apply")
+            save_btn = gr.Button("Download")
 
-        def reset_all():
-            return None, 1.0, 1.0, 1.0, 1.0, 1.0, 0.0, None, None
+            download = gr.File(visible=True)
 
-        reset_btn.click(fn=reset_all, outputs=[image_input, contrast_slider, sharpness_slider, brightness_slider, saturation_slider, upscale_slider, blur_slider, reference_output, enhanced_output])
+        with gr.Column():
+            original_out = gr.Image()
+            enhanced_out = gr.Image()
 
-        save_btn.click(fn=save_enhanced, inputs=all_inputs, outputs=download_file)
+    inputs = [image_input, contrast, sharpness, brightness, saturation, upscale, blur]
 
-    return demo
+    apply_btn.click(apply_enhancements, inputs, [original_out, enhanced_out])
+    save_btn.click(save_enhanced, inputs, download)
 
 
-if __name__ == "__main__":
-    app = build_app()
-    app.launch(share=True, server_name="127.0.0.1", server_port=7860)
+# 🚀 REQUIRED for Spaces
+demo.queue()
+demo.launch()