Update app.py

app.py

@@ -15,19 +15,19 @@ import io
 from dotenv import load_dotenv
 import time
 
-# Load .env
+# อ่าน .env
 load_dotenv()
-REPO_ID = os.getenv("REPO_ID", "your-username/worker-management")
+REPO_ID = os.getenv("REPO_ID", "protae5544/WorkerManagement")
 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}")
+    raise Exception(f"ข้อผิดพลาดในการโหลดฟอนต์: {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()
 
-# Import JSON data
+# นำเข้า JSON
 def import_json_to_db():
     try:
         json_path = hf_hub_download(repo_id=REPO_ID, filename="worker-database.json", repo_type="space")
@@ -93,16 +93,16 @@ def import_json_to_db():
         conn.commit()
         conn.close()
     except Exception as e:
-        print(f"Error importing JSON: {e}")
+        print(f"ข้อผิดพลาดในการนำเข้า 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}")
+        print(f"ข้อผิดพลาดในการสำรองฐานข้อมูล: {e}")
 
-# Generate next request number
+# ดึงเลขคำขอถัดไป
 def get_next_request_number():
     conn = sqlite3.connect("workers.db")
     c = conn.cursor()
@@ -116,24 +116,25 @@ 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}"
 
-# Create PDF with background
+# สร้าง 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"
     c = canvas.Canvas(local_path, pagesize=A4)
     
-    # Add background image (from index1.html)
+    # เพิ่มภาพพื้นหลัง (จาก index1.html)
     try:
-        # Try original Space first
+        # ลองดาวน์โหลดจาก Space เดิมก่อน
         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)
+        # ปรับตำแหน่งตาม x0: 58px, y0: 145px (แปลงเป็น cm: 58*0.0353 ≈ 2.05cm, 145*0.0353 ≈ 5.12cm)
+        c.drawImage(bg_path, 2.05*cm, A4[1] - (5.12*cm + 23.45*cm), width=17.62*cm, height=23.45*cm, preserveAspectRatio=True)
     except Exception as e:
-        print(f"Error loading background image (bg.png): {e}")
+        print(f"ข้อผิดพลาดในการโหลดภาพพื้นหลัง (bg.png): {e}")
     
-    # Apply styles from main (5).css
+    # ใช้สไตล์จาก 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
@@ -160,12 +161,12 @@ def create_receipt_pdf(request_number, english_name, foreign_reference_number, i
     c.drawString(2*cm, 17*cm, "ตำแหน่ง: นักวิชาการแรงงานชำนาญการ")
     
     # QR Code
-    pdf_url = f"[invalid url, do not cite]
+    pdf_url = f"https://huggingface.co/spaces/{REPO_ID}/raw/main/{filename}"
     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)
     
-    # Add print timestamp
+    # เพิ่ม print timestamp
     c.setFont("THSarabun", 20 / 2.83465)
     print_time = thai_date_time(time.time())
     c.drawString(2*cm, 2*cm, f"พิมพ์เมื่อ: {print_time}")
@@ -173,20 +174,20 @@ def create_receipt_pdf(request_number, english_name, foreign_reference_number, i
     c.showPage()
     c.save()
     
-    # Upload PDF
+    # อัพโหลด 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}")
+        print(f"ข้อผิดพลาดในการอัพโหลด PDF ไปยัง Space: {e}")
     
-    # Upload to original Space (if specified)
+    # อัพโหลดไปยัง Space เดิม (ถ้ามี)
     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}")
+            print(f"ข้อผิดพลาดในการอัพโหลดไปยัง Space เดิม: {e}")
     
-    # Save PDF info
+    # บันทึกข้อมูล PDF
     conn = sqlite3.connect("workers.db")
     c = conn.cursor()
     c.execute("INSERT OR REPLACE INTO pdfs (request_number, pdf_url) VALUES (?, ?)",
@@ -196,7 +197,7 @@ def create_receipt_pdf(request_number, english_name, foreign_reference_number, i
     
     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 หลัก"
@@ -204,7 +205,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:
@@ -212,7 +213,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 (
@@ -221,30 +222,30 @@ 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"[invalid url, do not cite]
+            url = f"https://huggingface.co/spaces/{REPO_ID}/raw/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}")
+            print(f"ข้อผิดพลาดในการอัพโหลดไฟล์แนบ: {e}")
     
     conn.commit()
     conn.close()
     
-    # Create PDF
+    # สร้าง 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 'ไม่มีไฟล์แนบ'}"
 
-# Download all PDFs
+# ดาวน์โหลด PDF ทั้งหมด
 def download_all_pdfs():
     conn = sqlite3.connect("workers.db")
     c = conn.cursor()
@@ -268,15 +269,15 @@ def download_all_pdfs():
     try:
         hf_hub_upload(repo_id=REPO_ID, path_in_repo="all_receipts.zip", path_or_fileobj=zip_path, repo_type="space")
     except Exception as e:
-        print(f"Error uploading ZIP: {e}")
+        print(f"ข้อผิดพลาดในการอัพโหลด ZIP: {e}")
     
     return f"pdfs/all_receipts.zip"
 
-# Initialize database and import JSON
+# เริ่มต้นฐานข้อมูลและนำเข้า JSON
 init_db()
 import_json_to_db()
 
-# Gradio interface
+# อินเท���ร์เฟซ Gradio
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("## เพิ่มข้อมูลพนักงานใหม่")
     with gr.Row():