Update app.py

app.py

@@ -1,3 +1,344 @@
+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: