Update app.py

app.py

@@ -1047,17 +1047,23 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         padding_info = None
         scaled_image = None
         scaled_mask = None
+
         
         if bbox_x1 is not None and bbox_y1 is not None and bbox_x2 is not None and bbox_y2 is not None:
             print(f"🎯 BBox Koordinaten erhalten: [{bbox_x1}, {bbox_y1}, {bbox_x2}, {bbox_y2}]")
             
-            # 1. MASKE in Abhängigkeit der BBox Koordinaten AUF ORIGINAL-BILD ERSTELLEN (speichern für späteres Compositing)
-            original_mask = create_face_mask(image, (bbox_x1, bbox_y1, bbox_x2, bbox_y2), mode)
+            # === WICHTIGE ÄNDERUNG: SAM 2 STATT create_face_mask ===
+            # 1. MASKE mit SAM 2 erzeugen (transparent für Benutzer)
+            original_mask = controlnet_processor.create_sam_mask(
+                image=image,
+                bbox_coords=(bbox_x1, bbox_y1, bbox_x2, bbox_y2),
+                mode=mode
+            )
             
             # 2. BILD UND MASKE GEMEINSAM SKALIEREN (mit Padding)
             scaled_image, scaled_mask, padding_info = scale_image_and_mask_together(
                 image.convert("RGB"),  # Originalbild
-                original_mask,          # Originalmaske
+                original_mask,          # SAM 2 Maske (oder Fallback)
                 target_size=IMG_SIZE
             )
 
@@ -1069,6 +1075,7 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
             scaled_image = image.convert("RGB").resize((IMG_SIZE, IMG_SIZE), Image.Resampling.LANCZOS)
             scaled_mask = Image.new("L", (IMG_SIZE, IMG_SIZE), 255)  # Volle Maske
 
+
         progress(0.1, desc="ControlNet läuft...")
 
         # ===== CONTROLNET: MAPS ERSTELLEN =====