File size: 2,678 Bytes
2c87105 cac4307 2c87105 78a6634 2c87105 78a6634 2c87105 78a6634 cac4307 2c87105 78a6634 2c87105 cac4307 2c87105 78a6634 cac4307 fb10459 78a6634 cac4307 78a6634 cac4307 78a6634 2c87105 cac4307 2c87105 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
import gradio as gr
from PIL import Image, ImageDraw, ImageFont
import random
import textwrap
# Enhanced function for image generation with better visual representation
def generate_image(text_description):
# Create a blank image with a gradient background to simulate a sunset
img = Image.new('RGB', (512, 512), color=(0, 0, 0)) # Start with black
draw = ImageDraw.Draw(img)
# Create a simple gradient for sunset (orange to purple)
for y in range(512):
r = int(255 * (1 - y / 512) * 0.8) # Fade red/orange
g = int(100 * (1 - y / 512)) # Minimal green
b = int(150 + 105 * (y / 512)) # Increase blue/purple
draw.line([(0, y), (512, y)], fill=(r, g, b))
# Load a default font (fallback to system font if arial.ttf is unavailable)
try:
font = ImageFont.truetype("arial.ttf", 25)
except:
font = ImageFont.load_default()
# Wrap text to fit within the image
wrapper = textwrap.TextWrapper(width=20) # Adjust width for better wrapping
wrapped_text = wrapper.fill(text=text_description)
lines = wrapped_text.split('\n')
# Draw each line of text at the bottom
y_text = 400
for line in lines:
bbox = draw.textbbox((0, 0), line, font=font)
line_width = bbox[2] - bbox[0]
line_height = bbox[3] - bbox[1]
draw.text((20, y_text), line, font=font, fill=(255, 255, 255))
y_text += line_height + 5
# Add graphical elements based on description
if "lake" in text_description.lower():
draw.rectangle([150, 300, 350, 400], fill=(0, 191, 255)) # Blue lake
if "mountains" in text_description.lower():
draw.polygon([(200, 250), (250, 200), (300, 250)], fill=(255, 255, 255)) # White mountain
draw.polygon([(250, 200), (300, 150), (350, 200)], fill=(255, 255, 255)) # Second mountain
return img
# Gradio interface
with gr.Blocks(title="Text-to-Image Generator") as demo:
gr.Markdown("# Text-to-Image Generator")
gr.Markdown("Enter a description below and generate an image!")
with gr.Row():
with gr.Column():
text_input = gr.Textbox(label="Description", placeholder="Type your image description here...", value="A serene lake surrounded by snow-capped mountains under a vibrant sunset sky with shades of orange and purple.")
generate_btn = gr.Button("Generate Image")
with gr.Column():
output_image = gr.Image(label="Generated Image")
# Connect the button to the function
generate_btn.click(fn=generate_image, inputs=text_input, outputs=output_image)
# Launch the app
demo.launch() |