Update app.py

app.py

@@ -15,19 +15,19 @@ import io
 from dotenv import load_dotenv
 import time
 
-# อ่าน .env (ถ้ามี)
+# Load .env
 load_dotenv()
 REPO_ID = os.getenv("REPO_ID", "your-username/worker-management")
 ORIGINAL_REPO_ID = os.getenv("ORIGINAL_REPO_ID", None)
 
-# ตั้งค่าฟอนต์
+# Register fonts
 try:
     pdfmetrics.registerFont(TTFont('THSarabun', 'THSarabunNew.ttf'))
     pdfmetrics.registerFont(TTFont('THSarabunBold', 'THSarabunNew-Bold.ttf'))
 except Exception as e:
     raise Exception(f"Error loading fonts: {e}")
 
-# ฟังก์ชันแปลงวันที่เป็นภาษาไทย
+# Convert timestamp to Thai format
 def thai_date_time(timestamp):
     months = {
         1: "มกราคม", 2: "กุมภาพันธ์", 3: "มีนาคม", 4: "เมษายน",
@@ -43,7 +43,7 @@ def thai_date_time(timestamp):
     period = "น"
     return f"{day:02d} {month} {year} {hour:02d}:{minute:02d} {period}"
 
-# เริ่มต้นฐานข้อมูล
+# Initialize database
 def init_db():
     conn = sqlite3.connect("workers.db")
     c = conn.cursor()
@@ -69,7 +69,7 @@ def init_db():
     conn.commit()
     conn.close()
 
-# นำเข้า JSON
+# Import JSON data
 def import_json_to_db():
     try:
         json_path = hf_hub_download(repo_id=REPO_ID, filename="worker-database.json", repo_type="space")
@@ -95,14 +95,14 @@ def import_json_to_db():
     except Exception as e:
         print(f"Error importing JSON: {e}")
 
-# สำรองฐานข้อมูล
+# Backup database
 def backup_db():
     try:
         hf_hub_upload(repo_id=REPO_ID, path_in_repo="workers.db", path_or_fileobj="workers.db", repo_type="space")
     except Exception as e:
         print(f"Error backing up database: {e}")
 
-# ดึงเลขคำขอถัดไป
+# Generate next request number
 def get_next_request_number():
     conn = sqlite3.connect("workers.db")
     c = conn.cursor()
@@ -116,20 +116,31 @@ def get_next_request_number():
     max_num = max([int(num[0].split('-')[-1]) for num in request_numbers])
     return f"WP-68-366-{max_num + 1:03d}"
 
-# สร้าง PDF
+# Create PDF with background
 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"
     c = canvas.Canvas(local_path, pagesize=A4)
     
-    # ใช้สไตล์จาก main (5).css
+    # Add background image (from index1.html)
+    try:
+        # Try original Space first
+        if ORIGINAL_REPO_ID:
+            bg_path = hf_hub_download(repo_id=ORIGINAL_REPO_ID, filename="assets/bg.png", repo_type="space")
+        else:
+            bg_path = hf_hub_download(repo_id=REPO_ID, filename="assets/bg.png", repo_type="space")
+        c.drawImage(bg_path, 0, 0, width=A4[0], height=A4[1], preserveAspectRatio=True)
+    except Exception as e:
+        print(f"Error loading background image (bg.png): {e}")
+    
+    # Apply styles from main (5).css
     c.setFont("THSarabunBold", 52.284 / 2.83465)  # fs0
     c.drawString(2*cm, 28*cm, "กระทรวงแรงงาน")
     c.setFont("THSarabun", 35.212 / 2.83465)  # fs3
     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, 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}")
@@ -148,34 +159,34 @@ def create_receipt_pdf(request_number, english_name, foreign_reference_number, i
     c.drawString(2*cm, 17.5*cm, "(ลงชื่อ) นางสาวอารีวรรณ โพธิ์นิ่มแดง (ผู้รับเงิน)")
     c.drawString(2*cm, 17*cm, "ตำแหน่ง: นักวิชาการแรงงานชำนาญการ")
     
-    # QR Code (ใช้ URL ถาวร)
-    pdf_url = f"https://huggingface.co/spaces/{REPO_ID}/raw/main/{filename}"
+    # QR Code
+    pdf_url = f"[invalid url, do not cite]
     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)
     
-    # เพิ่ม print timestamp ที่ด้านล่าง
-    c.setFont("THSarabun", 20 / 2.83465)  # ขนาดเล็กสำหรับ footer
+    # Add print timestamp
+    c.setFont("THSarabun", 20 / 2.83465)
     print_time = thai_date_time(time.time())
     c.drawString(2*cm, 2*cm, f"พิมพ์เมื่อ: {print_time}")
     
     c.showPage()
     c.save()
     
