Update app.py

app.py

@@ -1,285 +1,219 @@
 import gradio as gr
 import sqlite3
-import json
 import os
 import time
 from reportlab.lib.pagesizes import A4
 from reportlab.pdfgen import canvas
-from reportlab.lib.units import cm
 import qrcode
-from huggingface_hub import upload_file, hf_hub_download
+from huggingface_hub import upload_file
 import requests
 
 REPO_ID = "protae5544/WorkerManagement"
 
-# ดาวน์โหลดและโหลดข้อมูล JSON
-def load_json_data():
-    try:
-        # ลองดาวน์โหลด JSON จาก Space เดิม
-        json_path = hf_hub_download(
-            repo_id="protae5544/worberkit2024poe", 
-            filename="worker-database.json", 
-            repo_type="space"
-        )
-        with open(json_path, 'r', encoding='utf-8') as f:
-            data = json.load(f)
-        print(f"โหลดข้อมูล JSON สำเร็จ: {len(data)} รายการ")
-        return data
-    except Exception as e:
-        print(f"ไม่สามารถโหลด JSON: {e}")
-        # ข้อมูลตัวอย่างถ้าไม่มี JSON
-        return [
-            {
-                "เลขคำขอ": "WP-68-366-150",
-                "ชื่อภาษาอังกฤษ": "MR. AUNG KYAW",
-                "หมายเลขอ้างอิงคนต่างด้าว": "2492102076212",
-                "หมายเลขประจำตัว": "6685490000472",
-                "สัญชาติ": "เมียนมา",
-                "เลขที่บนขวาใบเสร็จ": "2100680001130",
-                "หมายเลขชำระเงิน": "IV680210/002350"
-            }
-        ]
-
 def setup_fonts():
     try:
         from reportlab.pdfbase import pdfmetrics
         from reportlab.pdfbase.ttfonts import TTFont
         
         font_url = 'https://fonts.gstatic.com/s/sarabun/v13/DtVjJx26TKEqsc-lWcmw-wKE7wM.ttf'
-        bold_url = 'https://fonts.gstatic.com/s/sarabun/v13/DtVmJx26TKEqsc-lWcmw5wq0hwOKBwXcb0A.ttf'
-        
-        for filename, url in [('THSarabun.ttf', font_url), ('THSarabunBold.ttf', bold_url)]:
-            if not os.path.exists(filename):
-                response = requests.get(url, timeout=10)
-                if response.status_code == 200:
-                    with open(filename, 'wb') as f:
-                        f.write(response.content)
-                    print(f"ดาวน์โหลดฟอนต์: {filename}")
+        if not os.path.exists('THSarabun.ttf'):
+            response = requests.get(font_url, timeout=10)
+            with open('THSarabun.ttf', 'wb') as f:
+                f.write(response.content)
         
         pdfmetrics.registerFont(TTFont('THSarabun', 'THSarabun.ttf'))
