Update app.py

app.py

@@ -1,388 +1,324 @@
-import gradio as gr
-from PIL import Image, ImageEnhance, ImageFilter, ImageOps
-import numpy as np
-import random
-import cv2
-import os
-
-def apply_transformations(
-        image,
-        apply_mosaic_trigger,
-        crop_box,
-        scale_factor,
-        rotation_angle,
-        h_flip,
-        v_flip,
-        shear_x,
-        shear_y,
-        brightness,
-        contrast,
-        saturation,
-        hue,
-        gamma,
-        grayscale,
-        invert,
-        blur_radius,
-        sharpen_factor,
-        noise_intensity,
-        cutout_n_holes,
-        cutout_ratio
-):
-    """
-    Applies a series of transformations to an input image based on user-defined parameters.
-
-    Args:
-        image (np.array): The input image in NumPy array format from Gradio.
-        apply_mosaic_trigger (bool): A flag to trigger the mosaic augmentation.
-        crop_box (tuple): Coordinates (x1, y1, x2, y2) for cropping.
-        scale_factor (float): Factor by which to scale the image.
-        rotation_angle (float): Angle in degrees for rotation.
-        h_flip (bool): Whether to flip horizontally.
-        v_flip (bool): Whether to flip vertically.
-        shear_x (float): Horizontal shear factor.
-        shear_y (float): Vertical shear factor.
-        brightness (float): Brightness enhancement factor (1.0 is original).
-        contrast (float): Contrast enhancement factor (1.0 is original).
-        saturation (float): Color saturation factor (1.0 is original).
-        hue (int): Hue shift value (-90 to 90).
-        gamma (float): Gamma correction value for exposure (1.0 is original).
-        grayscale (bool): Whether to convert to grayscale.
-        invert (bool): Whether to invert image colors.
-        blur_radius (float): Radius for Gaussian blur.
-        sharpen_factor (int): Number of times to apply the sharpen filter.
-        noise_intensity (float): Standard deviation for Gaussian noise.
-        cutout_n_holes (int): Number of rectangular holes to create.
-        cutout_ratio (float): Size of the cutout holes relative to image size.
-
-    Returns:
-        tuple: A tuple containing the original image preview, the transformed image,
-               and a reset boolean for the mosaic trigger.
-    """
-    if image is None:
-        return None, None, False
-
-    img = Image.fromarray(image).convert("RGB")
-    original_for_preview = img.copy()
-
-    # Enhanced Augmentation: Mosaic
-    if apply_mosaic_trigger:
-        w, h = img.size
-        cx, cy = w // 2, h // 2
-        crops = [img.crop((0, 0, cx, cy)), img.crop((cx, 0, w, cy)), img.crop((0, cy, w, h)), img.crop((cx, cy, w, h))]
-        random.shuffle(crops)
-        mosaic_img = Image.new('RGB', (w, h))
-        mosaic_img.paste(crops[0].resize((cx, cy), Image.Resampling.LANCZOS), (0, 0))
-        mosaic_img.paste(crops[1].resize((cx, cy), Image.Resampling.LANCZOS), (cx, 0))
-        mosaic_img.paste(crops[2].resize((cx, cy), Image.Resampling.LANCZOS), (0, cy))
-        mosaic_img.paste(crops[3].resize((cx, cy), Image.Resampling.LANCZOS), (cx, cy))
-        img = mosaic_img
-
-    # Geometric Augmentation: Crop
-    if crop_box is not None:
-        try:
-            x1, y1, x2, y2 = map(int, crop_box)
-            if x1 < x2 and y1 < y2:
-                img = img.crop((x1, y1, x2, y2))
-        except (ValueError, TypeError):
-            pass
-
-    # Geometric Augmentation: Scale
-    if scale_factor != 1.0:
-        w, h = img.size
-        new_size = (int(w * scale_factor), int(h * scale_factor))
-        img = img.resize(new_size, Image.Resampling.LANCZOS)
-
-    # Geometric Augmentation: Shear
-    if shear_x != 0 or shear_y != 0:
-        img = img.transform(img.size, Image.Transform.AFFINE, (1, shear_x, 0, shear_y, 1, 0), Image.Resampling.BICUBIC)
-
-    # Geometric Augmentation: Rotate
-    if rotation_angle != 0:
-        img = img.rotate(rotation_angle, expand=True, fillcolor=(128, 128, 128))
-
-    # Geometric Augmentation: Flip
-    if h_flip:
-        img = ImageOps.mirror(img)
-    if v_flip:
-        img = ImageOps.flip(img)
-
-    # Color Augmentation: Brightness, Contrast, Saturation
-    if brightness != 1.0: img = ImageEnhance.Brightness(img).enhance(brightness)
-    if contrast != 1.0: img = ImageEnhance.Contrast(img).enhance(contrast)
-    if saturation != 1.0: img = ImageEnhance.Color(img).enhance(saturation)
-
-    # Color Augmentation: Hue (using OpenCV for robust HSV conversion)
-    if hue != 0:
-        np_img_rgb = np.array(img)
-        hsv_img = cv2.cvtColor(np_img_rgb, cv2.COLOR_RGB2HSV)
-        hsv_img[:, :, 0] = (hsv_img[:, :, 0].astype(int) + hue) % 180
-        img = Image.fromarray(cv2.cvtColor(hsv_img, cv2.COLOR_HSV2RGB))
-
-    # Color Augmentation: Exposure (Gamma Correction)
-    if gamma != 1.0:
-        inv_gamma = 1.0 / gamma
-        table = np.array([((i / 255.0) ** inv_gamma) * 255 for i in np.arange(0, 256)]).astype("uint8")
-        if img.mode == 'L':
-            img = img.point(table)
-        else:
-            r, g, b = img.split()
-            r, g, b = r.point(table), g.point(table), b.point(table)
-            img = Image.merge('RGB', (r, g, b))
-
-    # Filter Augmentation: Sharpen
-    if sharpen_factor > 0:
-        for _ in range(int(sharpen_factor)):
-            img = img.filter(ImageFilter.SHARPEN)
-
-    # Filter Augmentation: Blur
-    if blur_radius > 0:
-        img = img.filter(ImageFilter.GaussianBlur(radius=blur_radius))
-
-    # Enhanced Augmentation: Cutout
-    if cutout_n_holes > 0 and cutout_ratio > 0:
-        w, h = img.size
-        np_img = np.array(img)
-        mask_value = int(np.mean(np_img))
-        hole_w, hole_h = int(w * cutout_ratio), int(h * cutout_ratio)
-        if hole_w > 0 and hole_h > 0:
-            for _ in range(cutout_n_holes):
-                y1, x1 = random.randint(0, h - hole_h), random.randint(0, w - hole_w)
-                y2, x2 = y1 + hole_h, x1 + hole_w
-                np_img[y1:y2, x1:x2] = mask_value
-            img = Image.fromarray(np_img)
-
-    # Filter Augmentation: Noise
-    if noise_intensity > 0:
-        np_img = np.array(img)
-        noise = np.random.normal(0, noise_intensity, np_img.shape)
-        np_img = np.clip(np_img + noise, 0, 255).astype(np.uint8)
-        img = Image.fromarray(np_img)
-
-    # Color Augmentation: Grayscale and Invert
-    if grayscale:
-        img = ImageOps.grayscale(img)
-    if invert:
-        img = ImageOps.invert(img.convert('RGB') if img.mode != 'L' else img)
-
-    # Return final images and reset the mosaic trigger to False
-    return original_for_preview, img, False
-
-
-# --- UI Helper Functions ---
-
-def process_selection(evt: gr.SelectData):
-    """Callback function to capture the coordinates of a user's selection on the input image for cropping."""
-    return (evt.index[0], evt.index[1], evt.index[2], evt.index[3])
-
-
-def update_slider(min_val, max_val, current_val):
-    """
-    Updates a Gradio slider's properties (min, max, value) dynamically.
-    It ensures the current value is clamped within the new min/max bounds.
-    """
-    if min_val > max_val:
-        min_val = max_val
-    new_val = max(min_val, min(max_val, current_val))
-    return gr.update(minimum=min_val, maximum=max_val, value=new_val)
-
-
-def reset_all_controls():
-    """Returns a tuple of default values to reset all UI components to their initial state."""
-    return (
-        # Main controls
-        False, None, 1.0, 0, False, False, 0.0, 0.0, 1.0, 1.0, 1.0, 0, 1.0, False, False, 0.0, 0, 0, 0, 0.0,
-        # Range controls
-        0.1, 3.0, -180, 180, -0.5, 0.5, -0.5, 0.5, 0.0, 3.0, 0.0, 3.0, 0.0, 3.0,
-        -90, 90, 0.2, 2.2, 0.0, 15.0, 0, 50, 0, 50, 0.0, 0.5,
-        # Image outputs
-        None, None
-    )
-
-
-def on_upload_or_clear(image):
-    """Handles the event of uploading or clearing an image, resetting all controls."""
-    outputs = reset_all_controls()
-    if image is None: return outputs
-    # Replace the last two None values with the new image and a cleared output
-    return outputs[:-2] + (image, None)
-
-
-# --- Gradio UI Layout ---
-
-with gr.Blocks(theme=gr.themes.Soft(), css="footer {display: none !important}") as demo:
-    gr.Markdown("# Advanced Image Augmentation Tool")
-    gr.Markdown(
-        "Upload an image or select an example. Apply transformations from the control panel and even define your own parameter ranges.")
-
-    # State objects to hold data that persists between interactions without being a visible component
-    crop_box_state = gr.State(None)
-    mosaic_trigger = gr.State(False)
-
-    with gr.Row(variant="panel"):
-        # --- Left Column: All Control Panels ---
-        with gr.Column(scale=1, min_width=400):
-            gr.Markdown("### Main Control Panel")
-
-            with gr.Accordion("Geometric Transformations", open=True):
-                scale_slider = gr.Slider(minimum=0.1, maximum=3.0, value=1.0, step=0.05, label="Scale")
-                rotation_slider = gr.Slider(minimum=-180, maximum=180, value=0, step=1, label="Rotation Angle")
-                with gr.Row():
-                    rotate_90_btn = gr.Button("Rotate 90°")
-                    rotate_180_btn = gr.Button("Rotate 180°")
-                    rotate_270_btn = gr.Button("Rotate 270°")
-                shear_x_slider = gr.Slider(minimum=-0.5, maximum=0.5, value=0.0, step=0.01, label="Shear X")
-                shear_y_slider = gr.Slider(minimum=-0.5, maximum=0.5, value=0.0, step=0.01, label="Shear Y")
-                with gr.Row():
-                    h_flip_check = gr.Checkbox(label="Horizontal Flip")
-                    v_flip_check = gr.Checkbox(label="Vertical Flip")
-
-            with gr.Accordion("Color & Tone Adjustments", open=True):
-                brightness_slider = gr.Slider(minimum=0.0, maximum=3.0, value=1.0, step=0.05, label="Brightness")
-                contrast_slider = gr.Slider(minimum=0.0, maximum=3.0, value=1.0, step=0.05, label="Contrast")
-                saturation_slider = gr.Slider(minimum=0.0, maximum=3.0, value=1.0, step=0.05, label="Saturation")
-                hue_slider = gr.Slider(minimum=-90, maximum=90, value=0, step=1, label="Hue")
-                gamma_slider = gr.Slider(minimum=0.2, maximum=2.2, value=1.0, step=0.05, label="Exposure (Gamma)")
-                with gr.Row():
-                    grayscale_check = gr.Checkbox(label="Grayscale")
-                    invert_check = gr.Checkbox(label="Invert Colors")
-
-            with gr.Accordion("Filters & Distortions", open=True):
-                blur_slider = gr.Slider(minimum=0.0, maximum=15.0, value=0.0, step=0.1, label="Blur Radius")
-                sharpen_slider = gr.Slider(minimum=0, maximum=5, value=0, step=1, label="Sharpen Intensity")
-                noise_slider = gr.Slider(minimum=0, maximum=50, value=0, step=1, label="Add Noise")
-
-            with gr.Accordion("Enhanced Augmentations", open=False):
-                gr.Markdown("**Cutout**: Erases random rectangular patches from the image.")
-                cutout_n_slider = gr.Slider(minimum=0, maximum=50, value=0, step=1, label="Number of Holes")
-                cutout_ratio_slider = gr.Slider(minimum=0.0, maximum=0.5, value=0.0, step=0.01, label="Hole Size Ratio")
-                gr.Markdown("**Mosaic**: Splits the image into 4 quadrants and shuffles them.")
-                mosaic_btn = gr.Button("Apply Mosaic")
-
-            # --- User-Defined Range Control Panel ---
-            with gr.Accordion("Parameter Range Control (Advanced)", open=False):
-                gr.Markdown("Define the min/max for the sliders in the main panel.")
-                with gr.Tabs():
-                    with gr.TabItem("Geometric"):
-                        with gr.Row():
-                            scale_min = gr.Number(0.1, label="Scale Min")
-                            scale_max = gr.Number(3.0, label="Scale Max")
-                        with gr.Row():
-                            rotation_min = gr.Number(-180, label="Rotation Min")
-                            rotation_max = gr.Number(180, label="Rotation Max")
-                        with gr.Row():
-                            shear_x_min = gr.Number(-0.5, label="Shear X Min")
-                            shear_x_max = gr.Number(0.5, label="Shear X Max")
-                        with gr.Row():
-                            shear_y_min = gr.Number(-0.5, label="Shear Y Min")
-                            shear_y_max = gr.Number(0.5, label="Shear Y Max")
-                    with gr.TabItem("Color"):
-                        with gr.Row():
-                            brightness_min = gr.Number(0.0, label="Brightness Min")
-                            brightness_max = gr.Number(3.0, label="Brightness Max")
-                        with gr.Row():
-                            contrast_min = gr.Number(0.0, label="Contrast Min")
-                            contrast_max = gr.Number(3.0, label="Contrast Max")
-                        with gr.Row():
-                            saturation_min = gr.Number(0.0, label="Saturation Min")
-                            saturation_max = gr.Number(3.0, label="Saturation Max")
-                        with gr.Row():
-                            hue_min = gr.Number(-90, label="Hue Min")
-                            hue_max = gr.Number(90, label="Hue Max")
-                        with gr.Row():
-                            gamma_min = gr.Number(0.2, label="Exposure Min")
-                            gamma_max = gr.Number(2.2, label="Exposure Max")
-                    with gr.TabItem("Filters/Other"):
-                        with gr.Row():
-                            blur_min = gr.Number(0.0, label="Blur Min")
-                            blur_max = gr.Number(15.0, label="Blur Max")
-                        with gr.Row():
-                            noise_min = gr.Number(0, label="Noise Min")
-                            noise_max = gr.Number(50, label="Noise Max")
-                        with gr.Row():
-                            cutout_n_min = gr.Number(0, label="Holes Min")
-                            cutout_n_max = gr.Number(50, label="Holes Max")
-                        with gr.Row():
-                            cutout_ratio_min = gr.Number(0.0, label="Ratio Min")
-                            cutout_ratio_max = gr.Number(0.5, label="Ratio Max")
-
-            reset_btn = gr.Button("Reset All Settings", variant="stop", size="lg")
-
-        # --- Right Column: Image Display ---
-        with gr.Column(scale=3):
-            image_input = gr.Image(type="numpy", label="Original Image (Select an area to crop)", tool="select")
-            image_output = gr.Image(label="Transformed Image", interactive=False)
-            gr.Examples(
-                examples=[os.path.join(os.path.dirname(__file__), f) for f in ["cat.jpg", "cheetah.jpg", "lion.jpg"] if
-                          os.path.exists(os.path.join(os.path.dirname(__file__), f))],
-                inputs=image_input,
-                label="Example Images"
-            )
-
-    # --- Event Handling Logic ---
-
-    # Define all inputs for the main transformation function
-    all_inputs = [
-        image_input, mosaic_trigger, crop_box_state, scale_slider, rotation_slider, h_flip_check, v_flip_check,
-        shear_x_slider, shear_y_slider, brightness_slider, contrast_slider, saturation_slider,
-        hue_slider, gamma_slider, grayscale_check, invert_check, blur_slider, sharpen_slider, noise_slider,
-        cutout_n_slider, cutout_ratio_slider
-    ]
-
-    # This list defines all components that need to be reset
-    all_resettable_outputs = [
-        mosaic_trigger, crop_box_state, scale_slider, rotation_slider, h_flip_check, v_flip_check,
-        shear_x_slider, shear_y_slider, brightness_slider, contrast_slider, saturation_slider,
-        hue_slider, gamma_slider, grayscale_check, invert_check, blur_slider, sharpen_slider, noise_slider,
-        cutout_n_slider, cutout_ratio_slider,
-        scale_min, scale_max, rotation_min, rotation_max, shear_x_min, shear_x_max, shear_y_min, shear_y_max,
-        brightness_min, brightness_max, contrast_min, contrast_max, saturation_min, saturation_max,
-        hue_min, hue_max, gamma_min, gamma_max, blur_min, blur_max, noise_min, noise_max,
-        cutout_n_min, cutout_n_max, cutout_ratio_min, cutout_ratio_max,
-        image_input, image_output
-    ]
-
-    # Link every main control component to the transformation function
-    for component in all_inputs:
-        if component not in [image_input, crop_box_state, mosaic_trigger]:
-            component.change(fn=apply_transformations, inputs=all_inputs,
-                             outputs=[image_input, image_output, mosaic_trigger])
-
-    # Handle specific events like image selection, upload, and button clicks
-    image_input.select(fn=process_selection, inputs=None, outputs=[crop_box_state]).then(
-        fn=apply_transformations, inputs=all_inputs, outputs=[image_input, image_output, mosaic_trigger])
-
-    image_input.upload(fn=on_upload_or_clear, inputs=[image_input], outputs=all_resettable_outputs)
-    image_input.clear(fn=on_upload_or_clear, inputs=[image_input], outputs=all_resettable_outputs)
-    reset_btn.click(fn=on_upload_or_clear, inputs=[image_input], outputs=all_resettable_outputs)
-
-    # Handle 90-degree rotation buttons
-    rotate_90_btn.click(lambda x: (x + 90) % 360, inputs=[rotation_slider], outputs=[rotation_slider])
-    rotate_180_btn.click(lambda: 180, inputs=None, outputs=[rotation_slider])
-    rotate_270_btn.click(lambda: -90, inputs=None, outputs=[rotation_slider])
-
-    # Handle mosaic button click
-    mosaic_btn.click(lambda: True, None, mosaic_trigger).then(
-        fn=apply_transformations, inputs=all_inputs, outputs=[image_input, image_output, mosaic_trigger])
-
-    # Link the range control inputs to update the main sliders
-    range_map = {
-        (scale_min, scale_max): scale_slider,
-        (rotation_min, rotation_max): rotation_slider,
-        (shear_x_min, shear_x_max): shear_x_slider,
-        (shear_y_min, shear_y_max): shear_y_slider,
-        (brightness_min, brightness_max): brightness_slider,
-        (contrast_min, contrast_max): contrast_slider,
-        (saturation_min, saturation_max): saturation_slider,
-        (hue_min, hue_max): hue_slider,
-        (gamma_min, gamma_max): gamma_slider,
-        (blur_min, blur_max): blur_slider,
-        (noise_min, noise_max): noise_slider,
-        (cutout_n_min, cutout_n_max): cutout_n_slider,
-        (cutout_ratio_min, cutout_ratio_max): cutout_ratio_slider
-    }
-    for (min_comp, max_comp), slider in range_map.items():
-        min_comp.change(fn=update_slider, inputs=[min_comp, max_comp, slider], outputs=[slider])
-        max_comp.change(fn=update_slider, inputs=[min_comp, max_comp, slider], outputs=[slider])
-
-# Launch the Gradio interface
+#
+# Advanced Image Augmentation Tool for Gradio
+#
+# This script creates a comprehensive web interface for applying various image transformations.
+# It uses the modern gr.ImageEditor component for interactive cropping, ensuring
+# compatibility with recent versions of the Gradio library.
+# Users can upload an image, interactively adjust parameters for augmentations,
+# and dynamically define the operational range (min/max values) for each slider.
+#
+
+import gradio as gr
+from PIL import Image, ImageEnhance, ImageFilter, ImageOps
+import numpy as np
+import random
+import cv2
+import os
+
+# --- Core Transformation Logic ---
+
+def apply_transformations(
+    image,
+    apply_mosaic_trigger,
+    crop_box,
+    scale_factor,
+    rotation_angle,
+    h_flip,
+    v_flip,
+    shear_x,
+    shear_y,
+    brightness,
+    contrast,
+    saturation,
+    hue,
+    gamma,
+    grayscale,
+    invert,
+    blur_radius,
+    sharpen_factor,
+    noise_intensity,
+    cutout_n_holes,
+    cutout_ratio
+):
+    """
+    Applies a series of transformations to an input image based on user-defined parameters.
+
+    Args:
+        image (dict or np.array): The input from gr.ImageEditor, which can be a dict
+                                  containing 'background' or a direct np.array.
+        apply_mosaic_trigger (bool): A flag to trigger the mosaic augmentation.
+        crop_box (tuple): Coordinates (x1, y1, x2, y2) for cropping.
+        ... (all other args are the same)
+
+    Returns:
+        tuple: A tuple containing the original image preview, the transformed image,
+               and a reset boolean for the mosaic trigger.
+    """
+    # Handle the input from gr.ImageEditor, which is a dictionary
+    if isinstance(image, dict) and image.get("background") is not None:
+        image_data = image["background"]
+    elif isinstance(image, np.ndarray):
+        image_data = image
+    else:
+        return None, None, False
+
+    # Convert NumPy array to PIL Image
+    img = Image.fromarray(image_data).convert("RGB")
+    original_for_preview = img.copy()
+
+    # Enhanced Augmentation: Mosaic
+    if apply_mosaic_trigger:
+        w, h = img.size
+        cx, cy = w // 2, h // 2
+        crops = [img.crop((0, 0, cx, cy)), img.crop((cx, 0, w, cy)), img.crop((0, cy, w, h)), img.crop((cx, cy, w, h))]
+        random.shuffle(crops)
+        mosaic_img = Image.new('RGB', (w, h))
+        mosaic_img.paste(crops[0].resize((cx, cy), Image.Resampling.LANCZOS), (0, 0))
+        mosaic_img.paste(crops[1].resize((cx, cy), Image.Resampling.LANCZOS), (cx, 0))
+        mosaic_img.paste(crops[2].resize((cx, cy), Image.Resampling.LANCZOS), (0, cy))
+        mosaic_img.paste(crops[3].resize((cx, cy), Image.Resampling.LANCZOS), (cx, cy))
+        img = mosaic_img
+
+    # Geometric Augmentation: Crop
+    if crop_box is not None:
+        try:
+            x1, y1, x2, y2 = map(int, crop_box)
+            if x1 < x2 and y1 < y2:
+                img = img.crop((x1, y1, x2, y2))
+        except (ValueError, TypeError):
+            pass
+
+    # Geometric Augmentation: Scale
+    if scale_factor != 1.0:
+        w, h = img.size
+        new_size = (int(w * scale_factor), int(h * scale_factor))
+        img = img.resize(new_size, Image.Resampling.LANCZOS)
+    
+    # Geometric Augmentation: Shear
+    if shear_x != 0 or shear_y != 0:
+        img = img.transform(img.size, Image.Transform.AFFINE, (1, shear_x, 0, shear_y, 1, 0), Image.Resampling.BICUBIC)
+
+    # Geometric Augmentation: Rotate
+    if rotation_angle != 0:
+        img = img.rotate(rotation_angle, expand=True, fillcolor=(128, 128, 128))
+
+    # Geometric Augmentation: Flip
+    if h_flip: img = ImageOps.mirror(img)
+    if v_flip: img = ImageOps.flip(img)
+
+    # Color Augmentation: Brightness, Contrast, Saturation
+    if brightness != 1.0: img = ImageEnhance.Brightness(img).enhance(brightness)
+    if contrast != 1.0: img = ImageEnhance.Contrast(img).enhance(contrast)
+    if saturation != 1.0: img = ImageEnhance.Color(img).enhance(saturation)
+    
+    # Color Augmentation: Hue (using OpenCV for robust HSV conversion)
+    if hue != 0:
+        np_img_rgb = np.array(img)
+        hsv_img = cv2.cvtColor(np_img_rgb, cv2.COLOR_RGB2HSV)
+        hsv_img[:, :, 0] = (hsv_img[:, :, 0].astype(int) + hue) % 180
+        img = Image.fromarray(cv2.cvtColor(hsv_img, cv2.COLOR_HSV2RGB))
+    
+    # Color Augmentation: Exposure (Gamma Correction)
+    if gamma != 1.0:
+        inv_gamma = 1.0 / gamma
+        table = np.array([((i / 255.0) ** inv_gamma) * 255 for i in np.arange(0, 256)]).astype("uint8")
+        if img.mode == 'L':
+            img = img.point(table)
+        else:
+            r, g, b = img.split()
+            r, g, b = r.point(table), g.point(table), b.point(table)
+            img = Image.merge('RGB', (r, g, b))
+
+    # Filter Augmentation: Sharpen
+    if sharpen_factor > 0:
+        for _ in range(int(sharpen_factor)): img = img.filter(ImageFilter.SHARPEN)
+
+    # Filter Augmentation: Blur
+    if blur_radius > 0: img = img.filter(ImageFilter.GaussianBlur(radius=blur_radius))
+    
+    # Enhanced Augmentation: Cutout
+    if cutout_n_holes > 0 and cutout_ratio > 0:
+        w, h = img.size
+        np_img = np.array(img)
+        mask_value = int(np.mean(np_img))
+        hole_w, hole_h = int(w * cutout_ratio), int(h * cutout_ratio)
+        if hole_w > 0 and hole_h > 0:
+            for _ in range(cutout_n_holes):
+                y1, x1 = random.randint(0, h - hole_h), random.randint(0, w - hole_w)
+                y2, x2 = y1 + hole_h, x1 + hole_w
+                np_img[y1:y2, x1:x2] = mask_value
+            img = Image.fromarray(np_img)
+
+    # Filter Augmentation: Noise
+    if noise_intensity > 0:
+        np_img = np.array(img)
+        noise = np.random.normal(0, noise_intensity, np_img.shape)
+        np_img = np.clip(np_img + noise, 0, 255).astype(np.uint8)
+        img = Image.fromarray(np_img)
+        
+    # Color Augmentation: Grayscale and Invert
+    if grayscale: img = ImageOps.grayscale(img)
+    if invert: img = ImageOps.invert(img.convert('RGB') if img.mode != 'L' else img)
+
+    # Return final images and reset the mosaic trigger to False
+    return original_for_preview, img, False
+
+# --- UI Helper Functions ---
+
+def process_selection(evt: gr.SelectData):
+    """Callback function to capture the coordinates of a user's selection on the input image for cropping."""
+    return (evt.index[0], evt.index[1], evt.index[2], evt.index[3])
+
+def update_slider(min_val, max_val, current_val):
+    """Updates a Gradio slider's properties (min, max, value) dynamically."""
+    if min_val > max_val: min_val = max_val
+    new_val = max(min_val, min(max_val, current_val))
+    return gr.update(minimum=min_val, maximum=max_val, value=new_val)
+
+def reset_all_controls():
+    """Returns a tuple of default values to reset all UI components to their initial state."""
+    return (
+        False, None, 1.0, 0, False, False, 0.0, 0.0, 1.0, 1.0, 1.0, 0, 1.0, False, False, 0.0, 0, 0, 0, 0.0,
+        0.1, 3.0, -180, 180, -0.5, 0.5, -0.5, 0.5, 0.0, 3.0, 0.0, 3.0, 0.0, 3.0,
+        -90, 90, 0.2, 2.2, 0.0, 15.0, 0, 50, 0, 50, 0.0, 0.5,
+        None, None
+    )
+
+def on_upload_or_clear(image):
+    """Handles the event of uploading or clearing an image, resetting all controls."""
+    outputs = reset_all_controls()
+    if image is None: return outputs
+    return outputs[:-2] + (image, None)
+
+# --- Gradio UI Layout ---
+
+with gr.Blocks(theme=gr.themes.Soft(), css="footer {display: none !important}") as demo:
+    gr.Markdown("# Advanced Image Augmentation Tool")
+    gr.Markdown("Upload an image or select an example. Apply transformations from the control panel and even define your own parameter ranges.")
+    
+    crop_box_state = gr.State(None)
+    mosaic_trigger = gr.State(False)
+
+    with gr.Row(variant="panel"):
+        with gr.Column(scale=1, min_width=400):
+            gr.Markdown("### Main Control Panel")
+            
+            with gr.Accordion("Geometric Transformations", open=True):
+                scale_slider = gr.Slider(0.1, 3.0, 1.0, step=0.05, label="Scale")
+                rotation_slider = gr.Slider(-180, 180, 0, step=1, label="Rotation Angle")
+                with gr.Row():
+                    rotate_90_btn = gr.Button("Rotate 90°")
+                    rotate_180_btn = gr.Button("Rotate 180°")
+                    rotate_270_btn = gr.Button("Rotate 270°")
+                shear_x_slider = gr.Slider(-0.5, 0.5, 0.0, step=0.01, label="Shear X")
+                shear_y_slider = gr.Slider(-0.5, 0.5, 0.0, step=0.01, label="Shear Y")
+                with gr.Row():
+                    h_flip_check = gr.Checkbox(label="Horizontal Flip")
+                    v_flip_check = gr.Checkbox(label="Vertical Flip")
+            
+            with gr.Accordion("Color & Tone Adjustments", open=True):
+                brightness_slider = gr.Slider(0.0, 3.0, 1.0, step=0.05, label="Brightness")
+                contrast_slider = gr.Slider(0.0, 3.0, 1.0, step=0.05, label="Contrast")
+                saturation_slider = gr.Slider(0.0, 3.0, 1.0, step=0.05, label="Saturation")
+                hue_slider = gr.Slider(-90, 90, 0, step=1, label="Hue")
+                gamma_slider = gr.Slider(0.2, 2.2, 1.0, step=0.05, label="Exposure (Gamma)")
+                with gr.Row():
+                    grayscale_check = gr.Checkbox(label="Grayscale")
+                    invert_check = gr.Checkbox(label="Invert Colors")
+
+            with gr.Accordion("Filters & Distortions", open=True):
+                blur_slider = gr.Slider(0.0, 15.0, 0.0, step=0.1, label="Blur Radius")
+                sharpen_slider = gr.Slider(0, 5, 0, step=1, label="Sharpen Intensity")
+                noise_slider = gr.Slider(0, 50, 0, step=1, label="Add Noise")
+            
+            with gr.Accordion("Enhanced Augmentations", open=False):
+                gr.Markdown("**Cutout**: Erases random rectangular patches from the image.")
+                cutout_n_slider = gr.Slider(0, 50, 0, step=1, label="Number of Holes")
+                cutout_ratio_slider = gr.Slider(0.0, 0.5, 0.0, step=0.01, label="Hole Size Ratio")
+                gr.Markdown("**Mosaic**: Splits the image into 4 quadrants and shuffles them.")
+                mosaic_btn = gr.Button("Apply Mosaic")
+
+            with gr.Accordion("Parameter Range Control (Advanced)", open=False):
+                gr.Markdown("Define the min/max for the sliders in the main panel.")
+                with gr.Tabs():
+                    with gr.TabItem("Geometric"):
+                        scale_min, scale_max = gr.Number(0.1, label="Scale Min"), gr.Number(3.0, label="Scale Max")
+                        rotation_min, rotation_max = gr.Number(-180, label="Rotation Min"), gr.Number(180, label="Rotation Max")
+                        shear_x_min, shear_x_max = gr.Number(-0.5, label="Shear X Min"), gr.Number(0.5, label="Shear X Max")
+                        shear_y_min, shear_y_max = gr.Number(-0.5, label="Shear Y Min"), gr.Number(0.5, label="Shear Y Max")
+                    with gr.TabItem("Color"):
+                        brightness_min, brightness_max = gr.Number(0.0, label="Brightness Min"), gr.Number(3.0, label="Brightness Max")
+                        contrast_min, contrast_max = gr.Number(0.0, label="Contrast Min"), gr.Number(3.0, label="Contrast Max")
+                        saturation_min, saturation_max = gr.Number(0.0, label="Saturation Min"), gr.Number(3.0, label="Saturation Max")
+                        hue_min, hue_max = gr.Number(-90, label="Hue Min"), gr.Number(90, label="Hue Max")
+                        gamma_min, gamma_max = gr.Number(0.2, label="Exposure Min"), gr.Number(2.2, label="Exposure Max")
+                    with gr.TabItem("Filters/Other"):
+                        blur_min, blur_max = gr.Number(0.0, label="Blur Min"), gr.Number(15.0, label="Blur Max")
+                        noise_min, noise_max = gr.Number(0, label="Noise Min"), gr.Number(50, label="Noise Max")
+                        cutout_n_min, cutout_n_max = gr.Number(0, label="Holes Min"), gr.Number(50, label="Holes Max")
+                        cutout_ratio_min, cutout_ratio_max = gr.Number(0.0, label="Ratio Min"), gr.Number(0.5, label="Ratio Max")
+
+            reset_btn = gr.Button("Reset All Settings", variant="stop", size="lg")
+
+        with gr.Column(scale=3):
+            # FIXED: Replaced gr.Image with gr.ImageEditor
+            image_input = gr.ImageEditor(type="numpy", label="Original Image (Select an area to crop)")
+            image_output = gr.Image(label="Transformed Image", interactive=False)
+            gr.Examples(
+                examples=[os.path.join(os.path.dirname(__file__), f) for f in ["cat.jpg", "cheetah.jpg", "lion.jpg"] if os.path.exists(os.path.join(os.path.dirname(__file__), f))],
+                inputs=image_input,
+                label="Example Images"
+            )
+
+    all_inputs = [
+        image_input, mosaic_trigger, crop_box_state, scale_slider, rotation_slider, h_flip_check, v_flip_check,
+        shear_x_slider, shear_y_slider, brightness_slider, contrast_slider, saturation_slider, 
+        hue_slider, gamma_slider, grayscale_check, invert_check, blur_slider, sharpen_slider, noise_slider,
+        cutout_n_slider, cutout_ratio_slider
+    ]
+    
+    all_resettable_outputs = [
+        mosaic_trigger, crop_box_state, scale_slider, rotation_slider, h_flip_check, v_flip_check,
+        shear_x_slider, shear_y_slider, brightness_slider, contrast_slider, saturation_slider, 
+        hue_slider, gamma_slider, grayscale_check, invert_check, blur_slider, sharpen_slider, noise_slider,
+        cutout_n_slider, cutout_ratio_slider,
+        scale_min, scale_max, rotation_min, rotation_max, shear_x_min, shear_x_max, shear_y_min, shear_y_max,
+        brightness_min, brightness_max, contrast_min, contrast_max, saturation_min, saturation_max,
+        hue_min, hue_max, gamma_min, gamma_max, blur_min, blur_max, noise_min, noise_max,
+        cutout_n_min, cutout_n_max, cutout_ratio_min, cutout_ratio_max,
+        image_input, image_output
+    ]
+
+    for component in all_inputs:
+        if component not in [image_input, crop_box_state, mosaic_trigger]:
+            component.change(fn=apply_transformations, inputs=all_inputs, outputs=[image_input, image_output, mosaic_trigger])
+            
+    # The 'select' event is built into ImageEditor
+    image_input.select(fn=process_selection, inputs=None, outputs=[crop_box_state]).then(
+        fn=apply_transformations, inputs=all_inputs, outputs=[image_input, image_output, mosaic_trigger])
+
+    # The 'change' event in ImageEditor fires on upload and clear
+    image_input.change(fn=on_upload_or_clear, inputs=[image_input], outputs=all_resettable_outputs)
+    reset_btn.click(fn=on_upload_or_clear, inputs=[image_input], outputs=all_resettable_outputs)
+    
+    rotate_90_btn.click(lambda x: (x + 90) % 360, inputs=[rotation_slider], outputs=[rotation_slider])
+    rotate_180_btn.click(lambda: 180, inputs=None, outputs=[rotation_slider])
+    rotate_270_btn.click(lambda: -90, inputs=None, outputs=[rotation_slider])
+    
+    mosaic_btn.click(lambda: True, None, mosaic_trigger).then(
+        fn=apply_transformations, inputs=all_inputs, outputs=[image_input, image_output, mosaic_trigger])
+
+    range_map = {
+        (scale_min, scale_max): scale_slider, (rotation_min, rotation_max): rotation_slider,
+        (shear_x_min, shear_x_max): shear_x_slider, (shear_y_min, shear_y_max): shear_y_slider,
+        (brightness_min, brightness_max): brightness_slider, (contrast_min, contrast_max): contrast_slider,
+        (saturation_min, saturation_max): saturation_slider, (hue_min, hue_max): hue_slider,
+        (gamma_min, gamma_max): gamma_slider, (blur_min, blur_max): blur_slider,
+        (noise_min, noise_max): noise_slider, (cutout_n_min, cutout_n_max): cutout_n_slider,
+        (cutout_ratio_min, cutout_ratio_max): cutout_ratio_slider
+    }
+    for (min_comp, max_comp), slider in range_map.items():
+        min_comp.change(fn=update_slider, inputs=[min_comp, max_comp, slider], outputs=[slider])
+        max_comp.change(fn=update_slider, inputs=[min_comp, max_comp, slider], outputs=[slider])
+
 demo.launch()