Update app.py

app.py

@@ -529,77 +529,84 @@ def add_text_to_image(
     """
     Add text to an image with customizable properties
     """
-    # Convert gradio image (numpy array) to PIL Image
-    if input_image is None:
-        return None
-    
-    image = Image.fromarray(input_image)
-    # remove background 
-    overlay_image = remove_background(image)
-
-    # Create a transparent overlay for the text
-    txt_overlay = Image.new('RGBA', image.size, (255, 255, 255, 0))
-    draw = ImageDraw.Draw(txt_overlay)
-    
-    # Create a font with specified size
     try:
-        font = ImageFont.truetype("DejaVuSans.ttf", int(font_size))
-    except:
-        # If DejaVu font is not found, try to use Arial or default
+        # 입력 이미지 처리 수정
+        if input_image is None:
+            return None
+            
+        # PIL Image 객체로 변환 (이미 PIL Image인 경우 그대로 사용)
+        if not isinstance(input_image, Image.Image):
+            if isinstance(input_image, np.ndarray):
+                image = Image.fromarray(input_image)
+            else:
+                raise ValueError("Unsupported image type")
+        else:
+            image = input_image.copy()
+
+        # 이미지를 RGBA 모드로 변환
+        if image.mode != 'RGBA':
+            image = image.convert('RGBA')
+
+        # Create a transparent overlay for the text
+        txt_overlay = Image.new('RGBA', image.size, (255, 255, 255, 0))
+        draw = ImageDraw.Draw(txt_overlay)
+        
+        # Create a font with specified size
         try:
-            font = ImageFont.truetype("arial.ttf", int(font_size))
+            font = ImageFont.truetype("DejaVuSans.ttf", int(font_size))
         except:
-            print("Using default font as system fonts not found")
-            font = ImageFont.load_default()
-    
-    # Convert color name to RGB
-    color_map = {
-        'White': (255, 255, 255),
-        'Black': (0, 0, 0),
-        'Red': (255, 0, 0),
-        'Green': (0, 255, 0),
-        'Blue': (0, 0, 255),
-        'Yellow': (255, 255, 0),
-        'Purple': (128, 0, 128)
-    }
-    rgb_color = color_map.get(color, (255, 255, 255))
-    
-    # Get text size for positioning
-    text_bbox = draw.textbbox((0, 0), text, font=font)
-    text_width = text_bbox[2] - text_bbox[0]
-    text_height = text_bbox[3] - text_bbox[1]
-
-    # Calculate actual x and y positions based on percentages
-    actual_x = int((image.width - text_width) * (x_position / 100))
-    actual_y = int((image.height - text_height) * (y_position / 100))
+            try:
+                font = ImageFont.truetype("arial.ttf", int(font_size))
+            except:
+                print("Using default font")
+                font = ImageFont.load_default()
+        
+        # Convert color name to RGB
+        color_map = {
+            'White': (255, 255, 255),
+            'Black': (0, 0, 0),
+            'Red': (255, 0, 0),
+            'Green': (0, 255, 0),
+            'Blue': (0, 0, 255),
+            'Yellow': (255, 255, 0),
+            'Purple': (128, 0, 128)
+        }
+        rgb_color = color_map.get(color, (255, 255, 255))
+        
+        # Get text size for positioning
+        text_bbox = draw.textbbox((0, 0), text, font=font)
+        text_width = text_bbox[2] - text_bbox[0]
+        text_height = text_bbox[3] - text_bbox[1]
 
-    # Create final color with opacity
-    text_color = (*rgb_color, int(opacity))
-    
-    # Draw the text with stroke for thickness
-    add_text_with_stroke(
-        draw, 
-        text, 
-        actual_x, 
-        actual_y, 
-        font, 
-        text_color, 
-        int(thickness)
-    )
+        # Calculate actual positions
+        actual_x = int((image.width - text_width) * (x_position / 100))
+        actual_y = int((image.height - text_height) * (y_position / 100))
 
-    # Combine the original image with the text overlay
-    if image.mode != 'RGBA':
-        image = image.convert('RGBA')
-    output_image = Image.alpha_composite(image, txt_overlay)
-    
-    # Convert back to RGB for display
-    output_image = output_image.convert('RGB')
+        # Create final color with opacity
+        text_color = (*rgb_color, int(opacity))
+        
+        # Draw text with stroke
+        add_text_with_stroke(
+            draw, 
+            text, 
+            actual_x, 
+            actual_y, 
+            font, 
+            text_color, 
+            int(thickness)
+        )
+
+        # Combine the original image with the text overlay
+        output_image = Image.alpha_composite(image, txt_overlay)
+        
+        # Convert back to RGB for display
+        output_image = output_image.convert('RGB')
+        
+        return output_image
 
-    # superimpose images
-    output_image = superimpose(output_image, overlay_image)
-    
-    # Convert PIL image back to numpy array for Gradio
-    return np.array(output_image)
+    except Exception as e:
+        print(f"Error in add_text_to_image: {str(e)}")
+        return input_image  # 에러 발생 시 원본 이미지 반환
     
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""