Update app.py

app.py

@@ -3,15 +3,22 @@ from PIL import Image, ImageDraw, ImageFont
 import random
 import textwrap
 
-# Improved function for image generation with better text handling
+# Enhanced function for image generation with better visual representation
 def generate_image(text_description):
-    # Create a blank image with a random background color
-    img = Image.new('RGB', (512, 512), color=(random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)))
+    # 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", 30)
+        font = ImageFont.truetype("arial.ttf", 25)
     except:
         font = ImageFont.load_default()
     
@@ -20,20 +27,21 @@ def generate_image(text_description):
     wrapped_text = wrapper.fill(text=text_description)
     lines = wrapped_text.split('\n')
     
-    # Draw each line of text
-    y_text = 20
+    # Draw each line of text at the bottom
+    y_text = 400
     for line in lines:
-        # Use textbbox to get the bounding box of the text
         bbox = draw.textbbox((0, 0), line, font=font)
-        line_width = bbox[2] - bbox[0]  # right - left
-        line_height = bbox[3] - bbox[1]  # bottom - top
+        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 spacing between lines
+        y_text += line_height + 5
     
-    # Optional: Add a simple graphical element to represent the description
-    if "lake" in text_description.lower() or "mountains" in text_description.lower():
-        draw.rectangle([200, 300, 300, 400], fill=(0, 191, 255))  # Blue rectangle for lake
-        draw.polygon([(250, 250), (300, 200), (350, 250)], fill=(255, 255, 255))  # White triangle for mountain
+    # 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