Update app.py

app.py

@@ -637,7 +637,6 @@ def text_to_image(prompt, model_id, steps, guidance_scale, progress=gr.Progress(
         return None, error_msg
 
 
-
 def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
                  mode, bbox_x1, bbox_y1, bbox_x2, bbox_y2,
                  progress=gr.Progress()):
@@ -724,6 +723,17 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
             # Für andere Modi: Normaler Prompt
             enhanced_prompt = prompt
 
+        # ===== HINTERGRUND-BOOSTER FÜR UMGEBUNGS-ÄNDERUNG =====
+        if mode == "environment_change":
+            # Booster für komplett neue Umgebung
+            background_boosters = "complete scene, full background, entire environment, "
+            
+            # Prüfe ob schon Booster enthalten
+            if not any(keyword in enhanced_prompt.lower() for keyword in 
+                      ["complete scene", "full background", "entire environment", "whole setting"]):
+                enhanced_prompt = background_boosters + enhanced_prompt
+                print(f"🌳 Hintergrund-Booster hinzugefügt: {background_boosters}")
+
         print(f"🎯 Finaler Prompt für {mode}: {enhanced_prompt}")
 
 
@@ -737,16 +747,20 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         if mode == "focus_change":
             # MODUS 2: Focus verändern → OpenPose + Canny
             keep_environment = False
-            controlnet_strength = adj_strength * 0.5
+            controlnet_strength = adj_strength * 0.7  # Höher für Strukturerhalt
             print(f"🎯 MODUS: Focus verändern → OpenPose+Canny (keep_environment=False)")
-        else:
-            # MODUS 1 & 3: Umgebung/Gesicht ändern → Depth + Canny
+            
+        elif mode == "environment_change":
+            # MODUS 1: Umgebung ändern → Depth + Canny
             keep_environment = True
-            controlnet_strength = adj_strength * 0.8
-            if mode == "environment_change":
-                print(f"🎯 MODUS: Umgebung ändern → Depth+Canny (keep_environment=True)")
-            else:
-                print(f"🎯 MODUS: Ausschließlich Gesicht → Depth+Canny (keep_environment=True)")
+            controlnet_strength = adj_strength * 0.3  # NIEDRIG für radikale Änderungen
+            print(f"🎯 MODUS: Umgebung ändern → Depth+Canny (keep_environment=True)")
+            
+        else:  # face_only_change
+            # MODUS 3: Ausschließlich Gesicht → Depth + Canny
+            keep_environment = True
+            controlnet_strength = adj_strength * 0.5  # Balance für Gesichtsdetails
+            print(f"🎯 MODUS: Ausschließlich Gesicht → Depth+Canny (keep_environment=True)")
             
         controlnet_steps = min(25, int(steps * 0.8))
         print(f"⚙️ ControlNet Settings: Strength={controlnet_strength:.3f}, Steps={controlnet_steps}")
@@ -833,8 +847,7 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         print(f"❌ Fehler in img_to_image: {e}")
         import traceback
         traceback.print_exc()
-        return None
-        
+        return None  
 
 def update_bbox_from_image(image):
     """Aktualisiert die Bounding-Box-Koordinaten wenn ein Bild hochgeladen wird"""