Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import numpy as np
-import io
 import PIL.Image as Image
 from rembg import remove
 from PIL import ImageDraw, ImageFont
@@ -27,16 +26,30 @@ def text_behind_image(input_image, text, text_color, font_size, text_opacity):
         # Get dimensions
         width, height = img.size
         
-        # First, remove background from the original image to get the person silhouette
+        # Step 1: Extract the person from the image
         try:
-            person_img = remove(img, model_name="u2net_human_seg")
+            # Try to use the human segmentation model first
+            person_only = remove(img, model_name="u2net_human_seg")
         except:
             # Fallback to default model if human_seg not available
-            person_img = remove(img)
+            person_only = remove(img)
+            
+        # Step 2: Create a mask from the person cutout
+        person_mask = Image.new('RGBA', (width, height), (0, 0, 0, 0))
+        person_mask.paste(person_only, (0, 0), person_only)
+        
+        # Step 3: Extract the background (original image without the person)
+        # Create inverted mask (where the person is black, background is white)
+        inverted_mask = Image.new('L', (width, height), 255)
+        inverted_mask.paste(0, (0, 0), person_only)
         
-        # Create a new image with white background for the text
-        text_bg = Image.new('RGBA', (width, height), (255, 255, 255, 255))
-        draw = ImageDraw.Draw(text_bg)
+        # Extract just the background
+        background = img.copy()
+        background.putalpha(inverted_mask)
+        
+        # Step 4: Create the text layer
+        text_layer = Image.new('RGBA', (width, height), (0, 0, 0, 0))
+        draw = ImageDraw.Draw(text_layer)
         
         # Prepare the text
         text = text.strip().upper()
@@ -113,9 +126,13 @@ def text_behind_image(input_image, text, text_color, font_size, text_opacity):
                 # Draw the text with opacity
                 draw.text((x, y), line, font=font, fill=text_color_rgb + (int(text_opacity * 255),))
         
-        # Now combine the text background with the person
-        # The text goes behind the person, so we composite person on top of text
-        final_image = Image.alpha_composite(text_bg, person_img)
+        # Step 5: Composite all layers together
+        # First, composite the text on top of the background
+        # We use the inverted mask so text only shows where the person isn't
+        background_with_text = Image.alpha_composite(background, text_layer)
+        
+        # Then, composite the person on top
+        final_image = Image.alpha_composite(background_with_text, person_mask)
         
         # Convert to RGB for display
         return np.array(final_image.convert('RGB'))