Update app.py

app.py

@@ -1073,7 +1073,7 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         # ===== CONTROLNET: MAPS ERSTELLEN =====
         print(f"📊 ControlNet Input Größe: {scaled_image.size}")
         
-        controlnet_maps = controlnet_processor.prepare_controlnet_maps(
+        controlnet_maps, debug_maps = controlnet_processor.prepare_controlnet_maps(
             image=scaled_image,
             keep_environment=keep_environment
         )
@@ -1159,15 +1159,25 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         mask_preview = original_mask.convert("RGB")
         raw_sam_mask_display = raw_mask.convert("RGB")
 
-        # 3. DREI Bilder zurückgeben
-        return final_image, raw_sam_mask_display, mask_preview
+    
+        if "pose" in debug_maps:
+            controlnet_map1 = debug_maps["pose"]
+            map1_label = "🎭 Pose Map"
+        else:
+            controlnet_map1 = debug_maps["depth"]
+            map1_label = "🏔️ Depth Map"
+
+        controlnet_map2 = debug_maps["canny"]
+
+        # Return 5 Werte:
+        return final_image, raw_sam_mask_display, mask_preview, controlnet_map1, controlnet_map2
 
 
     except Exception as e:
         print(f"❌ Fehler in img_to_image: {e}")
         import traceback
         traceback.print_exc()
-        return final_image, None, None
+        return final_image, None, None, None, None
 
 
 def update_bbox_from_image(image):
@@ -1556,6 +1566,20 @@ def main_ui():
                         height=300,
                         show_download_button=False
                     )
+
+                with gr.Row():                 
+                    pose_map_output = gr.Image(
+                        label="🎭 Pose/Depth Map", 
+                        type="pil", 
+                        height=300,
+                        show_download_button=False
+                    )
+                    canny_map_output = gr.Image(
+                        label="📐 Canny Edge Map", 
+                        type="pil", 
+                        height=300,
+                        show_download_button=False
+                    )
                 
                 #with gr.Row():
                 #    img_output = gr.Image(
@@ -1611,7 +1635,7 @@ def main_ui():
                     ],
                     #outputs=img_output,
                     #outputs=[img_output, mask_output],# NEU: Zwei Ausgaben
-                    outputs=[img_output, sam_raw_mask_output, processed_mask_output],
+                    outputs=[img_output, sam_raw_mask_output, processed_mask_output, pose_map_output, canny_map_output],
                     concurrency_limit=1
                 )