Update controlnet_module.py

controlnet_module.py

@@ -902,6 +902,7 @@ class ControlNetProcessor:
             print("   Schritt 2/2: Extrahiere Canny Edges...")
             canny_map = self.extract_canny_edges(image)
             conditioning_images = [depth_map, canny_map]
+            extra_maps = {"depth": depth_map, "canny": canny_map}  # NEU
         else:
             print("   Modus: OpenPose + Canny")
             print("   Schritt 1/2: Extrahiere Pose Map...")
@@ -909,6 +910,7 @@ class ControlNetProcessor:
             print("   Schritt 2/2: Extrahiere Canny Edges...")
             canny_map = self.extract_canny_edges(image)
             conditioning_images = [pose_map, canny_map]
+            extra_maps = {"pose": pose_map, "canny": canny_map}
 
         print("-" * 60)
         print(f"✅ {len(conditioning_images)} CONDITIONING-MAPS ERSTELLT")
@@ -916,7 +918,7 @@ class ControlNetProcessor:
             print(f"   Map {i+1}: {img.size}, Modus: {img.mode}")
         print("#" * 80)
         
-        return conditioning_images
+        return conditioning_images, extra_maps
 
 
 # Globale Instanz