app.py

import gradio as gr
import qrcode
import random
import os
from datetime import datetime
from PIL import Image

# QR 코드 생성을 위한 메인 함수
def create_qr(
   content,
   qr_type,
   fill_color,
   back_color, 
   box_size,
   border_size,
   error_correction
):
   # 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)
   
   # 파일 저장
   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="Advanced QR Code Generator") as demo:
       gr.Markdown(
           """
           # 🎯 Advanced QR Code Generator
           Create customized QR codes for various purposes with professional styling options.
           """
       )
       
       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.Markdown(
                   format_example_text("URL"),
                   label="Format Examples"
               )
               
               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"
               )
               
               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
               )
       
       # QR 타입 변경 시 예제 텍스트 업데이트
       def update_example(qr_type):
           return gr.Markdown.update(value=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
           ],
           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__":
   demo = create_interface()
   demo.launch(share=True)