-    # อัพโหลด PDF ไปยัง Space ใหม่
+    # Upload PDF
     try:
         hf_hub_upload(repo_id=REPO_ID, path_in_repo=filename, path_or_fileobj=local_path, repo_type="space")
     except Exception as e:
         print(f"Error uploading PDF to Space: {e}")
     
-    # อัพโหลดไปยัง Space เดิม (ถ้ามี)
+    # Upload to original Space (if specified)
     if ORIGINAL_REPO_ID:
         try:
             hf_hub_upload(repo_id=ORIGINAL_REPO_ID, path_in_repo=filename, path_or_fileobj=local_path, repo_type="space")
         except Exception as e:
             print(f"Error uploading to original Space: {e}")
     
-    # บันทึกข้อมูล PDF
+    # Save PDF info
     conn = sqlite3.connect("workers.db")
     c = conn.cursor()
     c.execute("INSERT OR REPLACE INTO pdfs (request_number, pdf_url) VALUES (?, ?)",
@@ -183,9 +194,9 @@ def create_receipt_pdf(request_number, english_name, foreign_reference_number, i
     conn.commit()
     conn.close()
     
-    return filename  # คืนค่า pdfs/receipt_xxx.pdf
+    return filename
 
-# ตรวจสอบข้อมูล
+# Validate input
 def validate_input(english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number):
     if not id_number or not id_number.isdigit() or len(id_number) != 13:
         return "หมายเลขประจำตัวต้องเป็นตัวเลข 13 หลัก"
@@ -193,7 +204,7 @@ def validate_input(english_name, foreign_reference_number, id_number, nationalit
         return "กรุณากรอกเลขที่ใบเสร็จและหมายเลขชำระเงิน"
     return None
 
-# เพิ่มข้อมูลพนักงาน
+# Add worker
 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:
@@ -201,7 +212,7 @@ def add_worker(english_name, foreign_reference_number, id_number, nationality, r
     
     request_number = get_next_request_number()
     
-    # บันทึกข้อมูล
+    # Save data
     conn = sqlite3.connect("workers.db")
     c = conn.cursor()
     c.execute('''INSERT INTO workers (
@@ -210,13 +221,13 @@ def add_worker(english_name, foreign_reference_number, id_number, nationality, r
         request_number, english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number
     ))
     
-    # จัดการไฟล์แนบ
+    # Handle attachments
     attachment_urls = []
     for i, file_data in enumerate(attachments or []):
         filename = f"attachments/file_{request_number}_{i}_{os.path.basename(file_data.name)}"
         try:
             hf_hub_upload(repo_id=REPO_ID, path_in_repo=filename, path_or_fileobj=file_data, repo_type="space")
-            url = f"https://huggingface.co/spaces/{REPO_ID}/raw/main/{filename}"
+            url = f"[invalid url, do not cite]
             c.execute("INSERT INTO attachments (request_number, file_url) VALUES (?, ?)", (request_number, url))
             attachment_urls.append(url)
         except Exception as e:
@@ -225,15 +236,15 @@ def add_worker(english_name, foreign_reference_number, id_number, nationality, r
     conn.commit()
     conn.close()
     
-    # สร้าง PDF
+    # Create PDF
     pdf_path = create_receipt_pdf(request_number, english_name, foreign_reference_number, id_number, nationality, receipt_number, payment_number)
     
-    # สำรองฐานข้อมูล
+    # Backup database
     backup_db()
     
     return f"เพิ่มข้อมูลพนักงานสำเร็จ! เลขคำขอ: {request_number}\nPDF: {pdf_path}\nไฟล์แนบ: {', '.join(attachment_urls) if attachment_urls else 'ไม่มีไฟล์แนบ'}"
 
-# ดาวน์โหลด PDF ทั้งหมด
+# Download all PDFs
 def download_all_pdfs():
     conn = sqlite3.connect("workers.db")
     c = conn.cursor()
@@ -261,11 +272,11 @@ def download_all_pdfs():
     
     return f"pdfs/all_receipts.zip"
 
-# เริ่มต้นฐานข้อมูลและนำเข้า JSON
+# Initialize database and import JSON
 init_db()
 import_json_to_db()
 
-# อินเทอร์เฟซ Gradio
+# Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("## เพิ่มข้อมูลพนักงานใหม่")
     with gr.Row():