Switch to Gemini + update requirements

app.py

@@ -5,6 +5,7 @@ from fastapi import FastAPI, UploadFile, Form
 from fastapi.responses import FileResponse, JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
+from PIL import ImageFilter, ImageOps
 from rembg import remove
 import google.generativeai as genai  # Gemini SDK
 import gradio as gr
@@ -69,31 +70,54 @@ 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):
+def process_image(input_img, bg_choice, bg_upload, logo_upload, logo_transparency, logo_position, blur_background=False):
     if input_img is None:
         return []
 
     temp_path = os.path.join(UPLOAD_DIR, f"upload_{int(time.time())}.png")
     input_img.save(temp_path)
 
+    # ---------------------------
     # Background selection
+    # ---------------------------
     if bg_upload is not None:
         bg = bg_upload.convert("RGBA").resize(input_img.size)
     else:
         bg_path = os.path.join(BG_DIR, bg_choice)
         bg = Image.open(bg_path).convert("RGBA").resize(input_img.size)
 
-    # Foreground (removed background)
+    # Optionally blur background
+    if blur_background:
+        bg = bg.filter(ImageFilter.GaussianBlur(radius=3))
+
+    # ---------------------------
+    # Foreground (with feathered mask)
+    # ---------------------------
     fg = remove(input_img.convert("RGBA"))
 
+    # Extract mask from alpha channel
+    mask = fg.split()[3]
+
+    # Feather the edges (soft transition)
+    mask = mask.filter(ImageFilter.GaussianBlur(radius=2))
+
+    # Rebuild foreground with feathered alpha
+    fg.putalpha(mask)
+
+    # ---------------------------
     # Merge main photo with background
+    # ---------------------------
     result = Image.alpha_composite(bg, fg)
 
+    # ---------------------------
     # If logo uploaded
+    # ---------------------------
     if logo_upload is not None:
         logo = logo_upload.convert("RGBA")
         scale = result.width // 5
         logo.thumbnail((scale, scale))
+
+        # Apply transparency
         alpha = logo.split()[3].point(lambda p: p * (logo_transparency / 100))
         logo.putalpha(alpha)
 
@@ -105,14 +129,10 @@ def process_image(input_img, bg_choice, bg_upload, logo_upload, logo_transparenc
             "Center": ((result.width - logo.width) // 2, (result.height - logo.height) // 2),
         }
         result.paste(logo, pos_map[logo_position], logo)
-        
-    '''
-    # Apply brand colour tint
-    if brand_color:
-        tint = Image.new("RGBA", result.size, brand_color + "20")
-        result = Image.alpha_composite(result, tint)
-        '''
 
+    # ---------------------------
+    # Save final result
+    # ---------------------------
     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)