Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import cv2 | |
| import numpy as np | |
| from PIL import Image, ImageDraw, ImageFont | |
| def combine_images(dress_image, design_image, position, size, text, text_position, font_size, font_color): | |
| # Convert images to numpy arrays | |
| dress = np.array(dress_image) | |
| design = np.array(design_image) | |
| # Convert position and size from strings to tuples of integers | |
| x, y = map(int, position.split(",")) | |
| width, height = map(int, size.split(",")) | |
| tx, ty = map(int, text_position.split(",")) | |
| # Resize design image to fit the size | |
| design_resized = cv2.resize(design, (width, height)) | |
| # Create a mask for the design image | |
| if design_resized.shape[2] == 4: # Handle transparency if PNG | |
| design_mask = design_resized[:, :, 3] > 0 | |
| design_resized = design_resized[:, :, :3] | |
| else: | |
| design_mask = np.ones(design_resized.shape[:2], dtype=bool) | |
| # Define region where the design will be placed | |
| h, w = design_resized.shape[:2] | |
| # Ensure the design fits within the dress image | |
| if x + w > dress.shape[1]: | |
| w = dress.shape[1] - x | |
| design_resized = design_resized[:, :w] | |
| design_mask = design_mask[:, :w] | |
| if y + h > dress.shape[0]: | |
| h = dress.shape[0] - y | |
| design_resized = design_resized[:h] | |
| design_mask = design_mask[:h] | |
| # Place the design on the dress image | |
| for c in range(3): | |
| dress[y:y+h, x:x+w, c] = dress[y:y+h, x:x+w, c] * (1 - design_mask) + design_resized[:, :, c] * design_mask | |
| # Convert to PIL for text addition | |
| final_image = Image.fromarray(dress) | |
| draw = ImageDraw.Draw(final_image) | |
| # Define font (you may need to provide the path to a ttf file or use a basic font) | |
| try: | |
| font = ImageFont.truetype("arial.ttf", int(font_size)) | |
| except IOError: | |
| font = ImageFont.load_default() | |
| # Add text to the image | |
| draw.text((tx, ty), text, font=font, fill=font_color) | |
| return final_image | |
| def interface(): | |
| with gr.Blocks() as demo: | |
| gr.Markdown("## Image Editor with Text Addition") | |
| dress_image = gr.Image(type="pil", label="Upload Dress Image") | |
| design_image = gr.Image(type="pil", label="Upload Design Image") | |
| position = gr.Textbox(placeholder="Enter position as x,y", value="100,100", label="Position") | |
| size = gr.Textbox(placeholder="Enter size as width,height", value="200,200", label="Size") | |
| text = gr.Textbox(placeholder="Enter text", label="Text") | |
| text_position = gr.Textbox(placeholder="Enter text position as x,y", value="50,50", label="Text Position") | |
| font_size = gr.Textbox(placeholder="Enter font size", value="24", label="Font Size") | |
| font_color = gr.Textbox(placeholder="Enter font color (e.g., #FFFFFF)", value="#000000", label="Font Color") | |
| btn = gr.Button("Fit Design") | |
| output = gr.Image(type="pil", label="Final Output") | |
| # Button click function | |
| btn.click( | |
| combine_images, | |
| inputs=[dress_image, design_image, position, size, text, text_position, font_size, font_color], | |
| outputs=output | |
| ) | |
| return demo | |
| demo = interface() | |
| demo.launch() | |