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

# QR 코드 생성 함수
def create_qr(content, qr_type, fill_color, back_color, box_size, border_size, error_correction, border_decoration="flowers"):
    # 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 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"
                    )
                
                error_correction = gr.Dropdown(
                    choices=[
                        "Low (7%)",
                        "Medium (15%)",
                        "Quartile (25%)",
                        "High (30%)"
                    ],
                    value="Medium (15%)",
                    label="Error Correction Level"
                )
                
                border_decoration = gr.Textbox(
                    label="Border Decoration Prompt",
                    placeholder="Enter decoration style (e.g., flowers, simple, modern)",
                    value="flowers"
                )
                
                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",
            server_port=7860,
            share=True,
            debug=True
        )
    except Exception as e:
        print(f"Error starting the application: {e}")