Switch to Gemini + update requirements

app.py

@@ -70,7 +70,7 @@ def replace_background(input_path, bg_choice):
     return result_path
 
 #def process_image(input_img, bg_choice, bg_upload, logo_upload, logo_transparency, logo_position, brand_color):
-def process_image(input_img, bg_choice, bg_upload, logo_upload, logo_transparency, logo_position, blur_background=False):
+def process_image(input_img, bg_choice, bg_upload, logo_upload, logo_transparency, logo_position, blur_background, blend_strength):
     if input_img is None:
         return []
 
@@ -98,14 +98,38 @@ def process_image(input_img, bg_choice, bg_upload, logo_upload, logo_transparenc
     # Extract mask from alpha channel
     mask = fg.split()[3]
 
-    # Feather the edges (soft transition)
+    # Feather edges for smooth transition
     mask = mask.filter(ImageFilter.GaussianBlur(radius=2))
-
-    # Rebuild foreground with feathered alpha
     fg.putalpha(mask)
 
     # ---------------------------
-    # Merge main photo with background
+    # Color matching (adjust subject brightness to background)
+    # ---------------------------
+    try:
+        from PIL import ImageStat, ImageEnhance
+
+        # Average brightness of background
+        stat_bg = ImageStat.Stat(bg.convert("L"))
+        bg_brightness = stat_bg.mean[0]
+
+        # Average brightness of foreground
+        stat_fg = ImageStat.Stat(fg.convert("L"))
+        fg_brightness = stat_fg.mean[0]
+
+        if fg_brightness > 0:
+            brightness_ratio = bg_brightness / fg_brightness
+
+            # Apply brightness adjustment with user-controlled strength
+            enhancer = ImageEnhance.Brightness(fg)
+            adjusted = enhancer.enhance(brightness_ratio)
+
+            # Blend original fg with adjusted fg
+            fg = Image.blend(fg, adjusted, alpha=blend_strength)
+    except Exception as e:
+        print("Color match failed:", e)
+
+    # ---------------------------
+    # Merge subject with background
     # ---------------------------
     result = Image.alpha_composite(bg, fg)
 
@@ -136,6 +160,7 @@ def process_image(input_img, bg_choice, bg_upload, logo_upload, logo_transparenc
     timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
     result_path = os.path.join(RESULTS_DIR, f"result_{timestamp}.png")
     result.save(result_path)
+
     cleanup_old_files(RESULTS_DIR)
     return [result_path]
 
@@ -223,6 +248,10 @@ with gr.Blocks(css="footer {display:none !important}") as demo:
                 value="Bottom-Right",
                 label="Logo Position"
             )
+        
+        blend_strength = gr.Slider(0, 1, value=0.5, step=0.1, label="Blending Strength (Realism)")
+        blur_background = gr.Checkbox(label="Blur Background", value=False)
+
             #brand_color = gr.ColorPicker(label="Brand Colour")
 
         btn = gr.Button("✨ Generate New Photo")
@@ -230,8 +259,7 @@ with gr.Blocks(css="footer {display:none !important}") as demo:
 
         btn.click(
             fn=process_image,
-            #inputs=[input_img, bg_choices, bg_upload, logo_upload, logo_transparency, logo_position, brand_color],
-            inputs=[input_img, bg_choices, bg_upload, logo_upload, logo_transparency, logo_position],
+            inputs=[input_img, bg_choices, bg_upload, logo_upload, logo_transparency, logo_position, blur_background, blend_strength],
             outputs=output_imgs
         )