Update app.py

app.py

@@ -3,7 +3,46 @@ import qrcode
 import random
 import os
 from datetime import datetime
-from PIL import Image
+from PIL import Image, ImageDraw
+from math import cos, sin, radians
+
+def create_border_decoration(qr_image, decoration_type="flowers"):
+    # 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
+    new_width = width + (padding * 2)
+    new_height = height + (padding * 2)
+    
+    # Create new image with white background
+    decorated_image = Image.new('RGB', (new_width, new_height), 'white')
+    
+    # Paste QR code in center
+    decorated_image.paste(qr_image, (padding, padding))
+    
+    # Get draw object
+    draw = ImageDraw.Draw(decorated_image)
+    
+    if decoration_type == "flowers":
+        # Draw simple flower patterns in corners
+        corner_positions = [
+            (padding//2, padding//2),  # Top-left
+            (new_width - padding//2, padding//2),  # Top-right
+            (padding//2, new_height - padding//2),  # Bottom-left
+            (new_width - padding//2, new_height - padding//2)  # Bottom-right
+        ]
+        
+        for x, y in corner_positions:
+            # Draw flower (simple version)
+            petal_size = 15
+            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
 
 # QR 코드 생성을 위한 메인 함수
 def create_qr(
@@ -13,7 +52,8 @@ def create_qr(
    back_color, 
    box_size,
    border_size,
-   error_correction
+   error_correction,
+   border_decoration="flowers"  # New parameter
 ):
    # QR 코드 데이터 포맷팅
    formatted_data = format_data(content, qr_type)
@@ -40,6 +80,10 @@ def create_qr(
    # QR 이미지 생성
    qr_img = qr.make_image(fill_color=fill_color, back_color=back_color)
    
+   # Add border decoration if specified
+   if border_decoration != "none":
+       qr_img = create_border_decoration(qr_img, border_decoration)
+   
    # 파일 저장
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    random_id = random.randint(1000, 9999)
@@ -126,8 +170,6 @@ def create_interface():
    
    # 인터페이스 구성
    with gr.Blocks(theme=theme, title="QR Canvas") as demo:
-
-# Inside create_interface() function, fixing the indentation:
        gr.Markdown(
            """
            # 🎯 QR CANVAS
@@ -135,11 +177,9 @@ def create_interface():
            """
        )
 
-       # Fixed indentation to match the surrounding block
        gr.HTML("""<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fginipick-QR-Canvas.hf.space">
                <img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fginipick-QR-Canvas.hf.space&countColor=%23263759" />
                </a>""")
-
        
        with gr.Row():
            with gr.Column(scale=2):
@@ -199,6 +239,17 @@ def create_interface():
                        label="Border Size"
                    )
                
+               border_decoration = gr.Dropdown(
+                   choices=[
+                       "none",
+                       "flowers",
+                       "simple",
+                       "modern"
+                   ],
+                   value="flowers",
+                   label="Border Decoration"
+               )
+               
                error_correction = gr.Dropdown(
                    choices=[
                        "Low (7%)",
@@ -245,7 +296,8 @@ def create_interface():
                back_color,
                box_size,
                border_size,
-               error_correction
+               error_correction,
+               border_decoration
            ],
            outputs=[output_image, output_data]
        )