Update app.py

app.py

@@ -3,10 +3,8 @@ import requests
 from PIL import Image, ImageDraw, ImageFont
 import textwrap
 import os
-import time
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
-# Function to download image based on prompt
 def download_image(prompt, index):
     try:
         url = f"https://pollinations.ai/p/{prompt}"
@@ -20,26 +18,51 @@ def download_image(prompt, index):
         print(f"Error downloading image {index}: {e}")
         return None
 
-# Function to add text to image
 def add_text_to_image(image_path, text):
     try:
         img = Image.open(image_path)
         draw = ImageDraw.Draw(img)
-        # Use a default font if arial.ttf is not available
+        
+        # Use a default font if custom font is not available
         try:
-            font = ImageFont.truetype("arial.ttf", 20)
+            font = ImageFont.truetype("arial.ttf", 30)
         except IOError:
-            font = ImageFont.load_default()
-        lines = textwrap.fill(text, width=40)
-        draw.text((10, 10), lines, font=font, fill="white", stroke_width=2, stroke_fill="black")
+            font = ImageFont.load_default().font_variant(size=30)
+        
+        # Calculate text size and position
+        img_width, img_height = img.size
+        max_text_width = int(img_width * 0.9)  # 90% of image width
+        lines = textwrap.wrap(text, width=40)
+        line_height = font.getsize('hg')[1] + 5  # Add some padding
+        text_height = line_height * len(lines)
+        
+        # Create semi-transparent background
+        overlay = Image.new('RGBA', img.size, (0, 0, 0, 0))
+        overlay_draw = ImageDraw.Draw(overlay)
+        overlay_draw.rectangle([(0, img_height - text_height - 20), (img_width, img_height)], 
+                               fill=(0, 0, 0, 180))
+        
+        # Paste the overlay onto the original image
+        img = img.convert('RGBA')
+        img = Image.alpha_composite(img, overlay)
+        draw = ImageDraw.Draw(img)
+        
+        # Add text
+        y_text = img_height - text_height - 10
+        for line in lines:
+            line_width, _ = draw.textsize(line, font=font)
+            x_text = (img_width - line_width) / 2
+            draw.text((x_text, y_text), line, font=font, fill=(255, 255, 255, 255))
+            y_text += line_height
+        
         img_with_text = f'text_added_{os.path.basename(image_path)}'
+        img = img.convert('RGB')
         img.save(img_with_text)
         return img_with_text
     except Exception as e:
         print(f"Error adding text to image: {e}")
         return None
 
-# Function to visualize each line of the story
 def visualize_story_lines(story, progress=gr.Progress()):
     lines = [line.strip() for line in story.split('\n') if line.strip()]
     images_with_text = []
@@ -61,13 +84,11 @@ def visualize_story_lines(story, progress=gr.Progress()):
     
     return [img for _, img in sorted(images_with_text)]
 
-# Gradio interface with state management
 def visualize_story_images(story):
     if not story.strip():
         return []
     return visualize_story_lines(story)
 
-# Create a Gradio interface
 with gr.Blocks() as iface:
     gr.Markdown("# Story Visualizer")