Update app.py

app.py

@@ -6,6 +6,7 @@ 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':
@@ -49,298 +50,248 @@ def create_border_decoration(qr_image, decoration_prompt="flowers"):
             
     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
+# 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())
+    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...")
+    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...")
+    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 인터페이스 생성
 # 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"  # Default value
-               )
-               
-               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
+    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",  # 모든 IP에서 접근 가능
-            server_port=7860,        # 포트 지정
-            share=True,              # 공유 링크 생성
-            debug=True              # 디버그 모드 활성화
+            server_name="0.0.0.0",
+            server_port=7860,
+            share=True,
+            debug=True
         )
     except Exception as e:
         print(f"Error starting the application: {e}")