Update app.py

app.py

@@ -1156,16 +1156,21 @@ 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")
 
-        # 2. Beide Bilder zurückgeben
-        return final_image, mask_preview
+        # 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
+
+        # 3. DREI Bilder zurückgeben
+        return final_image, raw_sam_mask, processed_mask
 
-        #return final_image
 
     except Exception as e:
         print(f"❌ Fehler in img_to_image: {e}")
         import traceback
         traceback.print_exc()
-        return None
+        return final_image, None, None
 
 
 def update_bbox_from_image(image):
@@ -1608,7 +1613,8 @@ def main_ui():
                         mode_radio, bbox_x1, bbox_y1, bbox_x2, bbox_y2
                     ],
                     #outputs=img_output,
-                    outputs=[img_output, mask_output], # NEU: Zwei Ausgaben
+                    #outputs=[img_output, mask_output],# NEU: Zwei Ausgaben
+                    outputs=[img_output, sam_raw_mask_output, processed_mask_output],
                     concurrency_limit=1
                 )