Update app.py

app.py

@@ -1,34 +1,42 @@
 import gradio as gr
 from PIL import Image, ImageEnhance
 
-def enhance_image(image, vibrancy_factor, sharpness_factor, yellow_reducer, brightness_boost):
+def enhance_image(image, 
+                 vibrancy=1.0, 
+                 sharpness=1.0, 
+                 yellow_reduce=0.0, 
+                 magenta_reduce=0.0, 
+                 brightness=0.0):
+    
+    # Convert to RGB
     image = image.convert("RGB")
     
-    # Targeted yellow reduction (without blue boost)
+    # Yellow reduction matrix (targets red/green channels)
     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
+        1.0 - yellow_reduce*0.4, 0, 0, 0,
+        0, 1.0 - yellow_reduce*0.6, 0, 0,
+        0, 0, 1.0 + yellow_reduce*0.2, 0  # Compensate with slight blue
     )
     image = image.convert("RGB", yellow_matrix)
     
-    # Apply magenta reduction
+    # Magenta reduction matrix (targets red/blue channels)
     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
+        1.0 - magenta_reduce*0.3, 0, 0, 0,
+        0, 1.0 + magenta_reduce*0.4, 0, 0,
+        0, 0, 1.0 - magenta_reduce*0.3, 0
     )
     image = image.convert("RGB", magenta_matrix)
     
     # Apply brightness boost
-    image = ImageEnhance.Brightness(image).enhance(1.0 + brightness_boost)
+    if brightness > 0:
+        image = ImageEnhance.Brightness(image).enhance(1.0 + brightness)
     
     # Apply vibrancy enhancements
-    image = ImageEnhance.Color(image).enhance(vibrancy_factor)
-    image = ImageEnhance.Contrast(image).enhance(1.0 + (vibrancy_factor-1)*0.1)
+    image = ImageEnhance.Color(image).enhance(vibrancy)
+    image = ImageEnhance.Contrast(image).enhance(1.0 + (vibrancy-1)*0.15)
     
     # Apply sharpness last
-    image = ImageEnhance.Sharpness(image).enhance(sharpness_factor)
+    image = ImageEnhance.Sharpness(image).enhance(sharpness)
     
     return image
 
@@ -38,12 +46,13 @@ iface = gr.Interface(
         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="Yellow Reduction"),
+        gr.Slider(0.0, 0.5, 0.0, step=0.05, label="Magenta Reduction"),
         gr.Slider(0.0, 0.5, 0.0, step=0.05, label="Brightness Boost")
     ],
     outputs=gr.Image(type="pil", label="Enhanced Image", format="jpeg"),
-    title="Professional Image Enhancer",
-    description="Targeted yellow/magenta removal with natural color preservation"
+    title="Professional Image Corrector",
+    description="Independent controls for yellow/magenta reduction with brightness preservation"
 )
 
 if __name__ == "__main__":