Update app.py

app.py

@@ -1,47 +1,49 @@
 import gradio as gr
 from PIL import Image, ImageEnhance, ImageFilter
 
-def enhance_vibrancy(image, vibrancy_factor):
+def enhance_image(image, vibrancy_factor, sharpness_factor, depth_factor):
     # Convert to RGB (JPEG doesn't support transparency)
     image = image.convert("RGB")
     
-    # Enhance color
-    enhancer = ImageEnhance.Color(image)
-    image = enhancer.enhance(vibrancy_factor)
-    
-    # Add minimal brightness/contrast boost
+    # 1. Vibrancy (color enhancement)
+    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)
     
-    # Add sharpness (1.5 base value, scaled slightly with vibrancy)
-    sharpness_factor = 1.5 + (vibrancy_factor-1)*0.3
+    # 2. Sharpness
     image = ImageEnhance.Sharpness(image).enhance(sharpness_factor)
     
-    # Add depth effect with subtle background blur
-    # Create a slightly blurred version
-    blurred = image.filter(ImageFilter.GaussianBlur(radius=2))
-    
-    # Enhance contrast on original to make subject pop
-    foreground = ImageEnhance.Contrast(image).enhance(1.2)
-    
-    # Blend original (sharp) with blurred background (70% foreground, 30% blur)
-    image = Image.blend(blurred, foreground, 0.7)
+    # 3. Depth (make subject pop)
+    if depth_factor > 1.0:
+        # Create blurred, darkened background
+        background = image.filter(ImageFilter.GaussianBlur(radius=depth_factor * 2))
+        background = ImageEnhance.Brightness(background).enhance(0.8 / depth_factor)  # Darken slightly
+        
+        # Boost foreground contrast and brightness
+        foreground = ImageEnhance.Contrast(image).enhance(1.0 + (depth_factor-1)*0.5)
+        foreground = ImageEnhance.Brightness(foreground).enhance(1.0 + (depth_factor-1)*0.2)
+        
+        # Blend: more depth_factor = more foreground dominance
+        alpha = 0.5 + (depth_factor-1) * 0.25  # 0.5 to 0.75 blend
+        image = Image.blend(background, foreground, alpha)
     
     return image
 
 iface = gr.Interface(
-    fn=enhance_vibrancy,
+    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, 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, 2.0, 1.0, step=0.1, label="Depth Level")
     ],
     outputs=gr.Image(
         type="pil",
         label="Enhanced Image (JPEG)",
         format="jpeg"  # Force JPEG output
     ),
-    title="  Vibrancy Booster",
-    description="enhancing colors, sharpness, and depth (1.0 = original vibrancy)"
+    title="Vibrancy, Sharpness & Depth Booster",
+    description="Adjust vibrancy, sharpness, and depth independently (1.0 = original)"
 )
 
 if __name__ == "__main__":