Update app-backup.py

app-backup.py

@@ -529,79 +529,85 @@ 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  # 에러 발생 시 원본 이미지 반환
     
-# UI 구성
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">
@@ -669,73 +675,73 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             )
 
         with gr.Column(scale=1):
-            with gr.Row():
+            with gr.Tab("Result"):
                 combined_image = gr.Image(
                     label="Combined Result",
                     show_download_button=True,
                     type="pil",
                     height=512
                 )
-            
-            # 텍스트 삽입 컨트롤 섹션 추가
-            with gr.Accordion("Add Text to Image", open=False):
-                text_input = gr.Textbox(
-                    label="Text to Add",
-                    placeholder="Enter text to overlay on image..."
-                )
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        font_size = gr.Slider(
-                            minimum=10,
-                            maximum=200,
-                            value=40,
-                            step=5,
-                            label="Font Size"
-                        )
-                        color_dropdown = gr.Dropdown(
-                            choices=["White", "Black", "Red", "Green", "Blue", "Yellow", "Purple"],
-                            value="White",
-                            label="Text Color"
-                        )
-                        thickness = gr.Slider(
-                            minimum=0,
-                            maximum=10,
-                            value=1,
-                            step=1,
-                            label="Text Thickness"
-                        )
-                    with gr.Column(scale=1):
-                        opacity_slider = gr.Slider(
-                            minimum=0,
-                            maximum=255,
-                            value=255,
-                            step=1,
-                            label="Opacity"
-                        )
-                        x_position = gr.Slider(
-                            minimum=0,
-                            maximum=100,
-                            value=50,
-                            step=1,
-                            label="X Position (%)"
-                        )
-                        y_position = gr.Slider(
-                            minimum=0,
-                            maximum=100,
-                            value=50,
-                            step=1,
-                            label="Y Position (%)"
-                        )
                 
-                add_text_btn = gr.Button("Apply Text", variant="secondary")
-            
-            with gr.Row():
-                extracted_image = gr.Image(
-                    label="Extracted Object",
-                    show_download_button=True,
-                    type="pil",
-                    height=256
-                )
+                # 텍스트 삽입 컨트롤을 더 명확하게 구분
+                with gr.Group():
+                    gr.Markdown("### Add Text to Image")
+                    text_input = gr.Textbox(
+                        label="Text Content",
+                        placeholder="Enter text to add to image..."
+                    )
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            font_size = gr.Slider(
+                                minimum=10,
+                                maximum=200,
+                                value=40,
+                                step=5,
+                                label="Font Size"
+                            )
+                            color_dropdown = gr.Dropdown(
+                                choices=["White", "Black", "Red", "Green", "Blue", "Yellow", "Purple"],
+                                value="White",
+                                label="Text Color"
+                            )
+                            thickness = gr.Slider(
+                                minimum=0,
+                                maximum=10,
+                                value=1,
+                                step=1,
+                                label="Text Thickness"
+                            )
+                        with gr.Column(scale=1):
+                            opacity_slider = gr.Slider(
+                                minimum=0,
+                                maximum=255,
+                                value=255,
+                                step=1,
+                                label="Opacity"
+                            )
+                            x_position = gr.Slider(
+                                minimum=0,
+                                maximum=100,
+                                value=50,
+                                step=1,
+                                label="X Position (%)"
+                            )
+                            y_position = gr.Slider(
+                                minimum=0,
+                                maximum=100,
+                                value=50,
+                                step=1,
+                                label="Y Position (%)"
+                            )
+                    add_text_btn = gr.Button("Apply Text", variant="primary")
+
+                with gr.Row():
+                    extracted_image = gr.Image(
+                        label="Extracted Object",
+                        show_download_button=True,
+                        type="pil",
+                        height=256
+                    )
 
     # 각 버튼에 대한 클릭 이벤트 처리
     def update_position(new_position):