app.py

import gradio as gr
import qrcode
import random
import os
from datetime import datetime
from PIL import Image, ImageDraw
from math import cos, sin, radians

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
    
    # 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 - now with explicit size
    decorated_image.paste(qr_image, (padding, padding, padding + width, padding + height))
    
    # Get draw object
    draw = ImageDraw.Draw(decorated_image)
    
    # 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
            (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

# 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,
   fill_color,
   back_color, 
   box_size,
   border_size,
   error_correction,
   border_decoration="flowers"  # Default value
):
   # QR 코드 데이터 포맷팅
   formatted_data = format_data(content, qr_type)
   
   # 에러 수정 레벨 설정
   error_levels = {
       "Low (7%)": qrcode.constants.ERROR_CORRECT_L,
       "Medium (15%)": qrcode.constants.ERROR_CORRECT_M,
       "Quartile (25%)": qrcode.constants.ERROR_CORRECT_Q,
       "High (30%)": qrcode.constants.ERROR_CORRECT_H
   }
   
   # QR 코드 생성
   qr = qrcode.QRCode(
       version=1,
       error_correction=error_levels[error_correction],
       box_size=box_size,
       border=border_size,
   )
   
   qr.add_data(formatted_data)
   qr.make(fit=True)
   
   # QR 이미지 생성
   qr_img = qr.make_image(fill_color=fill_color, back_color=back_color)
   
   # Add border decoration if specified
   if border_decoration.strip():  # Only if decoration prompt is not empty
       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)
   filename = f"qrfile/qr_{timestamp}_{random_id}.png"
   
   # 디렉토리 확인 및 생성
   os.makedirs("qrfile", exist_ok=True)
   
   # 이미지 저장
   qr_img.save(filename)
   cleanup_old_files("qrfile/", max_files=100)
   
   return filename, formatted_data

# 데이터 포맷팅 함수
def format_data(content, qr_type):
   if not content:
       return ""
       
   format_rules = {
       "URL": lambda x: f"https://{x}" if not x.startswith(('http://', 'https://')) else x,
       "Email": lambda x: f"mailto:{x}",
       "Phone": lambda x: f"tel:{x}",
       "SMS": lambda x: f"sms:{x}",
       "WhatsApp": lambda x: f"whatsapp://send?text={x}",
       "Location": lambda x: f"geo:{x}",
       "Wi-Fi": lambda x: f"WIFI:S:{x};;",
       "Text": lambda x: x,
       "vCard": lambda x: f"BEGIN:VCARD\nVERSION:3.0\n{x}\nEND:VCARD"
   }
   
   return format_rules[qr_type](content.strip())

# 파일 정리 함수
def cleanup_old_files(directory, max_files):
   files = [f for f in os.listdir(directory) if f.endswith('.png')]
   if len(files) > max_files:
       files.sort(key=lambda x: os.path.getctime(os.path.join(directory, x)))
       for f in files[:-max_files]:
           try:
               os.remove(os.path.join(directory, f))
           except:
               continue

def get_example_placeholder(qr_type):
   examples = {
       "URL": "example.com or https://example.com",
       "Email": "[email protected]",
       "Phone": "+1234567890",
       "SMS": "+1234567890",
       "WhatsApp": "Hello World!",
       "Location": "37.7749,-122.4194",
       "Wi-Fi": "MyWiFiNetwork",
       "Text": "Any text you want to encode",
       "vCard": "FN:John Doe\nTEL:+1234567890\nEMAIL:[email protected]"
   }
   return examples.get(qr_type, "Enter your content here...")

def format_example_text(qr_type):
   examples = {
       "URL": "• Direct URL: https://example.com\n• Without https: example.com",
       "Email": "• Basic: [email protected]\n• With subject: [email protected]?subject=Hello",
       "Phone": "• International: +1234567890\n• Local: 01012345678",
       "SMS": "• Basic: +1234567890\n• With message: +1234567890?body=Hello",
       "WhatsApp": "• Message: Hello World!\n• With number: +1234567890:Hello",
       "Location": "• Coordinates: 37.7749,-122.4194\n• With zoom: 37.7749,-122.4194,15z",
       "Wi-Fi": "• Network name only: MyWiFiNetwork\n• With password: WIFI:S:MyNetwork;P:password;;",
       "Text": "• Simple text: Hello World!\n• Multiple lines: Line 1\\nLine 2",
       "vCard": "• Basic:\nFN:John Doe\nTEL:+1234567890\nEMAIL:[email protected]\n• Extended:\nFN:John Doe\nTEL:+1234567890\nEMAIL:[email protected]\nADR:;;123 Street;City;State;12345;Country"
   }
   return examples.get(qr_type, "Enter your content here...")

