Update app.py

app.py

@@ -111,53 +111,32 @@ def fill_image(image, model_selection):
  target_height=1280
  overlap=48
  fade_width=24
- # Calculate target dimensions
- target_width = (target_height * target_ratio[0]) // target_ratio[1]
+ # Calculate the required height for 9:16 ratio
+ target_height = (source.width * target_ratio[1]) // target_ratio[0]
  
- # Resize the source image to fit the target width
- new_width = target_width
- new_height = int(source.height * (new_width / source.width))
- resized_source = source.resize((new_width, new_height), Image.LANCZOS)
+ # Calculate margins (only top and bottom)
+ margin_y = (target_height - source.height) // 2
+ 
+ # Calculate new output size
+ output_size = (source.width, target_height)
  
  # Create a white background
- background = Image.new('RGB', (target_width, target_height), (255, 255, 255))
+ background = Image.new('RGB', output_size, (255, 255, 255))
  
- # Calculate position to paste the resized image (centered vertically)
- margin_y = (target_height - new_height) // 2
+ # Calculate position to paste the original image
  position = (0, margin_y)
  
- # Paste the resized image onto the white background
- background.paste(resized_source, position)
+ # Paste the original image onto the white background
+ background.paste(source, position)
  
  # Create the mask
- mask = Image.new('L', (target_width, target_height), 255) # Start with all white
+ mask = Image.new('L', output_size, 255) # Start with all white
  mask_draw = ImageDraw.Draw(mask)
- 
- # Draw black rectangle for the resized image area (with overlap)
  mask_draw.rectangle([
- (-overlap, margin_y - overlap),
- (new_width + overlap, margin_y + new_height + overlap)
+ (overlap, margin_y + overlap),
+ (source.width - overlap, margin_y + source.height - overlap)
  ], fill=0)
  
- # Convert mask to numpy array for gradient creation
- mask_array = np.array(mask)
- 
- # Create gradient on the edges that overlap with the image
- for i in range(fade_width):
- alpha = i / fade_width
- # Top edge
- if margin_y - overlap + i < margin_y:
- mask_array[margin_y - overlap + i, :new_width] = int(255 * alpha)
- # Bottom edge
- if margin_y + new_height + overlap - i - 1 >= margin_y + new_height:
- mask_array[margin_y + new_height + overlap - i - 1, :new_width] = int(255 * alpha)
- # Left edge
- mask_array[margin_y:margin_y+new_height, i] = int(255 * alpha)
- # Right edge
- mask_array[margin_y:margin_y+new_height, new_width - i - 1] = int(255 * alpha)
- 
- mask = Image.fromarray(mask_array.astype('uint8'), 'L')
- 
  # Prepare the image for ControlNet
  cnet_image = background.copy()
  cnet_image.paste(0, (0, 0), mask)