Update app.py

app.py

@@ -42,6 +42,27 @@ def create_face_mask(image, bbox_coords, face_preserve):
     
     return mask
 
+def create_mask_preview(image, bbox_x1, bbox_y1, bbox_x2, bbox_y2, face_preserve):
+    """Erstellt eine Vorschau der Maske auf dem Originalbild"""
+    if image is None or None in [bbox_x1, bbox_y1, bbox_x2, bbox_y2]:
+        return None
+    
+    # Bild kopieren für Vorschau
+    preview = image.copy()
+    draw = ImageDraw.Draw(preview)
+    
+    # Rechteck zeichnen basierend auf face_preserve
+    if face_preserve:
+        # GESICHTSERHALTUNG: Roter Rahmen um geschützten Bereich
+        draw.rectangle([bbox_x1, bbox_y1, bbox_x2, bbox_y2], outline="red", width=3)
+        draw.text((bbox_x1, bbox_y1-25), "GESCHÜTZT", fill="red")
+    else:
+        # NUR GESICHT VERÄNDERN: Grüner Rahmen um Veränderungsbereich
+        draw.rectangle([bbox_x1, bbox_y1, bbox_x2, bbox_y2], outline="green", width=3)
+        draw.text((bbox_x1, bbox_y1-25), "WIRD VERÄNDERT", fill="green")
+    
+    return preview
+
 def auto_detect_face_area(image):
     """Optimierten Vorschlag für Gesichtsbereich ohne externe Bibliotheken"""
     width, height = image.size
@@ -323,12 +344,14 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale,
         
 
 def update_bbox_from_image(image):
-    """Aktualisiert die Bounding-Box-Koordinaten wenn ein Bild hochgeladen wird"""
+    """Aktualisiert die Bounding-Box-Koordinaten und erstellt Vorschau wenn ein Bild hochgeladen wird"""
     if image is None:
-        return None, None, None, None
+        return None, None, None, None, None
     
     bbox = auto_detect_face_area(image)
-    return bbox[0], bbox[1], bbox[2], bbox[3]
+    # Erstelle Vorschau mit Standard-Gesichtserhaltung (True)
+    preview = create_mask_preview(image, bbox[0], bbox[1], bbox[2], bbox[3], True)
+    return bbox[0], bbox[1], bbox[2], bbox[3], preview
 
 def main_ui():
     with gr.Blocks(
@@ -426,12 +449,21 @@ def main_ui():
                 gr.Markdown("**Lade ein Bild hoch und beschreibe die gewünschte Veränderung:**")
                 
                 with gr.Row():
-                    img_input = gr.Image(
-                        type="pil", 
-                        label="Eingabebild",
-                        height=300,
-                        sources=["upload"]  # Nur Upload-Button anzeigen
-                    )
+                    with gr.Column():
+                        img_input = gr.Image(
+                            type="pil", 
+                            label="Eingabebild mit Masken-Vorschau",
+                            height=300,
+                            sources=["upload"]  # Nur Upload-Button anzeigen
+                        )
+                    with gr.Column():
+                        gr.Markdown("**📐 Masken-Koordinaten**")
+                        with gr.Row():
+                            bbox_x1 = gr.Number(label="Links (x1)", value=100, precision=0)
+                            bbox_y1 = gr.Number(label="Oben (y1)", value=100, precision=0)
+                        with gr.Row():
+                            bbox_x2 = gr.Number(label="Rechts (x2)", value=300, precision=0)
+                            bbox_y2 = gr.Number(label="Unten (y2)", value=300, precision=0)
                 
                 with gr.Row():
                     with gr.Column():
@@ -474,43 +506,15 @@ def main_ui():
                     face_preserve = gr.Checkbox(
                         label="Gesicht, Tier, Gegenstand beibehalten", 
                         value=True,
-                        info="Aktiviert: Bildelement bleibt erhalten, Hintergrund wird verändert | Deaktiviert: Nur Bildelement wird verändert"
+                        info="🔴 ROT: Element geschützt | 🟢 GRÜN: Element wird verändert"
                     )
                 
-                with gr.Row():
-                    gr.Markdown("**Bildelementbereich anpassen**")
-                
-                with gr.Row():
-                    bbox_x1 = gr.Number(
-                        label="Links (x1)", 
-                        value=100, 
-                        precision=0,
-                        info="Linke Kante des Gesichtsbereichs"
-                    )
-                    bbox_y1 = gr.Number(
-                        label="Oben (y1)", 
-                        value=100, 
-                        precision=0,
-                        info="Obere Kante des Gesichtsbereichs"
-                    )
-                    bbox_x2 = gr.Number(
-                        label="Rechts (x2)", 
-                        value=300, 
-                        precision=0,
-                        info="Rechte Kante des Gesichtsbereichs"
-                    )
-                    bbox_y2 = gr.Number(
-                        label="Unten (y2)", 
-                        value=300, 
-                        precision=0,
-                        info="Untere Kante des Gesichtsbereichs"
-                    )
-
                 with gr.Row():
                    gr.Markdown(
                      "**Achtung:**\n"
                      "• **Automatische Bildelementerkennung** setzt Koordinaten beim Upload\n"  
-                     "• **Koordinaten nur bei erkennbaren Verzerrungen anpassen** (Bereiche leicht verschieben)"
+                     "• **Koordinaten anpassen** um Bereich zu vergrößern/verkleinern\n"
+                     "• **Farbiger Rahmen** zeigt an was geschützt/verändert wird"
                     )
                 
                           
@@ -523,13 +527,23 @@ def main_ui():
                         type="pil"
                     )
                 
-                # Event-Handler für Bild-Upload
+                # Event-Handler für Bild-Upload - Jetzt MIT Vorschau
                 img_input.change(
                     fn=update_bbox_from_image,
                     inputs=[img_input],
-                    outputs=[bbox_x1, bbox_y1, bbox_x2, bbox_y2]
+                    outputs=[bbox_x1, bbox_y1, bbox_x2, bbox_y2, img_input]
                 )
                 
+                # Event-Handler für Live-Masken-Vorschau bei Koordinaten-Änderungen
+                coordinates_inputs = [bbox_x1, bbox_y1, bbox_x2, bbox_y2, face_preserve]
+                
+                for coordinate in coordinates_inputs:
+                    coordinate.change(
+                        fn=create_mask_preview,
+                        inputs=[img_input] + coordinates_inputs,
+                        outputs=img_input
+                    )
+                
                 transform_btn.click(
                     fn=img_to_image,
                     inputs=[