# Gradio 인터페이스 생성
def create_interface():
   # 테마 설정
   theme = gr.themes.Soft(
       primary_hue="blue",
       secondary_hue="indigo",
   ).set(
       body_background_fill="*neutral_50",
       block_background_fill="*neutral_100",
       button_primary_background_fill="*primary_500",
   )
   
   # 인터페이스 구성
   with gr.Blocks(theme=theme, title="QR Canvas") as demo:
       gr.Markdown(
           """
           # 🎯 QR CANVAS
           Create customized QR codes for various purposes with professional styling options.
           """
       )

       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):
               # 입력 섹션
               qr_type = gr.Dropdown(
                   choices=[
                       "URL",
                       "Email",
                       "Phone", 
                       "SMS",
                       "WhatsApp",
                       "Location",
                       "Wi-Fi",
                       "Text",
                       "vCard"
                   ],
                   value="URL",
                   label="QR Code Type"
               )

               content = gr.Textbox(
                   label="Content",
                   placeholder="Enter your content here...",
                   lines=3
               )

               example_format = gr.Textbox(
                   value=format_example_text("URL"),
                   label="Format Examples",
                   interactive=False,
                   lines=6
               )
               
               with gr.Row():
                   fill_color = gr.ColorPicker(
                       label="QR Code Color",
                       value="#000000"
                   )
                   back_color = gr.ColorPicker(
                       label="Background Color",
                       value="#FFFFFF"
                   )
               
               with gr.Row():
                   box_size = gr.Slider(
                       minimum=1,
                       maximum=20,
                       value=10,
                       step=1,
                       label="QR Code Size"
                   )
                   border_size = gr.Slider(
                       minimum=0,
                       maximum=10,
                       value=4,
                       step=1,
                       label="Border Size"
                   )
               
               border_decoration = gr.Dropdown(
                   choices=[
                       "none",
                       "flowers",
                       "simple",
                       "modern"
                   ],
                   value="flowers",
                   label="Border Decoration"
               )
               
               error_correction = gr.Dropdown(
                   choices=[
                       "Low (7%)",
                       "Medium (15%)",
                       "Quartile (25%)",
                       "High (30%)"
                   ],
                   value="Medium (15%)",
                   label="Error Correction Level"
               )
               
               generate_btn = gr.Button(
                   "Generate QR Code",
                   variant="primary"
               )
           
           with gr.Column(scale=1):
               # 출력 섹션
               output_image = gr.Image(
                   label="Generated QR Code",
                   type="filepath"
               )
               output_data = gr.Textbox(
                   label="Formatted Data",
                   interactive=False
               )

       def update_example(qr_type):
           return format_example_text(qr_type)

       # 이벤트 핸들러
       qr_type.change(
           fn=update_example,
           inputs=[qr_type],
           outputs=example_format
       )

       generate_btn.click(
           fn=create_qr,
           inputs=[
               content,
               qr_type,
               fill_color,
               back_color,
               box_size,
               border_size,
               error_correction,
               border_decoration
           ],
           outputs=[output_image, output_data]
       )
       
       # 사용 안내
       gr.Markdown(
           """
           ### 📝 Instructions
           1. Select the QR code type from the dropdown menu
           2. Enter your content following the format examples shown
           3. Customize the appearance using the color pickers and sliders
           4. Click 'Generate QR Code' to create your custom QR code
           
           ### 💡 Tips
           - Use higher error correction levels for better scan reliability
           - Ensure sufficient contrast between QR code and background colors
           - Keep the content concise for better readability
           - Follow the format examples for best results
           """
       )
   
   return demo

if __name__ == "__main__":
    try:
        # 출력 디렉토리 생성
        os.makedirs("qrfile", exist_ok=True)
        
        # 인터페이스 생성 및 실행
        demo = create_interface()
        demo.launch(
            server_name="0.0.0.0",  # 모든 IP에서 접근 가능
            server_port=7860,        # 포트 지정
            share=True,              # 공유 링크 생성
            debug=True              # 디버그 모드 활성화
        )
    except Exception as e:
        print(f"Error starting the application: {e}")