Upload folder using huggingface_hub

image_styler_v2.py

@@ -87,14 +87,28 @@ def enhance_unified_image(harmonized_img, enhancer):
     
     return enhanced_img
 
-def embed_person_on_background(person_img, background_img):
-    # Preserve the aspect ratio and resize the person image to fit within the background
-    person_img = ImageOps.contain(person_img, background_img.size, method=Image.LANCZOS)
+def embed_person_on_background(person_img, background_img, position_x, position_y, scale):
+    # Scale the person image while keeping proportions
+    person_width, person_height = person_img.size
+    new_width = int(person_width * scale)
+    new_height = int(person_height * scale)
+    person_img = person_img.resize((new_width, new_height), Image.LANCZOS)
+    
+    # Calculate the position based on bottom-center transformation point
+    background_width, background_height = background_img.size
+    
+    # Default position: bottom-center of the background
+    default_x = (background_width - new_width) // 2
+    default_y = background_height - new_height
+
+    # Adjust the position based on sliders
+    position_x = default_x + int(position_x)
+    position_y = default_y + int(position_y)
 
     # Create a new image with the same size as the background and paste the person image onto it
     combined_img = Image.new("RGBA", background_img.size)
     combined_img.paste(background_img, (0, 0))
-    combined_img.paste(person_img, (0, 0), person_img)
+    combined_img.paste(person_img, (position_x, position_y), person_img)
 
     return combined_img
 
@@ -131,11 +145,7 @@ def enhance_image(image, contrast, brightness, color):
         enhanced_image = enhancer.enhance(factor)
     return enhanced_image
 
-def process_images(person_img, background_img, num_images, enhance, auto_match, contrast, brightness, color, unify):
-    # Validate parameters
-    if not (1 <= num_images <= 5):
-        raise ValueError("Number of Output Images must be between 1 and 5")
-    
+def process_images(person_img, background_img, enhance, auto_match, contrast, brightness, color, unify, position_x, position_y, scale):
     # Remove background from the person image
     person_no_bg = remove(person_img)
     
@@ -146,7 +156,7 @@ def process_images(person_img, background_img, num_images, enhance, auto_match,
         print(f"Applying enhancement with contrast={contrast}, brightness={brightness}, color={color}...")
         person_no_bg = enhance_image(person_no_bg, contrast, brightness, color)
     
-    combined_img = embed_person_on_background(person_no_bg, background_img)
+    combined_img = embed_person_on_background(person_no_bg, background_img, position_x, position_y, scale)
 
     if unify:
         print("Unifying image with AI...")
@@ -155,16 +165,14 @@ def process_images(person_img, background_img, num_images, enhance, auto_match,
         enhancer = load_enhancement_model('pretrained/enhancer.pth')
         combined_img = enhance_unified_image(combined_img, enhancer)
     
-    outputs = [combined_img] * num_images
-    
-    return outputs
+    return combined_img
 
-def gradio_interface(person_img, background_img, num_images, enhance, auto_match, contrast, brightness, color, unify):
+def gradio_interface(person_img, background_img, enhance, auto_match, contrast, brightness, color, unify, position_x, position_y, scale):
     try:
-        results = process_images(person_img, background_img, num_images, enhance, auto_match, contrast, brightness, color, unify)
-        return results + [None] * (5 - len(results))  # Ensure the number of returned images matches the expected output
+        result = process_images(person_img, background_img, enhance, auto_match, contrast, brightness, color, unify, position_x, position_y, scale)
+        return result
     except Exception as e:
-        return [str(e)] + [None] * 4
+        return str(e)
 
 def update_enhancement_controls(auto_match):
     # Disable enhancement sliders if auto-match is checked
@@ -175,11 +183,10 @@ def update_enhancement_controls(auto_match):
     }
 
 # Create Gradio interface
-with gr.Blocks() as interface:
+with gr.Blocks(css='#output_image {max-width: 800px !important; width: auto !important; height: auto !important;}') as interface:
     with gr.Row():
         person_img = gr.Image(type="pil", label="Upload Person Image")
         background_img = gr.Image(type="pil", label="Upload Background Image")
-    num_images = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Number of Output Images")
     enhance = gr.Checkbox(label="Enhance Image", value=False)
     auto_match = gr.Checkbox(label="Auto-Match Enhancers", value=False)
     contrast_slider = gr.Slider(minimum=0.5, maximum=3.0, step=0.1, value=1.0, label="Contrast")
@@ -189,14 +196,17 @@ with gr.Blocks() as interface:
     auto_match.change(fn=update_enhancement_controls, inputs=auto_match, outputs=[contrast_slider, brightness_slider, color_slider])
     
     unify = gr.Checkbox(label="Unify Image with AI", value=True)
+    position_x = gr.Slider(minimum=-500, maximum=500, step=1, value=0, label="Horizontal Position (pixels)")
+    position_y = gr.Slider(minimum=-500, maximum=500, step=1, value=0, label="Vertical Position (pixels)")
+    scale = gr.Slider(minimum=0.1, maximum=3.0, step=0.1, value=1.0, label="Scale")
 
-    outputs = [gr.Image(type="pil", label="Generated Image") for _ in range(5)]
+    output = gr.Image(type="pil", label="Generated Image", elem_id="output_image")
     run_button = gr.Button("Run")
     
     run_button.click(
         fn=gradio_interface,
-        inputs=[person_img, background_img, num_images, enhance, auto_match, contrast_slider, brightness_slider, color_slider, unify],
-        outputs=outputs
+        inputs=[person_img, background_img, enhance, auto_match, contrast_slider, brightness_slider, color_slider, unify, position_x, position_y, scale],
+        outputs=output
     )
 
 if __name__ == "__main__":