Update app.py

app.py

@@ -1,16 +1,45 @@
 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
+from huggingface_hub import upload_file, hf_hub_download
 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
@@ -19,221 +48,238 @@ def setup_fonts():
         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'
         
-        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)
-        
-        if not os.path.exists('THSarabunBold.ttf'):
-            response = requests.get(bold_url, timeout=10)
-            with open('THSarabunBold.ttf', 'wb') as f:
-                f.write(response.content)
+        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}")
         
         pdfmetrics.registerFont(TTFont('THSarabun', 'THSarabun.ttf'))
-        pdfmetrics.registerFont(TTFont('THSarabunBold', 'THSarabunBold.ttf'))
+        pdfmetrics.registerFont(TTSarabunBold', 'THSarabunBold.ttf'))
+        print("ลงทะเบียนฟอนต์สำเร็จ")
         return True
-    except:
+    except Exception as e:
+        print(f"ไม่สามารถตั้งค่าฟอนต์: {e}")
         return False
 
-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_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("THSarabunBold", 52.284/2.83465)  # ขนาดตามต้นฉบับ
-    else:
-        c.setFont("Helvetica-Bold", 18)
-    c.drawString(2*cm, 28*cm, "กระทรวงแรงงาน")
-    
-    if font_available:
-        c.setFont("THSarabun", 35.212/2.83465)
-    else:
-        c.setFont("Helvetica", 12)
-    c.drawString(2*cm, 27.5*cm, "ใบเสร็จรับเงิน (ต้นฉบับ)")
-    c.drawString(2*cm, 26*cm, f"เลขที่: {receipt_number}")
-    c.drawString(2*cm, 25.5*cm, "ที่ทำการ: สำนักบริหารแรงงานต่างด้าว")
-    c.drawString(2*cm, 25*cm, "วันที่: 01 เมษายน 2568")
-    c.drawString(2*cm, 24.5*cm, f"เลขที่ใบชำระเงิน: {payment_number}")
-    c.drawString(2*cm, 24*cm, f"เลขรับคำขอที่: {request_number}")
-    c.drawString(2*cm, 23.5*cm, f"ชื่อผู้ชำระเงิน: {english_name}")
-    c.drawString(2*cm, 23*cm, f"สัญชาติ: {nationality}")
-    c.drawString(2*cm, 22.5*cm, f"เลขอ้างอิงคนต่างด้าว: {foreign_reference_number}")
-    c.drawString(2*cm, 22*cm, f"หมายเลขประจำตัวคนต่างด้าว: {id_number}")
-    c.drawString(2*cm, 21.5*cm, "ชื่อนายจ้าง / สถานประกอบการ: บริษัท บาน กง เอ็นจิเนียริ่ง จำกัด")
-    c.drawString(2*cm, 21*cm, "เลขประจำตัวนายจ้าง: 0415567000061")
-    
-    # **รายการ - เหมือนต้นฉบับ**
-    if font_available:
-        c.setFont("THSarabunBold", 35.212/2.83465)
-    else:
-        c.setFont("Helvetica-Bold", 12)
-    c.drawString(2*cm, 20*cm, "รายการ")
-    
-    if font_available:
-        c.setFont("THSarabun", 35.212/2.83465)
-    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"Error creating QR code: {e}")
-    
-    # **เพิ่ม print timestamp ด้านล่าง - เหมือนต้นฉบับ**
-    if font_available:
-        c.setFont("THSarabun", 20/2.83465)
-    else:
-        c.setFont("Helvetica", 7)
-    
-    # แปลงเวลาเป็นภาษาไทย
+def thai_date_time(timestamp):
     months = {
         1: "มกราคม", 2: "กุมภาพันธ์", 3: "มีนาคม", 4: "เมษายน",
         5: "พฤษภาคม", 6: "มิถุนายน", 7: "กรกฎาคม", 8: "สิงหาคม",
         9: "กันยายน", 10: "ตุลาคม", 11: "พฤศจิกายน", 12: "ธันวาคม"
     }
-    dt = time.localtime(time.time())
-    thai_time = f"{dt.tm_mday:02d} {months[dt.tm_mon]} {dt.tm_year + 543} {dt.tm_hour:02d}:{dt.tm_min:02d} น"
-    
-    c.drawString(2*cm, 2*cm, f"พิมพ์เมื่อ: {thai_time}")
-    
-    c.showPage()
-    c.save()
+    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 create_receipt_pdf(worker_data):
+    request_number = worker_data["เลขคำขอ"]
+    filename = f"receipt_{request_number}.pdf"
     
-    # อัพโหลดไฟล์
     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:
+        
+    except Exception as e:
+        print(f"❌ ข้อผิดพลาดในการสร้าง PDF: {e}")
         return None
 
-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 "❌ กรุณากรอกข้อมูลให้ครบทุกช่อง"
-    
-    if not id_number.isdigit() or len(id_number) != 13:
-        return "❌ หมายเลขประจำตัวต้องเป็นตัวเลข 13 หลักเท่านั้น"
+# โหลดข้อมูล 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 "", "", "", "", "", ""
     
