Update app.py

app.py

@@ -25,7 +25,7 @@ from PIL import Image, ImageDraw, ImageFont
 from PIL import Image
 from gradio_client import Client, handle_file
 import uuid
-
+import traceback  # 상단에 추가
 
 def clear_memory():
     """메모리 정리 함수"""
@@ -568,6 +568,8 @@ def superimpose(image_with_text, overlay_image):
     # image_with_text.save("output_image.png")
     return image_with_text
 
+
+
 def add_text_to_image(
     input_image,
     text,
@@ -599,6 +601,10 @@ def add_text_to_image(
         # 이미지를 RGBA 모드로 변환
         image = image.convert('RGBA')
         
+        # 텍스트 레이어 생성
+        txt_overlay = Image.new('RGBA', image.size, (255, 255, 255, 0))
+        draw = ImageDraw.Draw(txt_overlay)
+
         # 폰트 설정
         try:
             font_path = {
@@ -635,31 +641,24 @@ def add_text_to_image(
         text_color = (*rgb_color, int(opacity))
 
         if text_position_type == "Text Behind Image":
-            # 텍스트 레이어 생성
-            text_layer = Image.new('RGBA', image.size, (255, 255, 255, 0))
-            text_draw = ImageDraw.Draw(text_layer)
+            # 배경 레이어 생성
+            background = Image.new('RGBA', image.size, (255, 255, 255, 255))
             
             # 텍스트 그리기
             for dx in range(-thickness, thickness + 1):
                 for dy in range(-thickness, thickness + 1):
-                    text_draw.text(
+                    draw.text(
                         (actual_x + dx, actual_y + dy),
                         text,
                         font=font,
                         fill=text_color
                     )
             
-            # 최종 이미지 합성
-            output_image = Image.new('RGBA', image.size, (255, 255, 255, 255))  # 흰색 배경
-            output_image = Image.alpha_composite(output_image, text_layer)  # 텍스트 추가
-            output_image = Image.alpha_composite(output_image, image)  # 원본 이미지 추가
-            
+            # 레이어 합성
+            output_image = Image.alpha_composite(background, txt_overlay)
+            output_image = Image.alpha_composite(output_image, image)
         else:
             # Text Over Image
-            txt_overlay = Image.new('RGBA', image.size, (255, 255, 255, 0))
-            draw = ImageDraw.Draw(txt_overlay)
-            
-            # 텍스트 그리기
             for dx in range(-thickness, thickness + 1):
                 for dy in range(-thickness, thickness + 1):
                     draw.text(
@@ -668,7 +667,6 @@ def add_text_to_image(
                         font=font,
                         fill=text_color
                     )
-            
             output_image = Image.alpha_composite(image, txt_overlay)
 
         # RGB로 변환하여 반환
@@ -676,7 +674,7 @@ def add_text_to_image(
 
     except Exception as e:
         print(f"Text addition error: {str(e)}")
-        traceback.print_exc()  # 상세한 에러 정보 출력
+        traceback.print_exc()
         return input_image