Update ui.py

ui.py

@@ -78,26 +78,19 @@ def mock_img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale
     
     return img, mask_preview, mask_preview, controlnet_map, canny_map
 
+
 def mock_update_live_preview(image, bbox_x1, bbox_y1, bbox_x2, bbox_y2, mode):
-    """Mock für Live-Vorschau"""
+    """Mock für Live-Vorschau MIT dynamischen Rahmen"""
     if not image:
         return None
     
-    preview = image.copy()
-    draw = ImageDraw.Draw(preview)
+    # Koordinaten sortieren
+    bbox_coords = sort_coordinates(bbox_x1, bbox_y1, bbox_x2, bbox_y2)
     
-    if all(v is not None for v in [bbox_x1, bbox_y1, bbox_x2, bbox_y2]):
-        x1, y1 = min(bbox_x1, bbox_x2), min(bbox_y1, bbox_y2)
-        x2, y2 = max(bbox_x1, bbox_x2), max(bbox_y1, bbox_y2)
-        
-        colors = {"environment_change": "green", 
-                 "focus_change": "orange", 
-                 "face_only_change": "red"}
-        
-        draw.rectangle([x1, y1, x2, y2], outline=colors.get(mode, "blue"), width=3)
-        draw.text((x1+5, y1-20), f"Mock: {mode}", fill=colors.get(mode, "blue"))
-    
-    return preview
+    # Verbesserte Vorschau mit dynamischen Rahmen
+    return create_preview_image(image, bbox_coords, mode)
+
+
 
 def mock_process_image_upload(image):
     """Mock für Bild-Upload: Setzt BBox in der Mitte"""