Update controlnet_module.py

controlnet_module.py

@@ -143,24 +143,24 @@ class ControlNetProcessor:
                     raise
             return self.pipe_canny
 
-    def generate_with_controlnet(
-        self, image, prompt, negative_prompt,
+
+    def generate_with_controlnet(self, image, prompt, negative_prompt,
         steps, guidance_scale, controlnet_strength,
         progress=None, keep_environment=False
     ):
-        """Generiert Bild mit ControlNet und Fortschrittsanzeige"""
-        try:
-            # --- ENTSCHEIDUNG: Welches ControlNet für welche Aufgabe? ---
-            if keep_environment:
-                # PERSON ÄNDERN, UMGEBUNG BEIBEHALTEN
-                controlnet_type = "canny"
-                print("🎯 ControlNet Modus: Umgebung beibehalten (Canny Edge)")
-                conditioning_image = self.extract_canny_edges(image)
-            else:
-                # UMGEBUNG ÄNDERN, PERSON BEIBEHALTEN  
-                controlnet_type = "openpose"
-                print("🎯 ControlNet Modus: Person beibehalten (OpenPose)")
-                conditioning_image = self.extract_pose(image)
+    try:
+        # --- KORREKTE LOGIK ---
+        if keep_environment:
+            # UMGEBUNG BEIBEHALTEN, PERSON ÄNDERN
+            controlnet_type = "canny"  # ✅ Canny behält Umgebung
+            print("🎯 ControlNet Modus: Umgebung beibehalten (Canny Edge)")
+            conditioning_image = self.extract_canny_edges(image)
+        else:
+            # PERSON BEIBEHALTEN, UMGEBUNG ÄNDERN  
+            controlnet_type = "openpose"  # ✅ OpenPose behält Person
+            print("🎯 ControlNet Modus: Person beibehalten (OpenPose)")
+            conditioning_image = self.extract_pose(image)   
+            
 
             pipe = self.load_controlnet_pipeline(controlnet_type)