Update app.py

app.py

@@ -1044,11 +1044,13 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
             
             # === 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(
+            processed_mask, raw_mask = controlnet_processor.create_sam_mask(
                 image=image,
                 bbox_coords=(bbox_x1, bbox_y1, bbox_x2, bbox_y2),
                 mode=mode
             )
+
+            original_mask = processed_mask
             
             # 2. BILD UND MASKE GEMEINSAM SKALIEREN (mit Padding)
             scaled_image, scaled_mask, padding_info = scale_image_and_mask_together(
@@ -1155,9 +1157,11 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
 
         # 1. Maske in RGB für die Anzeige konvertieren
         mask_preview = original_mask.convert("RGB")
+        raw_sam_mask = raw_mask.convert("RGB")
 
         # 1. SAM-Rohmaske speichern (vor allen Bearbeitungen)
         raw_sam_mask = original_mask.copy().convert("RGB")
+    
 
         # 2. Nachbearbeitete Maske für Inpainting (ist bereits original_mask nach Verarbeitung)
         processed_mask = mask_preview  # Deine existierende mask_preview