Update app.py

app.py

@@ -1,38 +1,33 @@
 import gradio as gr
 from PIL import Image, ImageEnhance
 
-def enhance_image(image, vibrancy_factor, sharpness_factor, blue_yellow_factor, magenta_factor):
-    # Convert to RGB (JPEG doesn't support transparency)
+def enhance_image(image, vibrancy_factor, sharpness_factor, yellow_reducer, brightness_boost):
     image = image.convert("RGB")
     
-    # Apply blue/yellow color balance
-    by_r_scale = 1.0 - blue_yellow_factor  # Reduce red when adding blue
-    by_g_scale = 1.0 - blue_yellow_factor  # Reduce green when adding blue
-    by_b_scale = 1.0 + blue_yellow_factor  # Increase blue
-    matrix_blue_yellow = (
-        by_r_scale, 0, 0, 0,
-        0, by_g_scale, 0, 0,
-        0, 0, by_b_scale, 0
+    # Targeted yellow reduction (without blue boost)
+    yellow_matrix = (
+        1.0 - yellow_reducer*0.3, 0, 0, 0,  # Reduce red contribution
+        0, 1.0 - yellow_reducer*0.5, 0, 0,   # Reduce green more (main yellow component)
+        0, 0, 1.0, 0                        # Leave blue unchanged
     )
-    image = image.convert("RGB", matrix_blue_yellow)
+    image = image.convert("RGB", yellow_matrix)
     
-    # Apply magenta reduction (green enhancement)
-    m_r_scale = 1.0 - magenta_factor  # Reduce red component of magenta
-    m_g_scale = 1.0 + magenta_factor  # Increase green to counteract magenta
-    m_b_scale = 1.0 - magenta_factor  # Reduce blue component of magenta
-    matrix_magenta = (
-        m_r_scale, 0, 0, 0,
-        0, m_g_scale, 0, 0,
-        0, 0, m_b_scale, 0
+    # Apply magenta reduction
+    magenta_matrix = (
+        1.0 - yellow_reducer*0.2, 0, 0, 0,  # Slight red reduction
+        0, 1.0 + yellow_reducer*0.3, 0, 0,  # Boost green to counteract magenta
+        0, 0, 1.0 - yellow_reducer*0.2, 0   # Slight blue reduction
     )
-    image = image.convert("RGB", matrix_magenta)
+    image = image.convert("RGB", magenta_matrix)
     
-    # Vibrancy (color enhancement)
+    # Apply brightness boost
+    image = ImageEnhance.Brightness(image).enhance(1.0 + brightness_boost)
+    
+    # Apply vibrancy enhancements
     image = ImageEnhance.Color(image).enhance(vibrancy_factor)
-    image = ImageEnhance.Brightness(image).enhance(1.0 + (vibrancy_factor-1)*0.2)
     image = ImageEnhance.Contrast(image).enhance(1.0 + (vibrancy_factor-1)*0.1)
     
-    # Sharpness
+    # Apply sharpness last
     image = ImageEnhance.Sharpness(image).enhance(sharpness_factor)
     
     return image
@@ -40,19 +35,15 @@ def enhance_image(image, vibrancy_factor, sharpness_factor, blue_yellow_factor,
 iface = gr.Interface(
     fn=enhance_image,
     inputs=[
-        gr.Image(type="pil", label="Original Image (WEBP)"),
-        gr.Slider(0.5, 2.0, 1.0, step=0.1, label="Vibrancy Level"),
-        gr.Slider(0.5, 3.0, 1.0, step=0.1, label="Sharpness Level"),
-        gr.Slider(-1.0, 1.0, 0.0, step=0.1, label="Blue/Yellow Balance"),
-        gr.Slider(-1.0, 1.0, 0.0, step=0.1, label="Magenta Reduction")
+        gr.Image(type="pil", label="Original Image"),
+        gr.Slider(0.5, 2.0, 1.0, step=0.1, label="Vibrancy"),
+        gr.Slider(0.5, 3.0, 1.0, step=0.1, label="Sharpness"),
+        gr.Slider(0.0, 0.5, 0.0, step=0.05, label="Yellow/Magenta Reduction"),
+        gr.Slider(0.0, 0.5, 0.0, step=0.05, label="Brightness Boost")
     ],
-    outputs=gr.Image(
-        type="pil",
-        label="Enhanced Image (JPEG)",
-        format="jpeg"
-    ),
-    title="Advanced Image Enhancer",
-    description="Adjust vibrancy, sharpness, color balance, and magenta levels (1.0 = original)"
+    outputs=gr.Image(type="pil", label="Enhanced Image", format="jpeg"),
+    title="Professional Image Enhancer",
+    description="Targeted yellow/magenta removal with natural color preservation"
 )
 
 if __name__ == "__main__":