-    # สร้างเลขคำขอ
-    conn = sqlite3.connect("workers.db")
-    c = conn.cursor()
-    c.execute("SELECT COUNT(*) FROM workers")
-    count = c.fetchone()[0]
-    conn.close()
+    # แยกเลขคำขอจากชื่อที่เลือก
+    request_number = selected_name.split("(")[-1].replace(")", "")
     
-    request_number = f"WP-68-366-{150 + count:03d}"
+    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('''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()
+    return "", "", "", "", "", ""
+
+def generate_pdf_from_selection(selected_name):
+    """สร้าง PDF จากการเลือกพนักงาน"""
+    if not selected_name:
+        return "❌ กรุณาเลือกพนักงาน"
     
-    # สร้าง PDF
-    pdf_url = create_pdf(request_number, english_name, foreign_reference_number, 
-                        id_number, nationality, receipt_number, payment_number)
+    request_number = selected_name.split("(")[-1].replace(")", "")
     
-    if pdf_url:
-        result = f"""✅ สร้างใบเสร็จสำเร็จ!
+    for worker in json_data:
+        if worker['เลขคำขอ'] == request_number:
+            pdf_url = create_receipt_pdf(worker)
+            if pdf_url:
+                return f"""✅ สร้างใบเสร็จสำเร็จ!
 
-📋 เลขคำขอ: {request_number}
-👤 ชื่อ: {english_name}
-🆔 หมายเลขประจำตัว: {id_number}
+📋 เลขคำขอ: {worker['เลขคำขอ']}
+👤 ชื่อ: {worker['ชื่อภาษาอังกฤษ']}
+🆔 หมายเลขประจำตัว: {worker['หมายเลขประจำตัว']}
 
 📄 ดาวน์โหลด PDF: {pdf_url}
 
-🔍 หน้าตาใบเสร็จจะเหมือนเดิมทุกประการ:
-- หัวเอกสาร "กระทรวงแรงงาน"
-- ข้อมูลเรียงตามต้นฉบับ
-- QR Code ที่มุมขวา
-- ลายเซ็นเจ้าหน้าที่
-- วันที่พิมพ์ด้านล่าง
-
-📱 สแกน QR Code ในใบเสร็จเพื่อดาวน์โหลดอีกครั้ง"""
-        return result
-    else:
-        return "❌ เกิดข้อผิดพลาดในการสร้าง PDF"
-
-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()
+✅ หน้าตาใบเสร็จเหมือนเดิม 100%
+📱 สแกน QR Code ในใบเสร็จเพื่อดาวน์โหลด"""
+            else:
+                return "❌ เกิดข้อผิดพลาดในการสร้าง PDF"
     
-    if not workers:
-        return "ยังไม่มีข้อมูลพนักงานในระบบ"
+    return "❌ ไม่พบข้อมูลพนักงาน"
+
+def show_all_workers():
+    """แสดงรายการพนักงานทั้งหมดจาก JSON"""
+    if not json_data:
+        return "ไม่มีข้อมูลพนักงาน"
     
-    result = f"📊 รายการพนักงานทั้งหมด ({len(workers)} คน)\n\n"
-    for i, worker in enumerate(workers, 1):
-        result += f"{i}. {worker[0]} | {worker[1]} | {worker[2]} | {worker[3]}\n"
+    result = f"📊 รายการพนักงานจากฐานข้อมูล ({len(json_data)} คน)\n\n"
+    for i, worker in enumerate(json_data, 1):
+        result += f"{i}. {worker['เลขคำขอ']} | {worker['ชื่อภาษาอังกฤษ']} | {worker['สัญชาติ']} | {worker['หมายเลขประจำตัว']}\n"
     
     return result
 
-init_db()
-
-with gr.Blocks(title="ระบบจัดการพนักงาน") as app:
+# สร้าง UI
+with gr.Blocks(title="ระบบจัดการพนักงาน - เหมือนเดิม") as app:
     gr.Markdown("# 🏢 ระบบจัดการข้อมูลพนักงานต่างด้าว")
     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)
     
-    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=12, 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("📋 ดูรายการ"):
-        view_btn = gr.Button("👥 ดูรายการพนักงานทั้งหมด")
-        worker_list = gr.Textbox(label="รายการ", lines=15, interactive=False)
+    generate_btn.click(
+        fn=generate_pdf_from_selection,
+        inputs=[worker_dropdown],
+        outputs=[result_output]
+    )
     
-    submit.click(add_worker, [name, ref_num, id_num, nation, receipt, payment], result)
-    view_btn.click(view_all, outputs=worker_list)
+    show_all_btn.click(
+        fn=show_all_workers,
+        outputs=[all_workers_output]
+    )
 
-app.launch()
+print("🚀 เริ่มต้นระบบ...")
+app.launch(server_name="0.0.0.0", server_port=7860)