Spaces:
Running
Running
| # Import necessary libraries | |
| from transformers import pipeline | |
| from PIL import Image, ImageDraw, ImageFont | |
| from gradio import Interface, Textbox, Radio, Dataframe, File, Label | |
| # Set up Hugging Face pipeline for text-to-image | |
| text_to_image = pipeline('text2image') | |
| # Define UI elements | |
| textarea = Textbox(lines=5, placeholder="Enter your message here...") | |
| radio_button = Radio([("Single", "single"), ("Married", "married"), ("Divorced", "divorced")]) | |
| dataframe = Dataframe(columns=["Item", "Cost"], datatype=["str", "number"]) | |
| file_upload = File(types=["image"]) | |
| label = Label() | |
| # Define the interface | |
| interface = Interface( | |
| fn=generate_image, | |
| inputs=[textarea, radio_button, dataframe, file_upload], | |
| outputs=label, | |
| capture_session=True, | |
| interpretation="default", | |
| examples=[ | |
| ["Hello, world!", "Single", [[("Suit", 5000), ("Laptop", 800), ("Car", 1800)]], None], | |
| ["I love you.", "Married", [[("Suit", 800), ("Watch", 1800), ("Car", 800)]], None], | |
| ], | |
| title="Text-to-Image Generator", | |
| description="Generate an image based on text input", | |
| ) | |
| # Generate image function | |
| def generate_image(textarea, radio_button, dataframe, file_upload): | |
| # Convert text to image file | |
| text_file = textarea.value | |
| font_name = text_file.split(".") | |
| if len(font_name) > 1: | |
| font_name = font_name[1] | |
| else: | |
| font_name = "default" | |
| font = ImageFont.truetype(TEXT_FONT, 20) | |
| line_width = calculate_line_width(textarea.value, font) | |
| line_height = font.getsize(textarea.value)[1] * 2 | |
| text_color = (0, 0, 0) | |
| text_image = Image.new("RGB", (line_width, line_height), (255, 255, 255)) | |
| text_draw = ImageDraw.Draw(text_image) | |
| text_draw.text((0, 0), textarea.value, font=font, fill=text_color) | |
| text_image = text_image.convert("L") | |
| text_image = text_image.point(lambda x: 255 - x) | |
| # Merge text image with user image | |
| try: | |
| background_image = Image.open(file_upload.value) | |
| except Exception: | |
| background_image = None | |
| if background_image is not None: | |
| background_image = background_image.convert("L") | |
| text_image = text_image.resize(background_image.size) | |
| text_image = text_image.convert("RGB") | |
| text_image.paste(background_image, box=(0, 0), mask=text_image) | |
| # Add radio button selection to label | |
| if radio_button.selected_value == "married": | |
| label_text = "Your personalized wedding invitation is almost here!" | |
| elif radio_button.selected_value == "single": | |
| label_text = "This is the image of your special message" | |
| else: | |
| label_text = "You are getting divorced?" | |
| # Add assets to label | |
| label_text += " 😂" | |
| for item in dataframe.value: | |
| cost = item[1] | |
| label_text += f" {item[0]}-\${cost:,}" | |
| # Display the generated image | |
| generated_image.show(label_text, text_image) | |
| # Define calculate_line_width function | |
| def calculate_line_width(text, font): | |
| max_width = 0 | |
| for line in text.split("\n"): | |
| line_width = font.getsize(line)[0] | |
| if line_width > max_width: | |
| max_width = line_width | |
| return max_width | |
| # Run the interface | |
| interface.launch() |