Update app.py

app.py

@@ -6,7 +6,7 @@ from datetime import datetime
 from PIL import Image, ImageDraw
 from math import cos, sin, radians
 
-def create_border_decoration(qr_image, decoration_prompt="🌸"):
+def create_border_decoration(qr_image, decoration_style="Flowers"):
     # Convert QR image to RGB mode first
     qr_image = qr_image.convert('RGB')
     
@@ -27,9 +27,9 @@ def create_border_decoration(qr_image, decoration_prompt="🌸"):
     # Get draw object
     draw = ImageDraw.Draw(decorated_image)
     
-    # 이모지에 따른 장식 설정
-    emoji_size = 12
-    gap = emoji_size * 2  # 이모지 사이의 간격
+    # 장식 크기 설정
+    deco_size = 12
+    gap = deco_size * 2  # 장식 사이의 간격
     
     # 테두리를 따라 점들의 위치 계산
     border_points = []
@@ -49,70 +49,97 @@ def create_border_decoration(qr_image, decoration_prompt="🌸"):
     # 좌측 테두리
     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:  # 벚꽃
+        if decoration_style == "Flowers":  # 꽃 패턴
+            # 꽃잎
             for angle in range(0, 360, 45):
-                x1 = x + emoji_size * cos(radians(angle))
-                y1 = y + emoji_size * sin(radians(angle))
+                x1 = x + deco_size * cos(radians(angle))
+                y1 = y + deco_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')
-        
-        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:  # 소용돌이
+            
+        elif decoration_style == "Hearts":  # 하트 패턴
+            # 작은 하트들을 연속으로 그리기
+            heart_size = 6
+            draw.ellipse([x-heart_size, y-heart_size, x, y], fill='red')
+            draw.ellipse([x, y-heart_size, x+heart_size, y], fill='red')
+            draw.polygon([(x-heart_size, y), (x+heart_size, y), (x, y+heart_size)], fill='red')
+            
+        elif decoration_style == "Waves":  # 웨이브 패턴
+            wave_size = 10
+            draw.arc([x-wave_size, y-wave_size//2, x+wave_size, y+wave_size//2],
+                    0, 180, fill='lightblue', width=2)
+            draw.arc([x-wave_size, y, x+wave_size, y+wave_size],
+                    0, 180, fill='blue', width=2)
+            
+        elif decoration_style == "Leaves":  # 잎 패턴
+            leaf_size = 8
+            # 메인 잎
+            draw.ellipse([x-leaf_size, y-leaf_size//2, x+leaf_size, y+leaf_size//2],
+                        fill='lightgreen')
+            # 작은 잎
+            draw.ellipse([x-leaf_size//2, y-leaf_size, x+leaf_size//2, y+leaf_size],
+                        fill='darkgreen')
+            
+        elif decoration_style == "Stars":  # 별 패턴
+            star_size = 6
+            points = []
             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)
+                angle = i * 72
+                # 외곽 점
+                x1 = x + star_size * cos(radians(angle))
+                y1 = y + star_size * sin(radians(angle))
+                points.append((x1, y1))
+                # 내부 점
+                x2 = x + (star_size/2) * cos(radians(angle + 36))
+                y2 = y + (star_size/2) * sin(radians(angle + 36))
+                points.append((x2, y2))
+            draw.polygon(points, fill='gold')
+            
+        elif decoration_style == "Chains":  # 체인 패턴
+            chain_size = 8
+            draw.ellipse([x-chain_size, y-chain_size//2, x+chain_size, y+chain_size//2],
+                        outline='gray', width=2)
+            
+        elif decoration_style == "Bubbles":  # 버블 패턴
+            bubble_sizes = [6, 4, 2]
+            for size in bubble_sizes:
+                draw.ellipse([x-size, y-size, x+size, y+size],
+                           outline='skyblue', width=1)
+            
+        elif decoration_style == "Vines":  # 덩굴 패턴
+            vine_size = 10
+            # 메인 덩굴
+            draw.arc([x-vine_size, y-vine_size, x+vine_size, y+vine_size],
+                    0, 180, fill='green', width=2)
+            # 작은 잎
+            draw.ellipse([x-3, y-3, x+3, y+3], fill='lightgreen')
+            
+        elif decoration_style == "Diamonds":  # 다이아몬드 패턴
+            diamond_size = 6
+            points = [
+                (x, y-diamond_size),  # 상단
+                (x+diamond_size, y),  # 우측
+                (x, y+diamond_size),  # 하단
+                (x-diamond_size, y)   # 좌측
+            ]
+            draw.polygon(points, outline='purple', width=1)
+            
+        elif decoration_style == "Lace":  # 레이스 패턴
+            lace_size = 8
+            # 레이스 원형 패턴
+            draw.arc([x-lace_size, y-lace_size, x+lace_size, y+lace_size],
+                    0, 180, fill='gray', width=1)
+            draw.arc([x-lace_size//2, y-lace_size//2, x+lace_size//2, y+lace_size//2],
+                    180, 360, fill='gray', width=1)
     
     return decorated_image
 
+
+
 def create_qr(content, qr_type, fill_color, back_color, box_size, border_size, error_correction, border_decoration="🌸 Cherry Blossoms"):
     # QR 코드 데이터 포맷팅
     formatted_data = format_data(content, qr_type)
@@ -202,7 +229,7 @@ def format_example_text(qr_type):
     }
     return examples.get(qr_type, "Enter your content here...")
 
-# Gradio 인터페이스 생성
+
 def create_interface():
     theme = gr.themes.Soft(
         primary_hue="blue",
@@ -214,6 +241,8 @@ def create_interface():
     )
     
     with gr.Blocks(theme=theme, title="QR Canvas") as demo:
+
+ 
         gr.Markdown(
             """
             # 🎯 QR CANVAS
@@ -282,24 +311,25 @@ def create_interface():
                     value="Medium (15%)",
                     label="Error Correction Level"
                 )
-                
+
                 border_decoration = gr.Dropdown(
                     choices=[
-                        "🌸 Cherry Blossoms",
-                        "🍀 Four Leaf Clover",
-                        "⭐ Stars",
-                        "❤️ Hearts",
-                        "🌟 Sparkles",
-                        "🍃 Leaves",
-                        "💫 Swirls",
-                        "✨ Twinkles",
-                        "🌊 Waves",
-                        "🎨 Rainbow"
+                        "Flowers",    # 꽃 패턴
+                        "Hearts",     # 하트 패턴
+                        "Waves",      # 웨이브 패턴
+                        "Leaves",     # 잎 패턴
+                        "Stars",      # 별 패턴
+                        "Chains",     # 체인 패턴
+                        "Bubbles",    # 버블 패턴
+                        "Vines",      # 덩굴 패턴
+                        "Diamonds",   # 다이아몬드 패턴
+                        "Lace"        # 레이스 패턴
                     ],
-                    value="🌸 Cherry Blossoms",
+                    value="Flowers",
                     label="Border Decoration Style"
                 )
-                
+        
+
                 generate_btn = gr.Button(
                     "Generate QR Code",
                     variant="primary"