Update app.py

app.py

@@ -1,10 +1,24 @@
 import gradio as gr
 from PIL import Image, ImageEnhance
 
-def enhance_image(image, vibrancy_factor, sharpness_factor):
+def enhance_image(image, vibrancy_factor, sharpness_factor, blue_yellow_factor):
     # Convert to RGB (JPEG doesn't support transparency)
     image = image.convert("RGB")
     
+    # Apply blue/yellow color balance
+    factor = blue_yellow_factor  # -1.0 (yellow) to 1.0 (blue)
+    r_scale = 1.0 - factor  # Reduce red when adding blue
+    g_scale = 1.0 - factor  # Reduce green when adding blue
+    b_scale = 1.0 + factor  # Increase blue
+    
+    # Create color balance matrix
+    matrix = (
+        r_scale, 0, 0, 0,
+        0, g_scale, 0, 0,
+        0, 0, b_scale, 0
+    )
+    image = image.convert("RGB", matrix)
+    
     # Vibrancy (color enhancement)
     image = ImageEnhance.Color(image).enhance(vibrancy_factor)
     image = ImageEnhance.Brightness(image).enhance(1.0 + (vibrancy_factor-1)*0.2)
@@ -20,15 +34,16 @@ iface = gr.Interface(
     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(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")
     ],
     outputs=gr.Image(
         type="pil",
         label="Enhanced Image (JPEG)",
-        format="jpeg"  # Force JPEG output
+        format="jpeg"
     ),
     title="Vibrancy & Sharpness Booster",
-    description="Adjust vibrancy and sharpness (1.0 = original)"
+    description="Adjust vibrancy, sharpness, and color balance (1.0 = original)"
 )
 
 if __name__ == "__main__":