Update app.py - changes to the UI

app.py

@@ -78,59 +78,99 @@ def add_text_to_image(img, pattern, line1, line2, line3, font_size, color):
     if img is None:
         return None, "Please supply an image first."
 
-    # Ensure editable copy
-    original_mode = img.mode
-    image = img.convert("RGBA")
+    # Create a working copy
+    image = img.convert("RGB")  # Ensure RGB mode
     draw = ImageDraw.Draw(image)
 
-    # Pick a font (robust fallback for different platforms)
-    try:
-        font = ImageFont.truetype("DejaVuSans-Bold.ttf", font_size)
-    except (OSError, IOError):
+    # More robust font loading with better fallbacks
+    font = None
+    font_paths = [
+        "DejaVuSans-Bold.ttf",
+        "arial.ttf",
+        "/System/Library/Fonts/Arial.ttf",
+        "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf",
+        "/usr/share/fonts/TTF/arial.ttf",
+    ]
+    
+    for font_path in font_paths:
         try:
-            # Try common system fonts
-            font = ImageFont.truetype("arial.ttf", font_size)  # Windows
+            font = ImageFont.truetype(font_path, font_size)
+            break
         except (OSError, IOError):
-            try:
-                font = ImageFont.truetype("/System/Library/Fonts/Arial.ttf", font_size)  # macOS
-            except (OSError, IOError):
-                try:
-                    font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", font_size)  # Linux
-                except (OSError, IOError):
-                    font = ImageFont.load_default()
+            continue
+    
+    # Final fallback to default font
+    if font is None:
+        try:
+            font = ImageFont.load_default()
+        except Exception:
+            # Create a basic font if all else fails
+            font = ImageFont.load_default()
 
     w, h = image.size
     positions = []
     
     # Define text positioning patterns
     if pattern == "2-lines-top":
-        positions = [(w / 2, h * 0.10), (w / 2, h * 0.20)]
+        positions = [(w // 2, int(h * 0.10)), (w // 2, int(h * 0.20))]
     elif pattern == "2-lines-bottom":
-        positions = [(w / 2, h * 0.80), (w / 2, h * 0.90)]
+        positions = [(w // 2, int(h * 0.80)), (w // 2, int(h * 0.90))]
     elif pattern == "2-lines-center":
-        positions = [(w / 2, h * 0.45), (w / 2, h * 0.55)]
+        positions = [(w // 2, int(h * 0.45)), (w // 2, int(h * 0.55))]
     elif pattern == "3-lines-center":
-        positions = [(w / 2, h * 0.40), (w / 2, h * 0.50), (w / 2, h * 0.60)]
+        positions = [(w // 2, int(h * 0.40)), (w // 2, int(h * 0.50)), (w // 2, int(h * 0.60))]
     elif pattern == "3-lines-top":
-        positions = [(w / 2, h * 0.10), (w / 2, h * 0.20), (w / 2, h * 0.30)]
+        positions = [(w // 2, int(h * 0.10)), (w // 2, int(h * 0.20)), (w // 2, int(h * 0.30))]
     elif pattern == "3-lines-bottom":
-        positions = [(w / 2, h * 0.70), (w / 2, h * 0.80), (w / 2, h * 0.90)]
+        positions = [(w // 2, int(h * 0.70)), (w // 2, int(h * 0.80)), (w // 2, int(h * 0.90))]
 
-    # Ensure color is in the right format
+    # Convert hex color to RGB tuple if needed
     if isinstance(color, str) and color.startswith('#'):
-        color = color  # PIL handles hex colors fine
-
-    # Draw text lines
+        # Convert hex to RGB
+        color = color.lstrip('#')
+        if len(color) == 6:
+            color = tuple(int(color[i:i+2], 16) for i in (0, 2, 4))
+        else:
+            color = (255, 255, 255)  # Default to white
+    
+    # Draw text with better visibility
     lines = [line1, line2, line3][:len(positions)]
+    text_drawn = False
+    
     for (x, y), txt in zip(positions, lines):
-        if txt and txt.strip():  # Only draw non-empty text
-            draw.text((x, y), txt, fill=color, font=font, anchor="mm")
-
-    # Convert back to original mode if needed
-    if original_mode != "RGBA":
-        image = image.convert(original_mode)
+        if txt and txt.strip():
+            # Add text stroke/outline for better visibility
+            stroke_width = max(1, font_size // 20)
+            
+            # Draw text outline (black)
+            for dx in [-stroke_width, 0, stroke_width]:
+                for dy in [-stroke_width, 0, stroke_width]:
+                    if dx != 0 or dy != 0:
+                        try:
+                            draw.text((x + dx, y + dy), txt, fill=(0, 0, 0), font=font, anchor="mm")
+                        except TypeError:
+                            # Fallback for older PIL versions without anchor
+                            bbox = draw.textbbox((0, 0), txt, font=font)
+                            text_w = bbox[2] - bbox[0]
+                            text_h = bbox[3] - bbox[1]
+                            draw.text((x - text_w//2 + dx, y - text_h//2 + dy), txt, fill=(0, 0, 0), font=font)
+            
+            # Draw main text
+            try:
+                draw.text((x, y), txt, fill=color, font=font, anchor="mm")
+            except TypeError:
+                # Fallback for older PIL versions
+                bbox = draw.textbbox((0, 0), txt, font=font)
+                text_w = bbox[2] - bbox[0]
+                text_h = bbox[3] - bbox[1]
+                draw.text((x - text_w//2, y - text_h//2), txt, fill=color, font=font)
+            
+            text_drawn = True
 
-    return image, "Text added successfully!"
+    if not text_drawn:
+        return img, "No text to add (all lines were empty)"
+    
+    return image, f"Text added successfully! Font size: {font_size}, Color: {color}"
 
 # Create the Gradio Interface
 with gr.Blocks(title="AI Image Generator & Text Overlay") as demo: