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="🌸"):
    # Convert QR image to RGB mode first
    qr_image = qr_image.convert('RGB')
    
    # Get the size of the QR code image
    width, height = qr_image.size
    
    # Create a new image with smaller padding for decoration
    padding = 30  # 패딩을 줄여서 QR 코드 크기를 더 크게
    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 (with RGB mode)
    decorated_image.paste(qr_image, (padding, padding))
    
    # Get draw object
    draw = ImageDraw.Draw(decorated_image)
    
    # 이모지에 따른 장식 설정
    emoji_size = 12
    gap = emoji_size * 2  # 이모지 사이의 간격
    
    # 테두리를 따라 점들의 위치 계산
    border_points = []
    
    # 상단 테두리
    for x in range(padding//2, new_width - padding//2, gap):
        border_points.append((x, padding//2))
        
    # 우측 테두리
    for y in range(padding//2, new_height - padding//2, gap):
        border_points.append((new_width - padding//2, y))
        
    # 하단 테두리
    for x in range(new_width - padding//2, padding//2, -gap):
        border_points.append((x, new_height - padding//2))
        
    # 좌측 테두리
    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:  # 벚꽃
            for angle in range(0, 360, 45):
                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='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:  # 소용돌이
            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)
    
    return decorated_image

# create_interface() 함수 내의 border_decoration 부분 수정
border_decoration = gr.Dropdown(
    choices=[
        "🌸 Cherry Blossoms",
        "🍀 Four Leaf Clover",
        "⭐ Stars",
        "❤️ Hearts",
        "🌟 Sparkles",
        "🍃 Leaves",
        "💫 Swirls",
        "✨ Twinkles",
        "🌊 Waves",
        "🎨 Rainbow"
    ],
    value="🌸 Cherry Blossoms",
    label="Border Decoration Style"
)

# create_qr 함수에서 border_decoration 처리 수정
def create_qr(..., border_decoration="🌸 Cherry Blossoms"):
    ...
    if border_decoration.strip():
        emoji = border_decoration.split()[0]  # Get only the emoji part
        qr_img = create_border_decoration(qr_img, emoji)
    ...

# 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: example@email.com\n• With subject: example@email.com?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:john@example.com\n• Extended:\nFN:John Doe\nTEL:+1234567890\nEMAIL:john@example.com\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}")