Update app.py

app.py

@@ -1,94 +1,40 @@
 import gradio as gr
 import sqlite3
-import json
-from huggingface_hub import hf_hub_download, upload_file
 import os
-from pathlib import Path
+import time
 from reportlab.lib.pagesizes import A4
-from reportlab.lib.units import cm
 from reportlab.pdfgen import canvas
-from reportlab.pdfbase import pdfmetrics
-from reportlab.pdfbase.ttfonts import TTFont
+from reportlab.lib.units import cm
 import qrcode
-import zipfile
-import io
-from dotenv import load_dotenv
-import time
+from huggingface_hub import upload_file
 import requests
 
-# อ่าน .env (ถ้ามี)
-load_dotenv()
-REPO_ID = os.getenv("REPO_ID", "protae5544/WorkerManagement")
-ORIGINAL_REPO_ID = os.getenv("ORIGINAL_REPO_ID", None)
+REPO_ID = "protae5544/WorkerManagement"
 
-# ดาวน์โหลดฟอนต์จาก Space อื่น
-def download_fonts():
+def setup_fonts():
     try:
-        # ดาวน์โหลดฟอนต์จาก protae5544/worberkit2024poe
-        font_urls = {
-            'THSarabunNew.ttf': 'https://huggingface.co/spaces/protae5544/worberkit2024poe/resolve/main/fontCSS/subset-THSarabunNew.woff2',
-            'THSarabunNew-Bold.ttf': 'https://huggingface.co/spaces/protae5544/worberkit2024poe/resolve/main/fontCSS/subset-THSarabunNew-Bold.woff2'
-        }
+        from reportlab.pdfbase import pdfmetrics
+        from reportlab.pdfbase.ttfonts import TTFont
         
-        # ใช้ฟอนต์ที่มีอยู่แล้วในระบบ หรือดาวน์โหลดจาก Google Fonts
-        google_fonts = {
-            'THSarabunNew.ttf': 'https://fonts.gstatic.com/s/sarabun/v13/DtVjJx26TKEqsc-lWcmw-wKE7wM.ttf',
-            'THSarabunNew-Bold.ttf': 'https://fonts.gstatic.com/s/sarabun/v13/DtVmJx26TKEqsc-lWcmw5wq0hwOKBwXcb0A.ttf'
-        }
+        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 google_fonts.items():
-            if not os.path.exists(filename):
-                try:
-                    response = requests.get(url, timeout=10)
-                    if response.status_code == 200:
-                        with open(filename, 'wb') as f:
-                            f.write(response.content)
-                        print(f"Downloaded font: {filename}")
-                except Exception as e:
-                    print(f"Error downloading {filename}: {e}")
-                    
-    except Exception as e:
-        print(f"Error in download_fonts: {e}")
-
-# ตั้งค่าฟอนต์
-def setup_fonts():
-    try:
-        # ดาวน์โหลดฟอนต์ก่อน
-        download_fonts()
+        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 os.path.exists('THSarabunNew.ttf'):
-            pdfmetrics.registerFont(TTFont('THSarabun', 'THSarabunNew.ttf'))
-            print("Registered THSarabun font")
+        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)
         
-        if os.path.exists('THSarabunNew-Bold.ttf'):
-            pdfmetrics.registerFont(TTFont('THSarabunBold', 'THSarabunNew-Bold.ttf'))
-            print("Registered THSarabunBold font")
-            
+        pdfmetrics.registerFont(TTFont('THSarabun', 'THSarabun.ttf'))
+        pdfmetrics.registerFont(TTFont('THSarabunBold', 'THSarabunBold.ttf'))
         return True
-    except Exception as e:
-        print(f"Warning: Could not load fonts: {e}")
+    except:
         return False
 
-# เริ่มต้นฟอนต์
-font_available = setup_fonts()
-
-# ฟังก์ชันแปลงวันที่เป็นภาษาไทย
-def thai_date_time(timestamp):
-    months = {
-        1: "มกราคม", 2: "กุมภาพันธ์", 3: "มีนาคม", 4: "เมษายน",
-        5: "พฤษภาคม", 6: "มิถุนายน", 7: "กรกฎาคม", 8: "สิงหาคม",
-        9: "กันยายน", 10: "ตุลาคม", 11: "พฤศจิกายน", 12: "ธันวาคม"
-    }
-    dt = time.localtime(timestamp)
-    day = dt.tm_mday
-    month = months[dt.tm_mon]
-    year = dt.tm_year + 543
-    hour = dt.tm_hour
-    minute = dt.tm_min
-    return f"{day:02d} {month} {year} {hour:02d}:{minute:02d} น."
-
-# เริ่มต้นฐานข้อมูล
 def init_db():
     conn = sqlite3.connect("workers.db")
     c = conn.cursor()
