Update app.py

app.py

@@ -7,16 +7,15 @@ from PIL import Image, ImageDraw
 from math import cos, sin, radians
 
 # 테두리 장식 함수
-# create_border_decoration 함수 수정
-def create_border_decoration(qr_image, decoration_prompt="flowers"):
+def create_border_decoration(qr_image, decoration_prompt="🌸"):
     # Convert QR image to RGB mode first
     qr_image = qr_image.convert('RGB')
     
     # Get the size of the QR code image
     width, height = qr_image.size
     
-    # Create a new image with padding for decoration
-    padding = 50  # Padding for decoration
+    # Create a new image with smaller padding for decoration
+    padding = 30  # 패딩을 줄여서 QR 코드 크기를 더 크게
     new_width = width + (padding * 2)
     new_height = height + (padding * 2)
     
@@ -29,51 +28,118 @@ def create_border_decoration(qr_image, decoration_prompt="flowers"):
     # Get draw object
     draw = ImageDraw.Draw(decorated_image)
     
-    # Parse decoration prompt and draw accordingly
-    if "flower" in decoration_prompt.lower():
-        # 꽃 크기 설정
-        flower_size = 12
-        gap = flower_size * 2  # 꽃 사이의 간격
+    # 이모지에 따른 장식 설정
+    emoji_size = 12
+    gap = emoji_size * 2  # 이모지 사이의 간격
+    
+    # 테두리를 따라 점들의 위치 계산
+    border_points = []
+    
+    # 상단 테두리
+    for x in range(padding//2, new_width - padding//2, gap):
+        border_points.append((x, padding//2))
         
-        # 테두리를 따라 점들의 위치 계산
-        border_points = []
+    # 우측 테두리
+    for y in range(padding//2, new_height - padding//2, gap):
+        border_points.append((new_width - padding//2, y))
         
-        # 상단 테두리
-        for x in range(padding//2, new_width - padding//2, gap):
-            border_points.append((x, padding//2))
-            
-        # 우측 테두리
-        for y in range(padding//2, new_height - padding//2, gap):
-            border_points.append((new_width - padding//2, y))
-            
-        # 하단 테두리
-        for x in range(new_width - padding//2, padding//2, -gap):
-            border_points.append((x, new_height - padding//2))
-            
-        # 좌측 테두리
-        for y in range(new_height - padding//2, padding//2, -gap):
-            border_points.append((padding//2, y))
+    # 하단 테두리
+    for x in range(new_width - padding//2, padding//2, -gap):
+        border_points.append((x, new_height - padding//2))
         
-        # 각 위치에 꽃 그리기
-        for x, y in border_points:
-            # 꽃잎 그리기 (8개의 꽃잎)
-            petal_colors = ['pink', 'lightpink', 'peachpuff']  # 다양한 색상 사용
-            for angle in range(0, 360, 45):  # 8 petals
-                x1 = x + flower_size * cos(radians(angle))
-                y1 = y + flower_size * sin(radians(angle))
-                petal_color = petal_colors[angle % len(petal_colors)]
-                draw.ellipse([x1-4, y1-4, x1+4, y1+4], fill=petal_color)
-            
-            # 꽃 중심 그리기
+    # 좌측 테두리
+    for y in range(new_height - padding//2, padding//2, -gap):
+        border_points.append((padding//2, y))
+    
+    # 각 위치에 이모지 그리기
+    for x, y in border_points:
+        if "🌸" in decoration_prompt:  # 벚꽃
+            for angle in range(0, 360, 45):
+                x1 = x + emoji_size * cos(radians(angle))
+                y1 = y + emoji_size * sin(radians(angle))
+                draw.ellipse([x1-4, y1-4, x1+4, y1+4], fill='pink')
             draw.ellipse([x-3, y-3, x+3, y+3], fill='yellow')
-            
-            # 잎 추가 (선택적)
-            if angle % 180 == 0:  # 양쪽에 잎 추가
-                draw.ellipse([x-8, y+8, x+8, y+16], fill='lightgreen')
-                draw.ellipse([x-8, y-16, x+8, y-8], fill='lightgreen')
+        
+        elif "🍀" in decoration_prompt:  # 네잎클로버
+            for angle in range(0, 360, 90):
+                x1 = x + emoji_size * cos(radians(angle))
+                y1 = y + emoji_size * sin(radians(angle))
+                draw.ellipse([x1-4, y1-4, x1+4, y1+4], fill='lightgreen')
+            draw.ellipse([x-2, y-2, x+2, y+2], fill='darkgreen')
+        
+        elif "⭐" in decoration_prompt:  # 별
+            for angle in range(0, 360, 72):
+                x1 = x + emoji_size * cos(radians(angle))
+                y1 = y + emoji_size * sin(radians(angle))
+                draw.polygon([(x1, y1), (x1+4, y1+4), (x1-4, y1+4)], fill='gold')
+        
+        elif "❤️" in decoration_prompt:  # 하트
+            draw.ellipse([x-6, y-6, x+6, y], fill='red')
+            draw.polygon([(x-6, y), (x+6, y), (x, y+8)], fill='red')
+        
+        elif "🌟" in decoration_prompt:  # 반짝이는 별
+            for angle in range(0, 360, 45):
+                x1 = x + emoji_size * cos(radians(angle))
+                y1 = y + emoji_size * sin(radians(angle))
+                draw.line([(x, y), (x1, y1)], fill='yellow', width=2)
+            draw.ellipse([x-3, y-3, x+3, y+3], fill='white')
+        
+        elif "🍃" in decoration_prompt:  # 잎
+            draw.ellipse([x-8, y-4, x+8, y+4], fill='lightgreen')
+            draw.ellipse([x-4, y-8, x+4, y+8], fill='green')
+        
+        elif "💫" in decoration_prompt:  # 소용돌이
+            for i in range(5):
+                size = emoji_size - (i * 2)
+                angle = i * 45
+                x1 = x + size * cos(radians(angle))
+                y1 = y + size * sin(radians(angle))
+                draw.ellipse([x1-3, y1-3, x1+3, y1+3], fill='lightyellow')
+        
+        elif "✨" in decoration_prompt:  # 반짝임
+            for angle in range(0, 360, 60):
+                x1 = x + emoji_size * cos(radians(angle))
+                y1 = y + emoji_size * sin(radians(angle))
+                draw.line([(x, y), (x1, y1)], fill='gold', width=1)
+        
+        elif "🌊" in decoration_prompt:  # 파도
+            draw.arc([x-8, y-4, x+8, y+4], 0, 180, fill='lightblue')
+            draw.arc([x-8, y, x+8, y+8], 0, 180, fill='blue')
+        
+        elif "🎨" in decoration_prompt:  # 무지개
+            colors = ['red', 'orange', 'yellow', 'green', 'blue', 'purple']
+            for i, color in enumerate(colors):
+                size = emoji_size - (i * 2)
+                draw.arc([x-size, y-size, x+size, y+size], 0, 180, fill=color)
     
     return decorated_image
 
+# create_interface() 함수 내의 border_decoration 부분 수정
+border_decoration = gr.Dropdown(
+    choices=[
+        "🌸 Cherry Blossoms",
+        "🍀 Four Leaf Clover",
+        "⭐ Stars",
+        "❤️ Hearts",
+        "🌟 Sparkles",
+        "🍃 Leaves",
+        "💫 Swirls",
+        "✨ Twinkles",
+        "🌊 Waves",
+        "🎨 Rainbow"
+    ],
+    value="🌸 Cherry Blossoms",
+    label="Border Decoration Style"
+)
+
+# create_qr 함수에서 border_decoration 처리 수정
+def create_qr(..., border_decoration="🌸 Cherry Blossoms"):
+    ...
+    if border_decoration.strip():
+        emoji = border_decoration.split()[0]  # Get only the emoji part
+        qr_img = create_border_decoration(qr_img, emoji)
+    ...
+
 # QR 코드 생성 함수
 def create_qr(content, qr_type, fill_color, back_color, box_size, border_size, error_correction, border_decoration="flowers"):
     # QR 코드 데이터 포맷팅