Update app.py

app.py

@@ -19,11 +19,13 @@ def detect_dress_and_warp(hoodie_img, design_img, warp_strength, scale):
     if contours:
         largest_contour = max(contours, key=cv2.contourArea)
         x, y, w, h = cv2.boundingRect(largest_contour)
+        center_x, center_y = x + w // 2, y + h // 2
     else:
-        x, y, w, h = 100, 200, hoodie.size[0]//2, hoodie.size[1]//2  # Default placement
+        center_x, center_y = hoodie.size[0] // 2, hoodie.size[1] // 2  # Default center placement
     
     # Resize and warp design
-    design = design.resize((int(w * scale), int(h * scale)))
+    new_w, new_h = int(w * scale), int(h * scale)
+    design = design.resize((new_w, new_h))
     design_cv = cv2.cvtColor(np.array(design), cv2.COLOR_RGBA2BGRA)
     
     # Apply warp effect
@@ -32,11 +34,15 @@ def detect_dress_and_warp(hoodie_img, design_img, warp_strength, scale):
     design_cv = cv2.warpAffine(design_cv, warp_map, (cols, rows))
     design = Image.fromarray(cv2.cvtColor(design_cv, cv2.COLOR_BGRA2RGBA))
     
+    # Calculate placement at the center of the detected dress
+    paste_x = center_x - new_w // 2
+    paste_y = center_y - new_h // 2
+    
     # Create a blank image with same size as hoodie
     temp_layer = Image.new("RGBA", hoodie.size, (0, 0, 0, 0))
     
-    # Paste design onto detected dress region
-    temp_layer.paste(design, (x, y), design)
+    # Paste design onto detected dress center
+    temp_layer.paste(design, (paste_x, paste_y), design)
     
     # Blend the layers
     final_image = Image.alpha_composite(hoodie, temp_layer)
@@ -56,7 +62,7 @@ demo = gr.Interface(
     ],
     outputs=gr.Image(type="pil", label="Generated Mockup"),
     title="Rookus Hoodie Mockup Generator",
-    description="Upload a hoodie mockup and your design to generate a realistic preview with automatic dress detection and warping."
+    description="Upload a hoodie mockup and your design to generate a realistic preview with automatic dress detection and warping at the center of the dress."
 )
 
-demo.launch()
+demo.launch()