Update app.py

app.py

@@ -1,16 +1,266 @@
-
 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))
 
-def texttoqr(Input):
-    qr_img = qrcode.make(Input)
-    count = random.randint(1, 99999999999999999999999999999999999999)
-    if (count > 0):             
-        name = 'qrfile/'+"qr-img"+ str(count) +".jpg"
-        qr_img.save(name)
-        return name 
+       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
 
-demo = gr.Interface(texttoqr, gr.Textbox(lines=1, placeholder="Enter Your Text Here..."), "image")     
-demo.launch()
+# 메인 실행
+if __name__ == "__main__":
+   demo = create_interface()
+   demo.launch(share=True)