Update app.py

app.py

@@ -581,9 +581,9 @@ def add_text_to_image(
     font_choice
 ):
     try:
-        if input_image is None:
-            return None
-            
+        if input_image is None or text.strip() == "":
+            return input_image
+
         # PIL Image 객체로 변환
         if not isinstance(input_image, Image.Image):
             if isinstance(input_image, np.ndarray):
@@ -622,8 +622,9 @@ def add_text_to_image(
         }
         rgb_color = color_map.get(color, (255, 255, 255))
         
-        # 텍스트 크기 계산
-        text_bbox = draw.textbbox((0, 0), text, font=font)
+        # 임시 Draw 객체 생성하여 텍스트 크기 계산
+        temp_draw = ImageDraw.Draw(image)
+        text_bbox = temp_draw.textbbox((0, 0), text, font=font)
         text_width = text_bbox[2] - text_bbox[0]
         text_height = text_bbox[3] - text_bbox[1]
 
@@ -635,32 +636,36 @@ def add_text_to_image(
         text_color = (*rgb_color, int(opacity))
         
         if text_position_type == "Text Behind Image":
-            # 원본 이미지에서 전경 객체만 추출
-            foreground = remove_background(image)
-            
-            # 배경 이미지 생성 (원본 이미지 복사)
-            background = image.copy()
-            
-            # 텍스트를 그릴 임시 레이어 생성
-            text_layer = Image.new('RGBA', image.size, (255, 255, 255, 0))
-            draw_text = ImageDraw.Draw(text_layer)
-            
-            # 텍스트 그리기
-            add_text_with_stroke(
-                draw_text,
-                text,
-                actual_x,
-                actual_y,
-                font,
-                text_color,
-                int(thickness)
-            )
-            
-            # 배경에 텍스트 합성
-            background = Image.alpha_composite(background, text_layer)
-            
-            # 텍스트가 있는 배경 위에 전경 객체 합성
-            output_image = Image.alpha_composite(background, foreground)
+            try:
+                # 원본 이미지에서 전경 객체만 추출
+                foreground = remove_background(image)
+                
+                # 배경 이미지 생성 (원본 이미지 복사)
+                background = image.copy()
+                
+                # 텍스트를 그릴 임시 레이어 생성
+                text_layer = Image.new('RGBA', image.size, (255, 255, 255, 0))
+                draw_text = ImageDraw.Draw(text_layer)
+                
+                # 텍스트 그리기
+                add_text_with_stroke(
+                    draw_text,
+                    text,
+                    actual_x,
+                    actual_y,
+                    font,
+                    text_color,
+                    int(thickness)
+                )
+                
+                # 배경에 텍스트 합성
+                background = Image.alpha_composite(background, text_layer)
+                
+                # 텍스트가 있는 배경 위에 전경 객체 합성
+                output_image = Image.alpha_composite(background, foreground)
+            except Exception as e:
+                print(f"Error in Text Behind Image processing: {str(e)}")
+                return input_image
         else:
             # 텍스트 오버레이 생성
             txt_overlay = Image.new('RGBA', image.size, (255, 255, 255, 0))
@@ -925,11 +930,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         queue=True
     )
 
+# 이벤트 바인딩 부분에서
     add_text_btn.click(
         fn=add_text_to_image,
         inputs=[
-            combined_image,
-            text_input,
+            combined_image,  # 첫 번째 인자로 이미지
+            text_input,      # 두 번째 인자로 텍스트
             font_size,
             color_dropdown,
             opacity_slider,
@@ -939,8 +945,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             text_position_type,
             font_choice
         ],
-        outputs=combined_image
+        outputs=combined_image,
+        api_name="add_text"  # API 이름 추가
     )
+    
+
 
 demo.queue(max_size=5)
 demo.launch(