Update app.py

app.py

@@ -1531,19 +1531,42 @@ def main_ui():
                     )
                 
                 transform_btn = gr.Button("🔄 Bild transformieren", variant="primary")
-                
+
+
                 with gr.Row():
                     img_output = gr.Image(
-                        label="✨ Transformiertes Bild (512×512 - SD-Technologie-Limit)",
+                        label="✨ Transformiertes Bild",
                         show_download_button=True,
                         type="pil",
                         height=400
                     )
-                    mask_output = gr.Image(
-                        label="🔍 Debug: SAM-Maske (Person sollte SCHWARZ sein)",
+    
+                with gr.Row():
+                    sam_raw_mask_output = gr.Image(
+                        label="🔍 SAM-Rohmaske (Vor Nachbearbeitung)",
                         type="pil",
-                        height=400
+                        height=300,
+                        show_download_button=False
+                    )
+                    processed_mask_output = gr.Image(
+                        label="🛠️ Nachbearbeitete Maske (Für Inpainting)",
+                        type="pil",
+                        height=300,
+                        show_download_button=False
                     )
+                
+                #with gr.Row():
+                #    img_output = gr.Image(
+                #        label="✨ Transformiertes Bild (512×512 - SD-Technologie-Limit)",
+                #        show_download_button=True,
+                #        type="pil",
+                #        height=400
+                #    )
+                #    mask_output = gr.Image(
+                #        label="🔍 Debug: SAM-Maske (Person sollte SCHWARZ sein)",
+                #        type="pil",
+                #        height=400
+                #    )