@@ -102,552 +48,192 @@ def init_db():
         payment_number TEXT,
         created_at TEXT
     )''')
-    c.execute('''CREATE TABLE IF NOT EXISTS attachments (
-        request_number TEXT,
-        file_url TEXT,
-        FOREIGN KEY(request_number) REFERENCES workers(request_number)
-    )''')
-    c.execute('''CREATE TABLE IF NOT EXISTS pdfs (
-        request_number TEXT,
-        pdf_url TEXT,
-        FOREIGN KEY(request_number) REFERENCES workers(request_number)
-    )''')
     conn.commit()
     conn.close()
 
-# สำรองฐานข้อมูล
-def create_receipt_pdf(request_number, english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number):
-    filename = f"pdfs/receipt_{request_number}.pdf"
-    local_path = f"temp_{request_number}.pdf"
+def create_pdf(request_number, english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number):
+    filename = f"receipt_{request_number}.pdf"
     
-    # สร้างโฟลเดอร์ pdfs ถ้าไม่มี
-    os.makedirs("pdfs", exist_ok=True)
-    
-    c = canvas.Canvas(local_path, pagesize=A4)
+    c = canvas.Canvas(filename, pagesize=A4)
     width, height = A4
     
-    # กำหนดตำแหน่งและขนาดตามเอกสารต้นฉบับ
-    # ใช้ Helvetica สำหรับความเข้ากันได้
-    
-    # หัวเอกสาร - กระทรวงแรงงาน
-    c.setFont("Helvetica-Bold", 20)
-    c.drawCentredText(width/2, height-80, "กระทรวงแรงงาน")
-    
-    c.setFont("Helvetica-Bold", 16)
-    c.drawCentredText(width/2, height-110, "ใบเสร็จรับเงิน (ต้นฉบับ)")
-    
-    # ข้อมูลส่วนหัว
-    c.setFont("Helvetica", 12)
-    y_pos = height - 150
-    
-    c.drawString(50, y_pos, f"เลขที่: {receipt_number}")
-    c.drawString(350, y_pos, "แบบ ภ.1600-034")
-    
-    y_pos -= 25
-    c.drawString(50, y_pos, "ที่ทำการ: สำนักบริหารแรงงานต่างด้าว")
-    
-    y_pos -= 25
-    c.drawString(50, y_pos, "วันที่: 01 เมษายน 2568")
-    
-    y_pos -= 25
-    c.drawString(50, y_pos, f"เลขที่ใบชำระเงิน: {payment_number}")
-    
-    y_pos -= 25
-    c.drawString(50, y_pos, f"เลขรับคำขอที่: {request_number}")
-    
-    # ข้อมูลผู้ชำระเงิน
-    y_pos -= 40
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(50, y_pos, "ข้อมูลผู้ชำระเงิน:")
-    
-    y_pos -= 25
-    c.setFont("Helvetica", 12)
-    c.drawString(50, y_pos, f"ชื่อผู้ชำระเงิน: {english_name}")
-    
-    y_pos -= 20
-    c.drawString(50, y_pos, f"สัญชาติ: {nationality}")
-    
-    y_pos -= 20
-    c.drawString(50, y_pos, f"เลขอ้างอิงคนต่างด้าว: {foreign_reference_number}")
-    
-    y_pos -= 20
-    c.drawString(50, y_pos, f"หมายเลขประจำตัวคนต่างด้าว: {id_number}")
-    
-    # ข้อมูลนายจ้าง
-    y_pos -= 40
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(50, y_pos, "ข้อมูลนายจ้าง:")
-    
-    y_pos -= 25
-    c.setFont("Helvetica", 12)
-    c.drawString(50, y_pos, "ชื่อนายจ้าง/สถานประกอบการ: บริษัท บาน กง เอ็นจิเนียริ่ง จำกัด")
-    
-    y_pos -= 20
-    c.drawString(50, y_pos, "เลขประจำตัวนายจ้าง: 0415567000061")
-    
-    # รายการค่าธรรมเนียม
-    y_pos -= 40
-    c.setFont("Helvetica-Bold", 14)
-    c.drawString(50, y_pos, "รายการค่าธรรมเนียม")
-    
-    # วาดเส้นใต้หัวข้อ
-    c.line(50, y_pos-5, 500, y_pos-5)
-    
-    y_pos -= 30
-    c.setFont("Helvetica", 12)
-    c.drawString(70, y_pos, "1. ค่าธรรมเนียมในการยื่นคำขอ ฉบับละ 100 บาท")
-    c.drawString(450, y_pos, "100.00")
-    
-    y_pos -= 25
-    c.drawString(70, y_pos, "2. ค่าธรรมเนียมการพิจารณา")
-    c.drawString(450, y_pos, "900.00")
-    
-    # เส้นคั่น
-    y_pos -= 15
-    c.line(400, y_pos, 500, y_pos)
-    
-    y_pos -= 25
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(70, y_pos, "รวมเป็นเงินทั้งสิ้น (บาท): (หนึ่งพันบาทถ้วน)")
-    c.drawString(430, y_pos, "1,000.00")
-    
-    # เส้นใต้ยอดรวม
-    c.line(400, y_pos-5, 500, y_pos-5)
-    c.line(400, y_pos-8, 500, y_pos-8)
-    
-    # ข้อความรับรอง
-    y_pos -= 40
-    c.setFont("Helvetica", 12)
-    c.drawString(50, y_pos, "ได้รับเงินไว้เป็นการถูกต้องแล้ว")
-    
-    # ลายเซ็น
-    y_pos -= 60
-    c.drawString(300, y_pos, "(ลงชื่อ) นางสาวอารีวรรณ โพธิ์นิ่มแดง")
-    c.drawString(350, y_pos-20, "(ผู้รับเงิน)")
-    
-    y_pos -= 45
-    c.drawString(300, y_pos, "ตำแหน่ง: นักวิชาการแรงงานชำนาญการ")
-    
-    # QR Code ที่มุมขวาล่าง
+    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.QRCode(version=1, box_size=8, border=2)
-        qr.add_data(pdf_url)
-        qr.make(fit=True)
-        qr_img = qr.make_image(fill_color="black", back_color="white")
-        qr_img.save("temp_qr.png")
-        c.drawImage("temp_qr.png", width-150, 50, width=80, height=80)
+        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}")
     
-    # Timestamp ที่ด้านล่าง
-    c.setFont("Helvetica", 8)
-    print_time = thai_date_time(time.time())
-    c.drawString(50, 30, f"พิมพ์เมื่อ: {print_time}")
+    # **เพิ่ม print timestamp ด้านล่าง - เหมือนต้นฉบับ**
+    if font_available:
+        c.setFont("THSarabun", 20/2.83465)
+    else:
+        c.setFont("Helvetica", 7)
+    
+    # แปลงเวลาเป็นภาษาไทย
+    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.rect(30, 30, width-60, height-60, stroke=1, fill=0)
+    c.drawString(2*cm, 2*cm, f"พิมพ์เมื่อ: {thai_time}")
     
     c.showPage()
     c.save()
     
-    # เพิ่ม HTML version สำหรับ preview และ print
-    create_html_receipt(request_number, english_name, foreign_reference_number, 
-                       id_number, nationality, receipt_number, payment_number)
-    
-    # อัพโหลด PDF
+    # อัพโหลดไฟล์
     try:
         upload_file(
-            path_or_fileobj=local_path,
+            path_or_fileobj=filename,
             path_in_repo=filename,
             repo_id=REPO_ID,
             repo_type="space"
         )
-        print(f"PDF uploaded successfully: {filename}")
-    except Exception as e:
-        print(f"Error uploading PDF: {e}")
+        return pdf_url
+    except:
         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 หลักเท่านั้น"
     
-    # บันทึกข้อมูล PDF ในฐานข้อมูล
+    # สร้างเลขคำขอ
     conn = sqlite3.connect("workers.db")
-    cursor = conn.cursor()
-    cursor.execute("INSERT OR REPLACE INTO pdfs (request_number, pdf_url) VALUES (?, ?)",
-                  (request_number, pdf_url))
-    conn.commit()
+    c = conn.cursor()
+    c.execute("SELECT COUNT(*) FROM workers")
+    count = c.fetchone()[0]
     conn.close()
     
-    return pdf_url
-
-# เพิ่มฟังก์ชันสร้าง HTML version
-def create_html_receipt(request_number, english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number):
-    html_filename = f"receipts/receipt_{request_number}.html"
-    local_html_path = f"temp_{request_number}.html"
+    request_number = f"WP-68-366-{150 + count:03d}"
     
-    os.makedirs("receipts", exist_ok=True)
+    # บันทึกข้อมูล
+    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()
     
-    pdf_url = f"https://huggingface.co/spaces/{REPO_ID}/resolve/main/pdfs/receipt_{request_number}.pdf"
-    print_time = thai_date_time(time.time())
+    # สร้าง PDF
+    pdf_url = create_pdf(request_number, english_name, foreign_reference_number, 
+                        id_number, nationality, receipt_number, payment_number)
     
-    html_content = f"""
-<!DOCTYPE html>
-<html lang="th">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>ใบเสร็จรับเงิน - {request_number}</title>
-    <script src="https://cdn.jsdelivr.net/npm/qrcode@1.5.3/build/qrcode.min.js"></script>
-    <style>
-        @import url('https://fonts.googleapis.com/css2?family=Sarabun:wght@400;600;700&display=swap');
-        
-        * {{
-            margin: 0;
-            padding: 0;
-            box-sizing: border-box;
-        }}
-        
-        body {{
-            font-family: 'Sarabun', Arial, sans-serif;
-            background: #f5f5f5;
-            padding: 20px;
-        }}
-        
-        .receipt-container {{
-            max-width: 800px;
-            margin: 0 auto;
-            background: white;
-            border: 2px solid #000;
-            padding: 30px;
-            box-shadow: 0 0 20px rgba(0,0,0,0.1);
-        }}
-        
-        .header {{
-            text-align: center;
-            margin-bottom: 30px;
-            border-bottom: 2px solid #000;
-            padding-bottom: 20px;
-        }}
-        
-        .header h1 {{
-            font-size: 24px;
-            font-weight: 700;
-            margin-bottom: 10px;
-        }}
-        
-        .header h2 {{
-            font-size: 18px;
-            font-weight: 600;
-        }}
-        
-        .info-section {{
-            margin-bottom: 25px;
-        }}
-        
-        .info-row {{
-            display: flex;
-            justify-content: space-between;
-            margin-bottom: 8px;
-            padding: 5px 0;
-        }}
-        
-        .info-label {{
-            font-weight: 600;
-        }}
-        
-        .section-title {{
-            font-size: 16px;
-            font-weight: 700;
-            margin: 20px 0 15px 0;
-            border-bottom: 1px solid #ccc;
-            padding-bottom: 5px;
-        }}
-        
-        .fee-table {{
-            width: 100%;
-            border-collapse: collapse;
-            margin: 20px 0;
-        }}
-        
-        .fee-table th,
-        .fee-table td {{
-            border: 1px solid #000;
-            padding: 10px;
-            text-align: left;
-        }}
-        
-        .fee-table th {{
-            background: #f0f0f0;
-            font-weight: 700;
-        }}
-        
-        .amount {{
-            text-align: right;
-            font-weight: 600;
-        }}
-        
-        .total-row {{
-            background: #f9f9f9;
-            font-weight: 700;
-        }}
-        
-        .signature-section {{
-            margin-top: 40px;
-            display: flex;
-            justify-content: space-between;
-            align-items: flex-end;
-        }}
-        
-        .signature-box {{
-            text-align: center;
-        }}
-        
-        .qr-section {{
-            display: flex;
-            flex-direction: column;
-            align-items: center;
-        }}
-        
-        .footer {{
-            margin-top: 30px;
-            text-align: center;
-            font-size: 12px;
-            color: #666;
-            border-top: 1px solid #ccc;
-            padding-top: 15px;
-        }}
-        
-        .download-section {{
-            text-align: center;
-            margin: 20px 0;
-            padding: 15px;
-            background: #e3f2fd;
-            border-radius: 8px;
-            border: 1px solid #2196f3;
-        }}
-        
-        .download-btn {{
-            background: #2196f3;
-            color: white;
-            padding: 10px 20px;
-            border: none;
-            border-radius: 5px;
-            cursor: pointer;
-            font-size: 16px;
-            margin: 5px;
-            text-decoration: none;
-            display: inline-block;
-        }}
-        
-        .download-btn:hover {{
-            background: #1976d2;
-        }}
-        
-        /* Print Styles */
-        @media print {{
-            body {{
-                background: white;
-                padding: 0;
-            }}
-            
-            .receipt-container {{
-                box-shadow: none;
-                border: 2px solid #000;
-                max-width: none;
-                width: 100%;
-            }}
-            
-            .download-section {{
-                display: none;
-            }}
-            
-            .no-print {{
-                display: none;
-            }}
-            
-            /* ปรับขนาดสำหรับการพิมพ์ */
-            .header h1 {{
-                font-size: 22px;
-            }}
-            
-            .header h2 {{
-                font-size: 16px;
-            }}
-            
-            .info-row {{
-                font-size: 14px;
-            }}
-            
-            .fee-table {{
-                font-size: 14px;
-            }}
-        }}
-        
-        @page {{
-            margin: 1cm;
-            size: A4;
-        }}
-    </style>
-</head>
-<body>
-    <div class="receipt-container">
-        <!-- Header -->
-        <div class="header">
-            <h1>กระทรวงแรงงาน</h1>
-            <h2>ใบเสร็จรับเงิน (ต้นฉบับ)</h2>
-        </div>
-        
-        <!-- Download Section (ไม่แสดงเวลาพิมพ์) -->
-        <div class="download-section no-print">
-            <h3>🔗 ดาวน์โหลดเอกสาร</h3>
-            <a href="{pdf_url}" class="download-btn" target="_blank">📄 ดาวน์โหลด PDF</a>
-            <button onclick="window.print()" class="download-btn">🖨️ พิมพ์เอกสาร</button>
-        </div>
-        
-        <!-- Receipt Info -->
-        <div class="info-section">
-            <div class="info-row">
-                <span class="info-label">เลขที่:</span>
-                <span>{receipt_number}</span>
-                <span class="info-label">แบบ ภ.1600-034</span>
-            </div>
-            <div class="info-row">
-                <span class="info-label">ที่ทำการ:</span>
-                <span>สำนักบริหารแรงงานต่างด้าว</span>
-            </div>
-            <div class="info-row">
-                <span class="info-label">วันที่:</span>
-                <span>01 เมษายน 2568</span>
-            </div>
-            <div class="info-row">
-                <span class="info-label">เลขที่ใบชำระเงิน:</span>
-                <span>{payment_number}</span>
-            </div>
-            <div class="info-row">
-                <span class="info-label">เลขรับคำขอที่:</span>
-                <span>{request_number}</span>
-            </div>
-        </div>
-        
-        <!-- Payer Info -->
-        <div class="section-title">ข้อมูลผู้ชำระเงิน</div>
-        <div class="info-section">
-            <div class="info-row">
-                <span class="info-label">ชื่อผู้ชำระเงิน:</span>
-                <span>{english_name}</span>
-            </div>
-            <div class="info-row">
-                <span class="info-label">สัญชาติ:</span>
-                <span>{nationality}</span>
-            </div>
-            <div class="info-row">
-                <span class="info-label">เลขอ้างอิงคนต่างด้าว:</span>
-                <span>{foreign_reference_number}</span>
-            </div>
-            <div class="info-row">
-                <span class="info-label">หมายเลขประจำตัวคนต่างด้าว:</span>
-                <span>{id_number}</span>
-            </div>
-        </div>
-        
-        <!-- Employer Info -->
-        <div class="section-title">ข้อมูลนายจ้าง</div>
-        <div class="info-section">
-            <div class="info-row">
-                <span class="info-label">ชื่อนายจ้าง/สถานประกอบการ:</span>
-                <span>บริษัท บาน กง เอ็นจิเนียริ่ง จำกัด</span>
-            </div>
-            <div class="info-row">
-                <span class="info-label">เลขประจำตัวนายจ้าง:</span>
-                <span>0415567000061</span>
-            </div>
-        </div>
-        
-        <!-- Fee Table -->
-        <div class="section-title">รายการค่าธรรมเนียม</div>
-        <table class="fee-table">
-            <thead>
-                <tr>
-                    <th>รายการ</th>
-                    <th class="amount">จำนวนเงิน (บาท)</th>
-                </tr>
-            </thead>
-            <tbody>
-                <tr>
-                    <td>1. ค่าธรรมเนียมในการยื่นคำขอ ฉบับละ 100 บาท</td>
-                    <td class="amount">100.00</td>
-                </tr>
-                <tr>
-                    <td>2. ค่าธรรมเนียมการพิจารณา</td>
-                    <td class="amount">900.00</td>
-                </tr>
-                <tr class="total-row">
-                    <td><strong>รวมเป็นเงินทั้งสิ้น (บาท): (หนึ่งพันบาทถ้วน)</strong></td>
-                    <td class="amount"><strong>1,000.00</strong></td>
-                </tr>
-            </tbody>
-        </table>
-        
-        <div style="margin: 20px 0; font-weight: 600;">
-            ได้รับเงินไว้เป็นการถูกต้องแล้ว
-        </div>
-        
-        <!-- Signature Section -->
-        <div class="signature-section">
-            <div class="signature-box">
-                <div style="margin-bottom: 50px;">_______________________</div>
-                <div>(ลงชื่อ) นางสาวอารีวรรณ โพธิ์นิ่มแดง</div>
-                <div>(ผู้รับเงิน)</div>
-                <div style="margin-top: 10px;">ตำแหน่ง: นักวิชาการแรงงานชำนาญการ</div>
-            </div>
-            
-            <div class="qr-section">
-                <div id="qrcode"></div>
-                <div style="font-size: 12px; margin-top: 5px;">สแกนเพื่อตรวจสอบ</div>
-            </div>
-        </div>
-        
-        <!-- Footer -->
-        <div class="footer">
-            <div>พิมพ์เมื่อ: {print_time}</div>
-            <div style="margin-top: 5px;">เอกสารนี้ออกโดยระบบอัตโนมัติ - บริษัท บาน กง เอ็นจิเนียริ่ง จำกัด</div>
-        </div>
-    </div>
+    if pdf_url:
+        result = f"""✅ สร้างใบเสร็จสำเร็จ!
+
+📋 เลขคำขอ: {request_number}
+👤 ชื่อ: {english_name}
+🆔 หมายเลขประจำตัว: {id_number}
+
+📄 ดาวน์โหลด 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()
     
-    <script>
-        // สร้าง QR Code
-        QRCode.toCanvas(document.getElementById('qrcode'), '{pdf_url}', {{
-            width: 80,
-            height: 80,
-            margin: 1,
-            color: {{
-                dark: '#000000',
-                light: '#FFFFFF'
-            }}
-        }}, function (error) {{
-            if (error) console.error(error);
-            console.log('QR Code generated successfully!');
-        }});
-        
-        // Auto-focus สำหรับการพิมพ์
-        window.addEventListener('load', function() {{
-            console.log('Receipt loaded successfully');
-        }});
-    </script>
-</body>
-</html>
-    """
+    if not workers:
+        return "ยังไม่มีข้อมูลพนักงานในระบบ"
     
-    # บันทึกไฟล์ HTML
-    with open(local_html_path, 'w', encoding='utf-8') as f:
-        f.write(html_content)
+    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"
     
-    # อัพโหลด HTML
-    try:
-        upload_file(
-            path_or_fileobj=local_html_path,
-            path_in_repo=html_filename,
-            repo_id=REPO_ID,
-            repo_type="space"
-        )
-        print(f"HTML uploaded successfully: {html_filename}")
-    except Exception as e:
-        print(f"Error uploading HTML: {e}")
+    return result
+
+init_db()
+
+with gr.Blocks(title="ระบบจัดการพนักงาน") as app:
+    gr.Markdown("# 🏢 ระบบจัดการข้อมูลพนักงานต่างด้าว")
+    gr.Markdown("### บริษัท บาน กง เอ็นจิเนียริ่ง จำกัด")
+    gr.Markdown("#### ✅ ใบเสร็จจะมีหน้าตาเหมือนเอกสารต้นฉบับทุกประการ")
+    
+    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)
+    
+    with gr.Tab("📋 ดูรายการ"):
+        view_btn = gr.Button("👥 ดูรายการพนักงานทั้งหมด")
+        worker_list = gr.Textbox(label="รายการ", lines=15, interactive=False)
+    
+    submit.click(add_worker, [name, ref_num, id_num, nation, receipt, payment], result)
+    view_btn.click(view_all, outputs=worker_list)
+
+app.launch()