Update app.py

app.py

@@ -218,7 +218,17 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         # PARAMETER-TUNING
         # -------------------------------
         adj_strength = min(0.85, strength * 1.25)
-        controlnet_strength = adj_strength * (0.8 if face_preserve else 0.5)
+        
+        # CONTROLNET-STRENGTH ANPASSEN ABHÄNGIG VOM MODUS
+        if face_preserve:
+            # UMGEBUNG BEIBEHALTEN, PERSON ÄNDERN → HOHE STRENGTH FÜR KOMBINIERTE STRATEGIE
+            controlnet_strength = adj_strength * 0.8  # 80% für kombinierte OpenPose + Canny
+            print(f"🎯 ControlNet Modus: Umgebung beibehalten (Strength = {controlnet_strength:.3f})")
+        else:
+            # PERSON BEIBEHALTEN, UMGEBUNG ÄNDERN → NORMALE STRENGTH FÜR OPENPOSE
+            controlnet_strength = adj_strength * 0.5  # 50% für OpenPose
+            print(f"🎯 ControlNet Modus: Person beibehalten (Strength = {controlnet_strength:.3f})")
+            
         controlnet_steps = min(25, int(steps * 0.8))
 
         print(f"🎯 Steps={steps}, ControlNet-Steps={controlnet_steps}, Strength={controlnet_strength:.3f}")
@@ -235,10 +245,12 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
             steps=controlnet_steps,
             guidance_scale=guidance_scale,
             controlnet_strength=controlnet_strength,
-            progress=progress
+            progress=progress,
+            keep_environment=face_preserve  # WICHTIG: Parameter hinzugefügt!
         )
 
         print(f"✅ ControlNet Output erhalten: {type(controlnet_output)}")
+        print(f"✅ Inpaint Input erhalten: {type(inpaint_input)}")
 
         # -------------------------------
         # INPAINT (STABLE DIFFUSION IMG2IMG)
@@ -246,6 +258,8 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         progress(0.3, desc="ControlNet abgeschlossen – starte Inpaint...")
 
         pipe = load_img2img()  # ← deine bestehende Funktion
+        
+        # inpaint_input ist IMMER das Originalbild (laut neuer ControlNet-Logik)
         img_resized = inpaint_input.convert("RGB").resize((IMG_SIZE, IMG_SIZE))
 
         adj_guidance = min(guidance_scale, 12.0)
@@ -306,7 +320,7 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         import traceback
         traceback.print_exc()
         return None
-
+        
 
 def update_bbox_from_image(image):
     """Aktualisiert die Bounding-Box-Koordinaten wenn ein Bild hochgeladen wird"""