Update app.py

app.py

@@ -6,7 +6,11 @@ from datetime import datetime
 from PIL import Image, ImageDraw
 from math import cos, sin, radians
 
-def create_border_decoration(qr_image, decoration_type="flowers"):
+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')
+    
     # Get the size of the QR code image
     width, height = qr_image.size
     
@@ -18,13 +22,14 @@ def create_border_decoration(qr_image, decoration_type="flowers"):
     # 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))
+    # Paste QR code in center - now with explicit size
+    decorated_image.paste(qr_image, (padding, padding, padding + width, padding + height))
     
     # Get draw object
     draw = ImageDraw.Draw(decorated_image)
     
-    if decoration_type == "flowers":
+    # Parse decoration prompt and draw accordingly
+    if "flower" in decoration_prompt.lower():
         # Draw simple flower patterns in corners
         corner_positions = [
             (padding//2, padding//2),  # Top-left
@@ -44,7 +49,14 @@ def create_border_decoration(qr_image, decoration_type="flowers"):
             
     return decorated_image
 
-# QR 코드 생성을 위한 메인 함수
+# Inside create_interface(), update the border_decoration section:
+               border_decoration = gr.Textbox(
+                   label="Border Decoration Prompt",
+                   placeholder="Enter decoration style (e.g., flowers, simple, modern)",
+                   value="flowers"  # Default value
+               )
+
+# Update the create_qr function:
 def create_qr(
    content,
    qr_type,
@@ -53,7 +65,7 @@ def create_qr(
    box_size,
    border_size,
    error_correction,
-   border_decoration="flowers"  # New parameter
+   border_decoration="flowers"  # Default value
 ):
    # QR 코드 데이터 포맷팅
    formatted_data = format_data(content, qr_type)
@@ -81,7 +93,7 @@ def create_qr(
    qr_img = qr.make_image(fill_color=fill_color, back_color=back_color)
    
    # Add border decoration if specified
-   if border_decoration != "none":
+   if border_decoration.strip():  # Only if decoration prompt is not empty
        qr_img = create_border_decoration(qr_img, border_decoration)
    
    # 파일 저장