Update app.py

app.py

@@ -1142,6 +1142,7 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         else:
             # Keine Maske: Einfach das generierte Bild zurückgeben
             final_image = generated_image
+            mask_preview = Image.new("RGB", (512, 512), color="gray")
             print(f"ℹ️ Keine Maske → Direkte Rückgabe des Bildes")
 
         end_time = time.time()
@@ -1152,7 +1153,13 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         print(f"⚙️ ControlNet: {'Depth+Canny' if keep_environment else 'OpenPose+Canny'}")
         print(f"📊 Finale Bildgröße: {final_image.size}")
 
-        return final_image
+        # 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
+
+        #return final_image
 
     except Exception as e:
         print(f"❌ Fehler in img_to_image: {e}")
@@ -1532,6 +1539,13 @@ def main_ui():
                         type="pil",
                         height=400
                     )
+                    mask_output = gr.Image(
+                        label="🔍 Debug: SAM-Maske (Person sollte SCHWARZ sein)",
+                        type="pil",
+                        height=400
+                    )
+
+
                 
                 # EVENT-HANDLER FÜR DYNAMISCHE BILDGRÖßEN
                 img_input.upload(
@@ -1570,7 +1584,8 @@ def main_ui():
                         strength_slider, img_steps, img_guidance, 
                         mode_radio, bbox_x1, bbox_y1, bbox_x2, bbox_y2
                     ],
-                    outputs=img_output,
+                    #outputs=img_output,
+                    outputs=[img_output, mask_output]  # NEU: Zwei Ausgaben
                     concurrency_limit=1
                 )