Spaces:
Build error
Build error
| import gradio as gr | |
| import numpy as np | |
| import io | |
| import PIL.Image as Image | |
| from rembg import remove | |
| from PIL import ImageDraw, ImageFont | |
| import textwrap | |
| # Function to remove background and place text behind person | |
| def text_behind_image(input_image, text, text_color, font_size, text_opacity): | |
| if input_image is None: | |
| return None | |
| # Convert color hex to RGB | |
| try: | |
| r = int(text_color[1:3], 16) | |
| g = int(text_color[3:5], 16) | |
| b = int(text_color[5:7], 16) | |
| text_color_rgb = (r, g, b) | |
| except: | |
| text_color_rgb = (255, 255, 255) # Default to white | |
| try: | |
| # Open the image | |
| img = Image.fromarray(input_image) | |
| # Get dimensions | |
| width, height = img.size | |
| # Create a new image with white background (this will be the canvas) | |
| background = Image.new('RGBA', (width, height), (255, 255, 255, 255)) | |
| # Create a drawing context for the text | |
| draw = ImageDraw.Draw(background) | |
| # Determine font size (relative to image size) | |
| font_size = int(min(width, height) * (font_size / 100)) # Convert percentage to actual size | |
| # Try to load a nice font, fall back to default if not available | |
| try: | |
| font = ImageFont.truetype("arial.ttf", font_size) | |
| except: | |
| font = ImageFont.load_default() | |
| # Word wrap the text to fit within the image width | |
| margin = 20 | |
| wrapper = textwrap.TextWrapper(width=int(width/font_size*2) - 2*margin) | |
| word_list = wrapper.wrap(text) | |
| # Calculate text block height | |
| text_height = len(word_list) * (font_size + 10) | |
| # Position text in the center | |
| y_text = (height - text_height) // 2 | |
| # Draw text with specified opacity | |
| for line in word_list: | |
| # Get line width | |
| line_width = font.getbbox(line)[2] - font.getbbox(line)[0] | |
| # Center the line | |
| x_text = (width - line_width) // 2 | |
| # Draw text with specified opacity | |
| text_color_with_opacity = text_color_rgb + (int(text_opacity * 255),) | |
| draw.text((x_text, y_text), line, font=font, fill=text_color_with_opacity) | |
| y_text += font_size + 10 | |
| # Remove background from the original image to get the person silhouette | |
| person_img = remove(img) | |
| # Composite the images: text in background, then person on top | |
| final_image = Image.alpha_composite(background.convert('RGBA'), person_img.convert('RGBA')) | |
| # Convert back to RGB for display | |
| return np.array(final_image.convert('RGB')) | |
| except Exception as e: | |
| print(f"Error processing image: {e}") | |
| return input_image # Return original image on error | |
| # Create Gradio interface | |
| with gr.Blocks(title="Text Behind Image") as demo: | |
| gr.Markdown("# Text Behind Image") | |
| gr.Markdown("Upload an image with a person and add text behind them") | |
| with gr.Row(): | |
| with gr.Column(): | |
| input_image = gr.Image(label="Upload Image", type="numpy") | |
| text_input = gr.Textbox(label="Text to place behind", placeholder="Enter text here...") | |
| with gr.Row(): | |
| text_color = gr.ColorPicker(label="Text Color", value="#FFFFFF") | |
| font_size = gr.Slider(label="Font Size (%)", minimum=1, maximum=30, value=10, step=1) | |
| text_opacity = gr.Slider(label="Text Opacity", minimum=0.1, maximum=1.0, value=0.8, step=0.1) | |
| submit_btn = gr.Button("Generate", variant="primary") | |
| with gr.Column(): | |
| output_image = gr.Image(label="Result", type="numpy") | |
| submit_btn.click( | |
| fn=text_behind_image, | |
| inputs=[input_image, text_input, text_color, font_size, text_opacity], | |
| outputs=output_image | |
| ) | |
| gr.Markdown("## How it works") | |
| gr.Markdown("1. Upload an image with a person") | |
| gr.Markdown("2. Enter the text you want to place behind the person") | |
| gr.Markdown("3. Customize text color, size, and opacity") | |
| gr.Markdown("4. Click 'Generate' to create your image") | |
| # Launch the app | |
| demo.launch() | |