Update app.py

app.py

@@ -7,10 +7,11 @@ from PIL import Image, ImageDraw
 from math import cos, sin, radians
 
 # 테두리 장식 함수
+# create_border_decoration 함수 수정
 def create_border_decoration(qr_image, decoration_prompt="flowers"):
-    # Convert QR image to RGB if it's not
-    if qr_image.mode != 'RGB':
-        qr_image = qr_image.convert('RGB')
+    # Convert QR image to RGBA if it's not
+    if qr_image.mode in ('1', 'L'):
+        qr_image = qr_image.convert('RGBA')
     
     # Get the size of the QR code image
     width, height = qr_image.size
@@ -21,10 +22,10 @@ def create_border_decoration(qr_image, decoration_prompt="flowers"):
     new_height = height + (padding * 2)
     
     # Create new image with white background
-    decorated_image = Image.new('RGB', (new_width, new_height), 'white')
+    decorated_image = Image.new('RGBA', (new_width, new_height), (255, 255, 255, 255))
     
-    # Paste QR code in center - now with explicit size
-    decorated_image.paste(qr_image, (padding, padding, padding + width, padding + height))
+    # Paste QR code in center - now with explicit size and convert to RGBA
+    decorated_image.paste(qr_image, (padding, padding, padding + width, padding + height), qr_image)
     
     # Get draw object
     draw = ImageDraw.Draw(decorated_image)
@@ -45,10 +46,11 @@ def create_border_decoration(qr_image, decoration_prompt="flowers"):
             for angle in range(0, 360, 45):  # 8 petals
                 x1 = x + petal_size * cos(radians(angle))
                 y1 = y + petal_size * sin(radians(angle))
-                draw.ellipse([x1-5, y1-5, x1+5, y1+5], fill='lightpink')
-            draw.ellipse([x-5, y-5, x+5, y+5], fill='yellow')  # Center of flower
-            
-    return decorated_image
+                draw.ellipse([x1-5, y1-5, x1+5, y1+5], fill=(255, 192, 203, 255))  # lightpink in RGBA
+            draw.ellipse([x-5, y-5, x+5, y+5], fill=(255, 255, 0, 255))  # yellow in RGBA
+    
+    # Convert back to RGB before saving
+    return decorated_image.convert('RGB')
 
 # QR 코드 생성 함수
 def create_qr(content, qr_type, fill_color, back_color, box_size, border_size, error_correction, border_decoration="flowers"):