Update app.py

app.py

@@ -56,6 +56,7 @@ prompt = "high quality"
 
 
 @spaces.GPU
+"""
 def fill_image(image, model_selection):
 
     margin = 256
@@ -100,7 +101,56 @@ def fill_image(image, model_selection):
     cnet_image.paste(image, (0, 0), mask)
 
     yield background, cnet_image
+"""
+
+def fill_image(image, model_selection):
+    source = image
+    target_ratio=(9, 16)
+    overlap=24
+    # Calculate the target width based on the 9:16 ratio
+    target_width = (source.height * target_ratio[0]) // target_ratio[1]
+    
+    # Calculate margins
+    margin_x = max(0, (target_width - source.width) // 2)
+    margin_y = 0  # No vertical expansion
+    
+    # Calculate new output size
+    output_size = (source.width + 2*margin_x, source.height + 2*margin_y)
+    
+    # Create a white background
+    background = Image.new('RGB', output_size, (255, 255, 255))
+    
+    # Calculate position to paste the original image
+    position = (margin_x, margin_y)
+    
+    # Paste the original image onto the white background
+    background.paste(source, position)
+    
+    # Create the mask
+    mask = Image.new('L', output_size, 255)  # Start with all white
+    mask_draw = ImageDraw.Draw(mask)
+    mask_draw.rectangle([
+        (position[0] + overlap, position[1] + overlap),
+        (position[0] + source.width - overlap, position[1] + source.height - overlap)
+    ], fill=0)
+    
+    # Prepare the image for ControlNet
+    cnet_image = background.copy()
+    cnet_image.paste(0, (0, 0), mask)
+
+    for image in pipe(
+        prompt_embeds=prompt_embeds,
+        negative_prompt_embeds=negative_prompt_embeds,
+        pooled_prompt_embeds=pooled_prompt_embeds,
+        negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
+        image=cnet_image,
+    ):
+        yield image, cnet_image
 
+    image = image.convert("RGBA")
+    cnet_image.paste(image, (0, 0), mask)
+
+    yield background, cnet_image
 
 def clear_result():
     return gr.update(value=None)