-        pdfmetrics.registerFont(TTSarabunBold', 'THSarabunBold.ttf'))
-        print("ลงทะเบียนฟอนต์สำเร็จ")
         return True
     except Exception as e:
-        print(f"ไม่สามารถตั้งค่าฟอนต์: {e}")
+        print(f"Font error: {e}")
         return False
 
-def thai_date_time(timestamp):
-    months = {
-        1: "มกราคม", 2: "กุมภาพันธ์", 3: "มีนาคม", 4: "เมษายน",
-        5: "พฤษภาคม", 6: "มิถุนายน", 7: "กรกฎาคม", 8: "สิงหาคม",
-        9: "กันยายน", 10: "ตุลาคม", 11: "พฤศจิกายน", 12: "ธันวาคม"
-    }
-    dt = time.localtime(timestamp)
-    return f"{dt.tm_mday:02d} {months[dt.tm_mon]} {dt.tm_year + 543} {dt.tm_hour:02d}:{dt.tm_min:02d} น"
+def init_db():
+    conn = sqlite3.connect("workers.db")
+    c = conn.cursor()
+    c.execute('''CREATE TABLE IF NOT EXISTS workers (
+        request_number TEXT PRIMARY KEY,
+        english_name TEXT,
+        foreign_reference_number TEXT,
+        id_number TEXT,
+        nationality TEXT,
+        receipt_number TEXT,
+        payment_number TEXT,
+        created_at TEXT
+    )''')
+    conn.commit()
+    conn.close()
 
-def create_receipt_pdf(worker_data):
-    request_number = worker_data["เลขคำขอ"]
+def create_pdf(request_number, english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number):
     filename = f"receipt_{request_number}.pdf"
     
+    c = canvas.Canvas(filename, pagesize=A4)
+    width, height = A4
+    
+    font_available = setup_fonts()
+    
+    if font_available:
+        c.setFont("THSarabun", 18)
+    else:
+        c.setFont("Helvetica-Bold", 16)
+    
+    # หัวเอกสาร
+    c.drawCentredText(width/2, height-60, "กระทรวงแรงงาน")
+    c.drawCentredText(width/2, height-85, "ใบเสร็จรับเงิน (ต้นฉบับ)")
+    
+    # ข้อมูลหลัก
+    y = height - 130
+    c.setFont("Helvetica", 12)
+    
+    info_lines = [
+        f"เลขที่: {receipt_number}",
+        "ที่ทำการ: สำนักบริหารแรงงานต่างด้าว", 
+        "วันที่: 01 เมษายน 2568",
+        f"เลขที่ใบชำระเงิน: {payment_number}",
+        f"เลขรับคำขอที่: {request_number}",
+        "",
+        f"ชื่อผู้ชำระเงิน: {english_name}",
+        f"สัญชาติ: {nationality}",
+        f"เลขอ้างอิงคนต่างด้าว: {foreign_reference_number}",
+        f"หมายเลขประจำตัวคนต่างด้าว: {id_number}",
+        "",
+        "ชื่อนายจ้าง: บริษัท บาน กง เอ็นจิเนียริ่ง จำกัด",
+        "เลขประจำตัวนายจ้าง: 0415567000061",
+        "",
+        "รายการค่าธรรมเนียม:",
+        "1. ค่าธรรมเนียมในการยื่นคำขอ ฉบับละ 100 บาท: 100.00",
+        "2. ค่าธรรมเนียมการพิจารณา: 900.00", 
+        "รวมเป็นเงินทั้งสิ้น (บาท): (หนึ่งพันบาทถ้วน) 1,000.00",
+        "",
+        "ได้รับเงินไว้เป็นการถูกต้องแล้ว",
+        "",
+        "(ลงชื่อ) นางสาวอารีวรรณ โพธิ์นิ่มแดง (ผู้รับเงิน)",
+        "ตำแหน่ง: นักวิชาการแรงงานชำนาญการ"
+    ]
+    
+    for line in info_lines:
+        c.drawString(50, y, line)
+        y -= 20
+    
+    # QR Code
+    pdf_url = f"https://huggingface.co/spaces/{REPO_ID}/resolve/main/{filename}"
+    try:
+        qr = qrcode.make(pdf_url)
+        qr.save("temp_qr.png")
+        c.drawImage("temp_qr.png", width-120, 50, width=70, height=70)
+    except Exception as e:
+        print(f"QR error: {e}")
+    
+    # เวลาที่พิมพ์
+    c.setFont("Helvetica", 8)
+    current_time = time.strftime('%d/%m/%Y %H:%M:%S')
+    c.drawString(50, 30, f"พิมพ์เมื่อ: {current_time}")
+    
+    c.save()
+    
+    # อัพโหลดไฟล์
     try:
-        os.makedirs("pdfs", exist_ok=True)
-        
-        c = canvas.Canvas(filename, pagesize=A4)
-        width, height = A4
-        
-        font_available = setup_fonts()
-        
-        # **หัวเอกสาร - ตำแหน่งเดิมทุกตัว**
-        if font_available:
-            c.setFont("THSarabunBold", 18)
-        else:
-            c.setFont("Helvetica-Bold", 18)
-        c.drawString(2*cm, 28*cm, "กระทรวงแรงงาน")
-        
-        if font_available:
-            c.setFont("THSarabun", 14)
-        else:
-            c.setFont("Helvetica", 12)
-        c.drawString(2*cm, 27.5*cm, "ใบเสร็จรับเงิน (ต้นฉบับ)")
-        c.drawString(2*cm, 26*cm, f"เลขที่: {worker_data['เลขที่บนขวาใบเสร็จ']}")
-        c.drawString(2*cm, 25.5*cm, "ที่ทำการ: สำนักบริหารแรงงานต่างด้าว")
-        c.drawString(2*cm, 25*cm, "วันที่: 01 เมษายน 2568")
-        c.drawString(2*cm, 24.5*cm, f"เลขที่ใบชำระเงิน: {worker_data['หมายเลขชำระเงิน']}")
-        c.drawString(2*cm, 24*cm, f"เลขรับคำขอที่: {worker_data['เลขคำขอ']}")
-        c.drawString(2*cm, 23.5*cm, f"ชื่อผู้ชำระเงิน: {worker_data['ชื่อภาษาอังกฤษ']}")
-        c.drawString(2*cm, 23*cm, f"สัญชาติ: {worker_data['สัญชาติ']}")
-        c.drawString(2*cm, 22.5*cm, f"เลขอ้างอิงคนต่างด้าว: {worker_data['หมายเลขอ้างอิงคนต่างด้าว']}")
-        c.drawString(2*cm, 22*cm, f"หมายเลขประจำตัวคนต่างด้าว: {worker_data['หมายเลขประจำตัว']}")
-        c.drawString(2*cm, 21.5*cm, "ชื่อนายจ้าง / สถานประกอบการ: บริษัท บาน กง เอ็นจิเนียริ่ง จำกัด")
-        c.drawString(2*cm, 21*cm, "เลขประจำตัวนายจ้าง: 0415567000061")
-        
-        # **รายการ - เดิมทุกบรรทัด**
-        if font_available:
-            c.setFont("THSarabunBold", 14)
-        else:
-            c.setFont("Helvetica-Bold", 12)
-        c.drawString(2*cm, 20*cm, "รายการ")
-        
-        if font_available:
-            c.setFont("THSarabun", 12)
-        else:
-            c.setFont("Helvetica", 12)
-        c.drawString(2*cm, 19.5*cm, "1. ค่าธรรมเนียมในการยื่นคำขอ ฉบับละ 100 บาท: 100.00")
-        c.drawString(2*cm, 19*cm, "2. ค่าธรรมเนียม: 900.00")
-        c.drawString(2*cm, 18.5*cm, "รวมเป็นเงินทั้งสิ้น (บาท): (หนึ่งพันบาทถ้วน) 1,000.00")
-        c.drawString(2*cm, 18*cm, "ได้รับเงินไว้เป็นการถูกต้องแล้ว")
-        c.drawString(2*cm, 17.5*cm, "(ลงชื่อ) นางสาวอารีวรรณ โพธิ์นิ่มแดง (ผู้รับเงิน)")
-        c.drawString(2*cm, 17*cm, "ตำแหน่ง: นักวิชาการแรงงานชำนาญการ")
-        
-        # **QR Code - ตำแหน่งเดิม**
-        pdf_url = f"https://huggingface.co/spaces/{REPO_ID}/resolve/main/{filename}"
-        try:
-            qr = qrcode.make(pdf_url)
-            qr.save("temp_qr.png")
-            c.drawImage("temp_qr.png", 15*cm, 15*cm, width=3*cm, height=3*cm)
-        except Exception as e:
-            print(f"ไม่สามารถสร้าง QR Code: {e}")
-        
-        # **เวลาพิมพ์ - ตำแหน่งเดิม**
-        c.setFont("Helvetica", 8)
-        print_time = thai_date_time(time.time())
-        c.drawString(2*cm, 2*cm, f"พิมพ์เมื่อ: {print_time}")
-        
-        c.showPage()
-        c.save()
-        
-        # **อัพโหลดไฟล์**
         upload_file(
             path_or_fileobj=filename,
             path_in_repo=filename,
             repo_id=REPO_ID,
             repo_type="space"
         )
-        
-        print(f"✅ สร้าง PDF สำเร็จ: {filename}")
         return pdf_url
-        
     except Exception as e:
-        print(f"❌ ข้อผิดพลาดในการสร้าง PDF: {e}")
+        print(f"Upload error: {e}")
         return None
 
-# โหลดข้อมูล JSON
-json_data = load_json_data()
-
-def get_worker_names():
-    """ดึงรายชื่อพนักงานสำหรับ dropdown"""
-    names = []
-    for worker in json_data:
-        name_display = f"{worker['ชื่อภาษาอังกฤษ']} ({worker['เลขคำขอ']})"
-        names.append(name_display)
-    return names
-
-def get_worker_by_selection(selected_name):
-    """ดึงข้อมูลพนักงานจากการเลือก"""
-    if not selected_name:
-        return "", "", "", "", "", ""
+def add_worker(english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number):
+    # ตรวจสอบข้อมูล
+    if not all([english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number]):
+        return "❌ กรุณากรอกข้อมูลให้ครบทุกช่อง"
     
-    # แยกเลขคำขอจากชื่อที่เลือก
-    request_number = selected_name.split("(")[-1].replace(")", "")
+    if not id_number.isdigit() or len(id_number) != 13:
+        return "❌ หมายเลขประจำตัวต้องเป็นตัวเลข 13 หลักเท่านั้น"
     
-    for worker in json_data:
-        if worker['เลขคำขอ'] == request_number:
-            return (
-                worker['ชื่อภาษาอังกฤษ'],
-                worker['หมายเลขอ้างอิงคนต่างด้าว'],
-                worker['หมายเลขประจำตัว'],
-                worker['สัญชาติ'],
-                worker['เลขที่บนขวาใบเสร็จ'],
-                worker['หมายเลขชำระเงิน']
-            )
+    # สร้างเลขคำขอ
+    conn = sqlite3.connect("workers.db")
+    c = conn.cursor()
+    c.execute("SELECT COUNT(*) FROM workers")
+    count = c.fetchone()[0]
+    conn.close()
     
-    return "", "", "", "", "", ""
-
-def generate_pdf_from_selection(selected_name):
-    """สร้าง PDF จากการเลือกพนักงาน"""
-    if not selected_name:
-        return "❌ กรุณาเลือกพนักงาน"
+    request_number = f"WP-68-366-{150 + count:03d}"
+    
+    # บันทึกข้อมูล
+    conn = sqlite3.connect("workers.db")
+    c = conn.cursor()
+    c.execute('''INSERT INTO workers VALUES (?, ?, ?, ?, ?, ?, ?, ?)''', 
+              (request_number, english_name, foreign_reference_number, id_number, 
+               nationality, receipt_number, payment_number, str(time.time())))
+    conn.commit()
+    conn.close()
     
-    request_number = selected_name.split("(")[-1].replace(")", "")
+    # สร้าง PDF
+    pdf_url = create_pdf(request_number, english_name, foreign_reference_number, 
+                        id_number, nationality, receipt_number, payment_number)
     
-    for worker in json_data:
-        if worker['เลขคำขอ'] == request_number:
-            pdf_url = create_receipt_pdf(worker)
-            if pdf_url:
-                return f"""✅ สร้างใบเสร็จสำเร็จ!
+    if pdf_url:
+        result_text = f"""✅ เสร็จสิ้น! 
 
-📋 เลขคำขอ: {worker['เลขคำขอ']}
-👤 ชื่อ: {worker['ชื่อภาษาอังกฤษ']}
-🆔 หมายเลขประจำตัว: {worker['หมายเลขประจำตัว']}
+📋 เลขคำขอ: {request_number}
+👤 ชื่อ: {english_name}
+🆔 หมายเลขประจำตัว: {id_number}
 
 📄 ดาวน์โหลด PDF: {pdf_url}
 
-✅ หน้าตาใบเสร็จเหมือนเดิม 100%
-📱 สแกน QR Code ในใบเสร็จเพื่อดาวน์โหลด"""
-            else:
-                return "❌ เกิดข้อผิดพลาดในการสร้าง PDF"
-    
-    return "❌ ไม่พบข้อมูลพนักงาน"
+🔍 วิธีใช้:
+1. คลิกลิงก์ด้านบนเพื่อดาวน์โหลด PDF
+2. สแกน QR Code ในใบเสร็จเพื่อดาวน์โหลดอีกครั้ง
+3. ใบเสร็จพร้อมใช้งาน!"""
+        return result_text
+    else:
+        return "❌ เกิดข้อผิดพลาดในการสร้าง PDF"
 
-def show_all_workers():
-    """แสดงรายการพนักงานทั้งหมดจาก JSON"""
-    if not json_data:
-        return "ไม่มีข้อมูลพนักงาน"
+def view_all():
+    conn = sqlite3.connect("workers.db")
+    c = conn.cursor()
+    c.execute("SELECT request_number, english_name, nationality, id_number FROM workers ORDER BY request_number DESC")
+    workers = c.fetchall()
+    conn.close()
+    
+    if not workers:
+        return "ยังไม่มีข้อมูลพนักงานในระบบ"
     
-    result = f"📊 รายการพนักงานจากฐานข้อมูล ({len(json_data)} คน)\n\n"
-    for i, worker in enumerate(json_data, 1):
-        result += f"{i}. {worker['เลขคำขอ']} | {worker['ชื่อภาษาอังกฤษ']} | {worker['สัญชาติ']} | {worker['หมายเลขประจำตัว']}\n"
+    result_text = f"📊 รายการพนักงานทั้งหมด ({len(workers)} คน)\n\n"
+    for i, worker in enumerate(workers, 1):
+        result_text += f"{i}. {worker[0]} | {worker[1]} | {worker[2]} | {worker[3]}\n"
     
-    return result
+    return result_text
+
+# เริ่มต้นระบบ
+init_db()
 
 # สร้าง UI
-with gr.Blocks(title="ระบบจัดการพนักงาน - เหมือนเดิม") as app:
+with gr.Blocks(title="ระบบจัดการพนักงาน") as app:
     gr.Markdown("# 🏢 ระบบจัดการข้อมูลพนักงานต่างด้าว")
     gr.Markdown("### บริษัท บาน กง เอ็นจิเนียริ่ง จำกัด")
-    gr.Markdown("#### ✅ ดึงข้อมูลจาก JSON อัตโนมัติ | หน้าตาเอกสารเหมือนเดิม 100%")
-    
-    with gr.Tab("📝 สร้างใบเสร็จจาก JSON"):
-        gr.Markdown("### เลือกพนักงานจากฐานข้อมูล")
-        
-        worker_dropdown = gr.Dropdown(
-            choices=get_worker_names(),
-            label="🔍 เลือกพนักงาน",
-            interactive=True
-        )
-        
-        gr.Markdown("**ข้อมูลที่เลือก:**")
-        with gr.Row():
-            name_display = gr.Textbox(label="ชื่อภาษาอังกฤษ", interactive=False)
-            ref_display = gr.Textbox(label="หมายเลขอ้างอิง", interactive=False)
-        
-        with gr.Row():
-            id_display = gr.Textbox(label="หมายเลขประจำตัว", interactive=False)
-            nation_display = gr.Textbox(label="สัญชาติ", interactive=False)
-        
-        with gr.Row():
-            receipt_display = gr.Textbox(label="เลขที่ใบเสร็จ", interactive=False)
-            payment_display = gr.Textbox(label="หมายเลขชำระเงิน", interactive=False)
-        
-        generate_btn = gr.Button("🎯 สร้างใบเสร็จ (เหมือนเดิม 100%)", variant="primary", size="lg")
-        result_output = gr.Textbox(label="ผลลัพธ์", lines=10, interactive=False)
-    
-    with gr.Tab("📋 ดูข้อมูลทั้งหมด"):
-        show_all_btn = gr.Button("👥 แสดงข้อมูลจาก JSON")
-        all_workers_output = gr.Textbox(label="รายการพนักงานทั้งหมด", lines=20, interactive=False)
     
-    # เชื่อมต่อ Events
-    worker_dropdown.change(
-        fn=get_worker_by_selection,
-        inputs=[worker_dropdown],
-        outputs=[name_display, ref_display, id_display, nation_display, receipt_display, payment_display]
-    )
+    with gr.Tab("📝 เพิ่มพนักงานใหม่"):
+        gr.Markdown("### กรอกข้อมูลพนักงาน")
+        
+        name = gr.Textbox(label="ชื่อภาษาอังกฤษ", placeholder="MR. AUNG KYAW")
+        ref_num = gr.Textbox(label="หมายเลขอ้างอิงคนต่างด้าว", placeholder="2492102076212")
+        id_num = gr.Textbox(label="หมายเลขประจำตัว (13 หลัก)", placeholder="6685490000472")
+        nation = gr.Dropdown(["เมียนมา", "กัมพูชา", "ลาว", "เวียดนาม"], 
+                           label="สัญชาติ", value="เมียนมา")
+        receipt = gr.Textbox(label="เลขที่ใบเสร็จ", placeholder="2100680001130")
+        payment = gr.Textbox(label="หมายเลขชำระเงิน", placeholder="IV680210/002350")
+        
+        submit = gr.Button("💾 สร้างใบเสร็จ", variant="primary", size="lg")
+        result = gr.Textbox(label="ผลลัพธ์", lines=8, interactive=False)
     
-    generate_btn.click(
-        fn=generate_pdf_from_selection,
-        inputs=[worker_dropdown],
-        outputs=[result_output]
-    )
+    with gr.Tab("📋 ดูรายการ"):
+        view_btn = gr.Button("👥 ดูรายการพนักงานทั้งหมด")
+        worker_list = gr.Textbox(label="รายการ", lines=15, interactive=False)
     
-    show_all_btn.click(
-        fn=show_all_workers,
-        outputs=[all_workers_output]
-    )
+    # เชื่อมต่อฟังก์ชัน
+    submit.click(add_worker, [name, ref_num, id_num, nation, receipt, payment], result)
+    view_btn.click(view_all, outputs=worker_list)
 
-print("🚀 เริ่มต้นระบบ...")
-app.launch(server_name="0.0.0.0", server_port=7860)
+app.launch()