app.py

import gradio as gr
import sqlite3
import json
from huggingface_hub import hf_hub_download, upload_file
import os
from pathlib import Path
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
import qrcode
import zipfile
import io
from dotenv import load_dotenv
import time

# อ่าน .env (ถ้ามี)
load_dotenv()
REPO_ID = os.getenv("REPO_ID", "protae5544/WorkerManagement")
ORIGINAL_REPO_ID = os.getenv("ORIGINAL_REPO_ID", None)

# ตั้งค่าฟอนต์
try:
    pdfmetrics.registerFont(TTFont('THSarabun', 'THSarabunNew.ttf'))
    pdfmetrics.registerFont(TTFont('THSarabunBold', 'THSarabunNew-Bold.ttf'))
except Exception as e:
    print(f"Warning: Could not load fonts: {e}")

# ฟังก์ชันแปลงวันที่เป็นภาษาไทย
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()
    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
    )''')
    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 backup_db():
    try:
        upload_file(
            path_or_fileobj="workers.db",
            path_in_repo="workers.db",
            repo_id=REPO_ID,
            repo_type="space"
        )
    except Exception as e:
        print(f"Error backing up database: {e}")

# ดึงเลขคำขอถัดไป
def get_next_request_number():
    conn = sqlite3.connect("workers.db")
    c = conn.cursor()
    c.execute("SELECT request_number FROM workers")
    request_numbers = c.fetchall()
    conn.close()
    
    if not request_numbers:
        return "WP-68-366-150"
    
    try:
        max_num = max([int(num[0].split('-')[-1]) for num in request_numbers if num[0].startswith('WP-68-366-')])
        return f"WP-68-366-{max_num + 1:03d}"
    except:
        return "WP-68-366-150"

# สร้าง PDF
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"
    
    # สร้างโฟลเดอร์ pdfs ถ้าไม่มี
    os.makedirs("pdfs", exist_ok=True)
    
    c = canvas.Canvas(local_path, pagesize=A4)
    
    # ใช้ฟอนต์ไทยถ้ามี ไม่งั้นใช้ Helvetica
    try:
        c.setFont("THSarabunBold", 18)
        font_available = True
    except:
        c.setFont("Helvetica-Bold", 16)
        font_available = False
    
    # Header
    c.drawString(2*cm, 28*cm, "กระทรวงแรงงาน" if font_available else "Ministry of Labour")
    
    try:
        c.setFont("THSarabun", 14) if font_available else c.setFont("Helvetica", 12)
    except:
        c.setFont("Helvetica", 12)
        
    c.drawString(2*cm, 27.5*cm, "ใบเสร็จรับเงิน (ต้นฉบับ)" if font_available else "Receipt (Original)")
    c.drawString(2*cm, 26*cm, f"เลขที่: {receipt_number}" if font_available else f"Number: {receipt_number}")
    c.drawString(2*cm, 25.5*cm, "ที่ทำการ: สำนักบริหารแรงงานต่างด้าว" if font_available else "Office: Bureau of Foreign Workers Administration")
    c.drawString(2*cm, 25*cm, "วันที่: 01 เมษายน 2568" if font_available else "Date: 01 April 2025")
    c.drawString(2*cm, 24.5*cm, f"เลขที่ใบชำระเงิน: {payment_number}" if font_available else f"Payment Number: {payment_number}")
    c.drawString(2*cm, 24*cm, f"เลขรับคำขอที่: {request_number}" if font_available else f"Request Number: {request_number}")
    c.drawString(2*cm, 23.5*cm, f"ชื่อผู้ชำระเงิน: {english_name}" if font_available else f"Payer Name: {english_name}")
    c.drawString(2*cm, 23*cm, f"สัญชาติ: {nationality}" if font_available else f"Nationality: {nationality}")
    c.drawString(2*cm, 22.5*cm, f"เลขอ้างอิงคนต่างด้าว: {foreign_reference_number}" if font_available else f"Foreign Reference: {foreign_reference_number}")
    c.drawString(2*cm, 22*cm, f"หมายเลขประจำตัวคนต่างด้าว: {id_number}" if font_available else f"Foreign Worker ID: {id_number}")
    c.drawString(2*cm, 21.5*cm, "ชื่อนายจ้าง/สถานประกอบการ: บริษัท บาน กง เอ็นจิเนียริ่ง จำกัด" if font_available else "Employer: Ban Kong Engineering Co., Ltd.")
    c.drawString(2*cm, 21*cm, "เลขประจำตัวนายจ้าง: 0415567000061" if font_available else "Employer ID: 0415567000061")
    
    # รายการ
    try:
        c.setFont("THSarabunBold", 14) if font_available else c.setFont("Helvetica-Bold", 12)
    except:
        c.setFont("Helvetica-Bold", 12)
        
    c.drawString(2*cm, 20*cm, "รายการ" if font_available else "Items")
    
    try:
        c.setFont("THSarabun", 12) if font_available else c.setFont("Helvetica", 10)
    except:
        c.setFont("Helvetica", 10)
        
    c.drawString(2*cm, 19.5*cm, "1. ค่าธรรมเนียมในการยื่นคำขอ ฉบับละ 100 บาท: 100.00" if font_available else "1. Application fee 100 Baht each: 100.00")
    c.drawString(2*cm, 19*cm, "2. ค่าธรรมเนียม: 900.00" if font_available else "2. Service fee: 900.00")
    c.drawString(2*cm, 18.5*cm, "รวมเป็นเงินทั้งสิ้น (บาท): (หนึ่งพันบาทถ้วน) 1,000.00" if font_available else "Total (Baht): (One thousand Baht) 1,000.00")
    c.drawString(2*cm, 18*cm, "ได้รับเงินไว้เป็นการถูกต้องแล้ว" if font_available else "Payment received correctly")
    c.drawString(2*cm, 17.5*cm, "(ลงชื่อ) นางสาวอารีวรรณ โพธิ์นิ่มแดง (ผู้รับเงิน)" if font_available else "(Signature) Ms. Areewan Pho Nim Daeng")
    c.drawString(2*cm, 17*cm, "ตำแหน่ง: นักวิชาการแรงงานชำนาญการ" if font_available else "Position: Senior Labor Academic")
    
    # QR Code
    pdf_url = f"https://huggingface.co/spaces/{REPO_ID}/resolve/main/{filename}"
    try:
        qr = qrcode.QRCode(version=1, box_size=10, border=5)
        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", 15*cm, 15*cm, width=3*cm, height=3*cm)
    except Exception as e:
        print(f"Error creating QR code: {e}")
    
    # Timestamp
    try:
        c.setFont("THSarabun", 8) if font_available else c.setFont("Helvetica", 8)
    except:
        c.setFont("Helvetica", 8)
        
    print_time = thai_date_time(time.time())
    c.drawString(2*cm, 2*cm, f"พิมพ์เมื่อ: {print_time}" if font_available else f"Printed: {print_time}")
    
    c.showPage()
    c.save()
    
    # อัพโหลด PDF
    try:
        upload_file(
            path_or_fileobj=local_path,
            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 None
    
    # บันทึกข้อมูล 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()
    conn.close()
    
    return pdf_url

# ตรวจสอบข้อมูล
def validate_input(english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number):
    if not english_name or not english_name.strip():
        return "กรุณากรอกชื่อภาษาอังกฤษ"
    if not foreign_reference_number or not foreign_reference_number.strip():
        return "กรุณากรอกหมายเลขอ้างอิงคนต่างด้าว"
    if not id_number or not id_number.isdigit() or len(id_number) != 13:
        return "หมายเลขประจำตัวต้องเป็นตัวเลข 13 หลัก"
    if not nationality or not nationality.strip():
        return "กรุณากรอกสัญชาติ"
    if not receipt_number or not receipt_number.strip():
        return "กรุณากรอกเลขที่ใบเสร็จ"
    if not payment_number or not payment_number.strip():
        return "กรุณากรอกหมายเลขชำระเงิน"
    return None

# เพิ่มข้อมูลพนักงาน
def add_worker(english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number, attachments):
    # ตรวจสอบข้อมูล
    error = validate_input(english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number)
    if error:
        return f"❌ {error}"
    
    try:
        request_number = get_next_request_number()
        
        # บันทึกข้อมูลในฐานข้อมูล
        conn = sqlite3.connect("workers.db")
        c = conn.cursor()
        c.execute('''INSERT INTO workers (
            request_number, english_name, foreign_reference_number, id_number, 
            nationality, receipt_number, payment_number, created_at
        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)''', (
            request_number, english_name, foreign_reference_number, id_number, 
            nationality, receipt_number, payment_number, time.time()
        ))
        
        # จัดการไฟล์แนบ
        attachment_urls = []
        if attachments:
            for i, file_data in enumerate(attachments):
                if file_data is not None:
                    try:
                        filename = f"attachments/file_{request_number}_{i}_{os.path.basename(file_data.name)}"
                        upload_file(
                            path_or_fileobj=file_data.name,
                            path_in_repo=filename,
                            repo_id=REPO_ID,
                            repo_type="space"
                        )
                        url = f"https://huggingface.co/spaces/{REPO_ID}/resolve/main/{filename}"
                        c.execute("INSERT INTO attachments (request_number, file_url) VALUES (?, ?)", 
                                (request_number, url))
                        attachment_urls.append(url)
                    except Exception as e:
                        print(f"Error uploading attachment: {e}")
        
        conn.commit()
        conn.close()
        
        # สร้าง PDF
        pdf_url = create_receipt_pdf(request_number, english_name, foreign_reference_number, 
                                   id_number, nationality, receipt_number, payment_number)
        
        if pdf_url:
            # สำรองฐานข้อมูล
            backup_db()
            
            result = f"✅ เพิ่มข้อมูลพนักงานสำเร็จ!\n"
            result += f"📋 เลขคำขอ: {request_number}\n"
            result += f"📄 PDF: {pdf_url}\n"
            if attachment_urls:
                result += f"📎 ไฟล์แนบ: {len(attachment_urls)} ไฟล์"
            else:
                result += f"📎 ไฟล์แนบ: ไม่มี"
            return result
        else:
            return f"❌ เกิดข้อผิดพลาดในการสร้าง PDF สำหรับเลขคำขอ: {request_number}"
            
    except Exception as e:
        return f"❌ เกิดข้อผิดพลาด: {str(e)}"

# ดาวน์โหลด PDF ทั้งหมด
def download_all_pdfs():
    try:
        conn = sqlite3.connect("workers.db")
        c = conn.cursor()
        c.execute("SELECT request_number, pdf_url FROM pdfs")
        pdfs = c.fetchall()
        conn.close()
        
        if not pdfs:
            return "❌ ไม่มี PDF ในระบบ"
        
        zip_buffer = io.BytesIO()
        with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
            for request_number, pdf_url in pdfs:
                try:
                    # ดาวน์โหลด PDF จาก Space
                    local_path = hf_hub_download(
                        repo_id=REPO_ID, 
                        filename=f"pdfs/receipt_{request_number}.pdf", 
                        repo_type="space"
                    )
                    zip_file.write(local_path, f"receipt_{request_number}.pdf")
                except Exception as e:
                    print(f"Error downloading PDF {request_number}: {e}")
                    continue
        
        zip_path = "all_receipts.zip"
        with open(zip_path, 'wb') as f:
            f.write(zip_buffer.getvalue())
        
        # อัพโหลด ZIP file
        upload_file(
            path_or_fileobj=zip_path,
            path_in_repo="downloads/all_receipts.zip",
            repo_id=REPO_ID,
            repo_type="space"
        )
        
        download_url = f"https://huggingface.co/spaces/{REPO_ID}/resolve/main/downloads/all_receipts.zip"
        return f"✅ สร้างไฟล์ ZIP สำเร็จ!\n📥 ดาวน์โหลด: {download_url}\n📊 จำนวน PDF: {len(pdfs)} ไฟล์"
        
    except Exception as e:
        return f"❌ เกิดข้อผิดพลาด: {str(e)}"

# ดูรายการพนักงาน
def get_worker_list():
    try:
        conn = sqlite3.connect("workers.db")
        c = conn.cursor()
        c.execute("""SELECT request_number, english_name, nationality, id_number, 
                           receipt_number, payment_number FROM workers ORDER BY request_number DESC""")
        workers = c.fetchall()
        conn.close()
        
        if not workers:
            return "ไม่มีข้อมูลพนักงานในระบบ"
        
        result = f"📋 รายการพนักงานทั้งหมด ({len(workers)} คน)\n" + "="*50 + "\n"
        for worker in workers:
            result += f"🆔 {worker[0]} | {worker[1]} | {worker[2]} | {worker[3]}\n"
        
        return result
    except Exception as e:
        return f"❌ เกิดข้อผิดพลาด: {str(e)}"

# เริ่มต้นฐานข้อมูล
init_db()

# UI Gradio
with gr.Blocks(
    theme=gr.themes.Soft(),
    title="ระบบจัดการข้อมูลพนักงานต่างด้าว",
    css="""
    .gradio-container {
        max-width: 1200px !important;
        margin: auto !important;
    }
    .form-container {
        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
        border-radius: 15px;
        padding: 20px;
        margin: 10px 0;
    }
    """
) as demo:
    
    gr.Markdown("""
    # 🏢 ระบบจัดการข้อมูลพนักงานต่างด้าว
    ### Worker Management System - Ban Kong Engineering Co., Ltd.
    """)
    
    with gr.Tab("📝 เพิ่มข้อมูลพนักงานใหม่"):
        gr.Markdown("### กรอกข้อมูลพนักงานใหม่")
        
        with gr.Row():
            english_name = gr.Textbox(
                label="🏷️ ชื่อภาษาอังกฤษ",
                placeholder="เช่น MR. AUNG KYAW",
                lines=1
            )
            foreign_reference_number = gr.Textbox(
                label="🔢 หมายเลขอ้างอิงคนต่างด้าว",
                placeholder="เช่น 2492102076212",
                lines=1
            )
        
        with gr.Row():
            id_number = gr.Textbox(
                label="🆔 หมายเลขประจำตัว (13 หลัก)",
                placeholder="เช่น 6685490000472",
                lines=1
            )
            nationality = gr.Dropdown(
                label="🌍 สัญชาติ",
                choices=["เมียนมา", "กัมพูชา", "ลาว", "เวียดนาม"],
                value="เมียนมา"
            )
        
        with gr.Row():
            receipt_number = gr.Textbox(
                label="🧾 เลขที่บนขวาใบเสร็จ",
                placeholder="เช่น 2100680001130",
                lines=1
            )
            payment_number = gr.Textbox(
                label="💳 หมายเลขชำระเงิน",
                placeholder="เช่น IV680210/002350",
                lines=1
            )
        
        attachments = gr.File(
            label="📎 แนบไฟล์ (สูงสุด 5MB ต่อไฟล์)",
            file_count="multiple",
            file_types=[".jpg", ".png", ".pdf", ".jpeg"],
            max_file_size="5MB"
        )
        
        submit_btn = gr.Button(
            "💾 บันทึกข้อมูลและสร้าง PDF",
            variant="primary",
            size="lg"
        )
        
        output = gr.Textbox(
            label="📋 ผลลัพธ์",
            lines=6,
            interactive=False
        )
    
    with gr.Tab("📊 จัดการข้อมูล"):
        gr.Markdown("### ดูและจัดการข้อมูลพนักงาน")
        
        with gr.Row():
            view_workers_btn = gr.Button("👥 ดูรายการพนักงานทั้งหมด", variant="secondary")
            download_all_btn = gr.Button("📥 ดาวน์โหลด PDF ทั้งหมด", variant="primary")
        
        management_output = gr.Textbox(
            label="📋 ผลลัพธ์",
            lines=10,
            interactive=False
        )
    
    with gr.Tab("ℹ️ คำแนะนำ"):
        gr.Markdown("""
        ## 📖 วิธีการใช้งาน
        
        ### 1. เพิ่มข้อมูลพนักงานใหม่
        - กรอกข้อมูลในฟอร์มให้ครบถ้วน
        - หมายเลขประจำตัวต้องเป็นตัวเลข 13 หลัก
        - สามารถแนบไฟล์เอกสารได้ (JPG, PNG, PDF)
        - ระบบจะสร้าง PDF ใบเสร็จพร้อม QR Code อัตโนมัติ
        
        ### 2. ดาวน์โหลด PDF
        - ใช้ QR Code ในใบเสร็จเพื่อดาวน์โหลด PDF แต่ละไฟล์
        - หรือดาวน์โหลด PDF ทั้งหมดเป็นไฟล์ ZIP
        
        ### 3. ข้อมูลที่เก็บ
        - ข้อมูลพนักงาน
        - ไฟล์ PDF ใบเสร็จ
        - ไฟล์แนบเอกสาร
        - ประวัติการทำรายการ
        
        ### 4. คุณสมบัติ
        - 🔒 ข้อมูลปลอดภัย บันทึกใน Hugging Face Space
        - 📱 QR Code สำหรับดาวน์โหลดใบเสร็จ
        - 📊 รายงานและสถิติ
        - 💾 สำรองข้อมูลอัตโนมัติ
        
        ---
        **🏢 พัฒนาโดย: Ban Kong Engineering Co., Ltd.**
        """)
    
    # Event handlers
    submit_btn.click(
        fn=add_worker,
        inputs=[english_name, foreign_reference_number, id_number, nationality, 
                receipt_number, payment_number, attachments],
        outputs=output
    )
    
    view_workers_btn.click(
        fn=get_worker_list,
        outputs=management_output
    )
    
    download_all_btn.click(
        fn=download_all_pdfs,
        outputs=management_output
    )

if __name__ == "__main__":
    demo.launch()