Upload app.py

app.py

@@ -88,28 +88,23 @@ def extract_head_crop(image_pil, face_bbox):
     crop_y1 = max(0, int(cy - crop_size/1.2)) # Più spazio sopra per i capelli
     crop_x2 = min(image_pil.width, int(cx + crop_size/1.5))
     
-    # Meno spazio sotto: fermiamoci alla base del collo, evitando le spalle
-    # y2 è circa il mento, y2 + h*0.5 è la fine del collo
-    crop_y2 = min(image_pil.height, int(y2 + h * 0.45)) 
+    # Taglio ancora più drastico: fermiamoci poco sotto il mento per escludere a priori qualsiasi spalla
+    crop_y2 = min(image_pil.height, int(y2 + h * 0.35)) 
     
     return image_pil.crop((crop_x1, crop_y1, crop_x2, crop_y2))
 
-def fade_bottom(image_rgba, fade_height_ratio=0.15):
-    """Applica una sfumatura trasparente alla base dell'immagine per fondere dolcemente il collo."""
+def cut_bottom(image_rgba, cut_height_ratio=0.20):
+    """Rende completamente trasparente la base dell'immagine (Hard Cut) per eliminare il collo inferiore."""
     width, height = image_rgba.size
-    fade_height = int(height * fade_height_ratio)
-    if fade_height == 0:
+    cut_height = int(height * cut_height_ratio)
+    if cut_height == 0:
         return image_rgba
         
     r, g, b, a = image_rgba.split()
     alpha_np = np.array(a)
     
-    # Crea un gradiente da 1.0 (opaco) a 0.0 (trasparente)
-    gradient = np.linspace(1, 0, fade_height)
-    gradient = np.tile(gradient[:, None], (1, width))
-    
-    # Applica il gradiente alla parte bassa del canale alpha
-    alpha_np[-fade_height:, :] = (alpha_np[-fade_height:, :] * gradient).astype(np.uint8)
+    # Taglio netto: zero alpha nella parte inferiore
+    alpha_np[-cut_height:, :] = 0
     
     a = Image.fromarray(alpha_np)
     image_rgba.putalpha(a)
@@ -155,11 +150,11 @@ def generate_avatar(user_image, selected_style):
         user_faces = face_app.get(cv_user_image)
         user_face = sorted(user_faces, key=lambda x: (x.bbox[2]-x.bbox[0])*(x.bbox[3]-x.bbox[1]))[-1]
 
-    # 4. Ritaglio, rimozione sfondo e sfumatura collo
+    # 4. Ritaglio, rimozione sfondo e Hard Cut del collo
     print("Estrazione della testa con Rembg...")
     cropped_user = extract_head_crop(user_image, user_face.bbox)
     head_rgba = rembg.remove(cropped_user, session=rembg_session)
-    head_rgba = fade_bottom(head_rgba, fade_height_ratio=0.15) # Sfuma il collo
+    head_rgba = cut_bottom(head_rgba, cut_height_ratio=0.20) # Taglio netto trasparente
     
     # 5. Analisi della testa scontornata per trovare le nuove coordinate del volto
     head_rgb_cv = cv2.cvtColor(np.array(head_rgba.convert("RGB")), cv2.COLOR_RGB2BGR)