Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import pandas as pd
+import json
 import io
 import zipfile
 from datetime import datetime
@@ -39,8 +40,10 @@ def analyze_pdf_fields(pdf_path):
                     if "/T" in field_obj:
                         field_name = str(field_obj["/T"]).strip("()")
                         field_type = str(field_obj.get("/FT", "Unknown"))
+                        field_value = str(field_obj.get("/V", "")).strip("()")
                         all_fields[field_name] = {
                             'type': field_type,
+                            'default_value': field_value,
                             'method': 'AcroForm'
                         }
         
@@ -55,8 +58,10 @@ def analyze_pdf_fields(pdf_path):
                             if "/T" in annot_obj:
                                 field_name = str(annot_obj["/T"]).strip("()")
                                 field_type = str(annot_obj.get("/FT", "Widget"))
+                                field_value = str(annot_obj.get("/V", "")).strip("()")
                                 all_fields[field_name] = {
                                     'type': field_type,
+                                    'default_value': field_value,
                                     'page': page_num + 1,
                                     'method': 'Annotation'
                                 }
@@ -67,6 +72,166 @@ def analyze_pdf_fields(pdf_path):
     except Exception as e:
         return {"error": str(e)}
 
+def generate_csv_template(pdf_fields, num_rows=5):
+    """สร้าง CSV template จาก PDF fields"""
+    if not pdf_fields or "error" in pdf_fields:
+        return None, "ไม่สามารถสร้าง CSV template ได้"
+    
+    # สร้าง DataFrame ว่าง
+    template_data = {}
+    
+    # เพิ่มคอลัมน์ ID
+    template_data['id'] = list(range(1, num_rows + 1))
+    
+    # เพิ่มคอลัมน์จาก PDF fields
+    for field_name, field_info in pdf_fields.items():
+        if field_name and field_name.strip():
+            clean_name = field_name.strip()
+            # ใส่ค่าตัวอย่าง
+            if field_info.get('default_value') and field_info['default_value'].strip():
+                sample_value = field_info['default_value']
+            else:
+                # สร้างค่าตัวอย่าง
+                if 'name' in clean_name.lower():
+                    sample_value = f"ชื่อตัวอย่าง {{}}"
+                elif 'date' in clean_name.lower() or 'วันที่' in clean_name:
+                    sample_value = "2024-01-01"
+                elif 'email' in clean_name.lower():
+                    sample_value = "example{}@email.com"
+                elif 'phone' in clean_name.lower() or 'เบอร์' in clean_name:
+                    sample_value = "08-1234-567{}"
+                elif 'address' in clean_name.lower() or 'ที่อยู่' in clean_name:
+                    sample_value = "123 ถนนตัวอย่าง กรุงเทพ {}"
+                else:
+                    sample_value = f"ข้อมูลตัวอย่าง {{}}"
+            
+            # สร้างข้อมูลตัวอย่าง
+            template_data[clean_name] = [
+                sample_value.format(i) if '{}' in sample_value else sample_value 
+                for i in range(1, num_rows + 1)
+            ]
+    
+    # สร้าง DataFrame
+    df = pd.DataFrame(template_data)
+    return df, "สร้าง CSV template สำเร็จ"
+
+def generate_json_template(pdf_fields):
+    """สร้าง JSON template จาก PDF fields"""
+    if not pdf_fields or "error" in pdf_fields:
+        return None, "ไม่สามารถสร้าง JSON template ได้"
+    
+    template = {
+        "pdf_info": {
+            "total_fields": len(pdf_fields),
+            "generation_time": datetime.now().isoformat()
+        },
+        "fields": {},
+        "sample_data": []
+    }
+    
+    # เพิ่มข้อมูล fields
+    for field_name, field_info in pdf_fields.items():
+        if field_name and field_name.strip():
+            clean_name = field_name.strip()
+            template["fields"][clean_name] = {
+                "type": field_info.get('type', 'Unknown'),
+                "default_value": field_info.get('default_value', ''),
+                "page": field_info.get('page', 1),
+                "method": field_info.get('method', 'Unknown')
+            }
+    
+    # สร้างข้อมูลตัวอย่าง
+    for i in range(1, 4):  # 3 ตัวอย่าง
+        sample_record = {"id": i}
+        for field_name in template["fields"].keys():
+            if 'name' in field_name.lower():
+                sample_record[field_name] = f"ชื่อตัวอย่าง {i}"
+            elif 'date' in field_name.lower() or 'วันที่' in field_name:
+                sample_record[field_name] = f"2024-0{i}-01"
+            elif 'email' in field_name.lower():
+                sample_record[field_name] = f"example{i}@email.com"
+            else:
+                sample_record[field_name] = f"ข้อมูลตัวอย่าง {i}"
+        template["sample_data"].append(sample_record)
+    
+    return template, "สร้าง JSON template สำเร็จ"
+
+def create_template_files(pdf_file, num_rows=5):
+    """สร้างไฟล์ template ทั้งหมด"""
+    if pdf_file is None:
+        return None, None, "❌ กรุณาอัพโหลดไฟล์ PDF"
+    
+    try:
+        # วิเคราะห์ PDF
+        pdf_fields = analyze_pdf_fields(pdf_file)
+        
+        if not pdf_fields or "error" in pdf_fields:
+            return None, None, "❌ ไม่พบ form fields ใน PDF หรือไม่สามารถอ่านได้"
+        
+        # สร้าง CSV template
+        csv_df, csv_msg = generate_csv_template(pdf_fields, num_rows)
+        
+        # สร้าง JSON template
+        json_template, json_msg = generate_json_template(pdf_fields)
+        
+        if csv_df is None or json_template is None:
+            return None, None, "❌ ไม่สามารถสร้าง template ได้"
+        
+        # สร้าง ZIP file
+        zip_buffer = io.BytesIO()
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        
+        with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
+            # เพิ่ม CSV
+            csv_buffer = io.StringIO()
+            csv_df.to_csv(csv_buffer, index=False, encoding='utf-8-sig')
+            zip_file.writestr(f"template_{timestamp}.csv", csv_buffer.getvalue())
+            
+            # เพิ่ม JSON
+            json_str = json.dumps(json_template, ensure_ascii=False, indent=2)
+            zip_file.writestr(f"template_{timestamp}.json", json_str)
+            
+            # เพิ่ม README
+            readme_content = f"""# PDF Form Template Files
+Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
+
+## Files included:
+1. template_{timestamp}.csv - CSV template with sample data
+2. template_{timestamp}.json - JSON template with field information
+
+## PDF Fields Found: {len(pdf_fields)}
+{chr(10).join([f"- {name}: {info.get('type', 'Unknown')}" for name, info in pdf_fields.items()])}
+
+## How to use:
+1. Edit the CSV file with your actual data
+2. Use the CSV file with the original PDF form filler
+3. The JSON file contains detailed field information for reference
+
+## Note:
+- Make sure column names in CSV match the field names in PDF
+- You can add more rows to the CSV as needed
+- Keep the column headers exactly as shown
+"""
+            zip_file.writestr("README.txt", readme_content)
+        
+        zip_buffer.seek(0)
+        
+        # บันทึกไฟล์ชั่วคราว
+        temp_zip_path = os.path.join(tempfile.gettempdir(), f"pdf_templates_{timestamp}.zip")
+        with open(temp_zip_path, 'wb') as f:
+            f.write(zip_buffer.getvalue())
+        
+        result_msg = f"✅ สร้าง template สำเร็จ!\n"
+        result_msg += f"🔍 พบ {len(pdf_fields)} fields ใน PDF\n"
+        result_msg += f"📄 CSV template: {num_rows} แถวตัวอย่าง\n"
+        result_msg += f"📋 JSON template: รายละเอียดครบถ้วน\n"
+        result_msg += f"📁 README: คำแนะนำการใช้งาน"
+        
+        return temp_zip_path, pdf_fields, result_msg
+        
+    except Exception as e:
+        return None, None, f"❌ เกิดข้อผิดพลาด: {str(e)}"
+
 def fill_pdf_form(pdf_path, field_data):
     """เติมข้อมูลในฟอร์ม PDF"""
     try:
@@ -318,7 +483,7 @@ def process_pdf_csv(pdf_file, csv_file, filename_column, file_prefix, use_form_f
 def analyze_pdf_info(pdf_file):
     """วิเคราะห์ข้อมูล PDF"""
     if pdf_file is None:
-        return "ไม่มีไฟล์ PDF"
+        return "ยังไม่มีไฟล์ PDF"
     
     try:
         reader = PdfReader(pdf_file)
@@ -346,7 +511,7 @@ def analyze_pdf_info(pdf_file):
 def analyze_csv_info(csv_file):
     """วิเคราะห์ข้อมูล CSV"""
     if csv_file is None:
-        return "ไม่มีไฟล์ CSV"
+        return "ยังไม่มีไฟล์ CSV"
     
     try:
         df, error = read_csv_safe(csv_file)
@@ -378,109 +543,79 @@ def analyze_csv_info(csv_file):
 
 # สร้าง Gradio Interface
 def create_interface():
-    with gr.Blocks(title="PDF Form Filler", theme=gr.themes.Soft()) as app:
+    with gr.Blocks(title="PDF Form Filler & Template Generator", theme=gr.themes.Soft()) as app:
         gr.Markdown("""
-        # 📄 เครื่องมือเติมข้อมูล PDF จาก CSV
+        # 📄 เครื่องมือเติมข้อมูล PDF จาก CSV + สร้าง Template
         
         **เครื่องมือนี้สามารถ:**
-        - เติมข้อมูลลงในฟอร์ม PDF ที่มี form fields
-        - สร้าง PDF ใหม่หากไม่มี form fields หรือเติมไม่ได้
-        - รองรับ CSV หลาย encoding (UTF-8, TIS-620, CP874, etc.)
-        - ส่งออกเป็นไฟล์ ZIP
+        - 🔄 **สร้าง CSV/JSON Template จาก PDF** (Reverse Engineering)
+        - 📝 เติมข้อมูลลงในฟอร์ม PDF ที่มี form fields
+        - 🆕 สร้าง PDF ใหม่หากไม่มี form fields หรือเติมไม่ได้
+        - 🌐 รองรับ CSV หลาย encoding (UTF-8, TIS-620, CP874, etc.)
+        - 📦 ส่งออกเป็นไฟล์ ZIP
         """)
         
-        with gr.Row():
-            with gr.Column(scale=1):
-                gr.Markdown("## 📁 อัพโหลดไฟล์")
-                
-                pdf_file = gr.File(
-                    label="PDF Template",
-                    file_types=[".pdf"],
-                    type="filepath"
-                )
+        with gr.Tabs():
+            # Tab 1: Template Generator
+            with gr.TabItem("🔄 สร้าง Template"):
+                gr.Markdown("""
+                ## 🔄 สร้าง CSV/JSON Template จาก PDF
+                **อัพโหลด PDF ที่มี form fields แล้วสร้าง template ให้อัตโนมัติ**
+                """)
                 
-                csv_file = gr.File(
-                    label="CSV Data", 
-                    file_types=[".csv"],
-                    type="filepath"
-                )
-                
-                gr.Markdown("## ⚙️ ตั้งค่า")
-                
-                filename_column = gr.Textbox(
-                    label="คอลัมน์สำหรับชื่อไฟล์ (ถ้ามี)",
-                    placeholder="เช่น name, id, etc.",
-                    value=""
-                )
-                
-                file_prefix = gr.Textbox(
-                    label="คำนำหน้าชื่อไฟล์",
-                    value="document"
-                )
-                
-                use_form_fields = gr.Checkbox(
-                    label="ใช้ Form Fields (ถ้าพบ)",
-                    value=True
-                )
-                
-                process_btn = gr.Button(
-                    "🚀 สร้าง PDF ทั้งหมด",
-                    variant="primary",
-                    size="lg"
-                )
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        template_pdf = gr.File(
+                            label="📄 PDF ที่ต้องการสร้าง Template",
+                            file_types=[".pdf"],
+                            type="filepath"
+                        )
+                        
+                        num_sample_rows = gr.Slider(
+                            label="จำนวนแถวตัวอย่างใน CSV",
+                            minimum=1,
+                            maximum=20,
+                            value=5,
+                            step=1
+                        )
+                        
+                        generate_template_btn = gr.Button(
+                            "🔄 สร้าง Template",
+                            variant="primary",
+                            size="lg"
+                        )
+                    
+                    with gr.Column(scale=2):
+                        template_pdf_info = gr.Markdown("ยังไม่มีไฟล์ PDF")
+                        
+                        template_result_file = gr.File(
+                            label="📦 ไฟล์ Template (ZIP)",
+                            visible=False
+                        )
+                        
+                        template_result_message = gr.Markdown("")
             
-            with gr.Column(scale=2):
-                gr.Markdown("## 📊 ข้อมูลไฟล์")
+            # Tab 2: PDF Form Filler
+            with gr.TabItem("📝 เติมข้อมูล PDF"):
+                gr.Markdown("""
+                ## 📝 เติมข้อมูล PDF จาก CSV
+                **ใช้ CSV ที่สร้างจาก Template หรือ CSV ของคุณเอง**
+                """)
                 
-                pdf_info = gr.Markdown("ยังไม่มีไฟล์ PDF")
-                csv_info = gr.Markdown("ยังไม่มีไฟล์ CSV")
-                
-                gr.Markdown("## 📥 ผลลัพธ์")
-                
-                result_file = gr.File(
-                    label="ไฟล์ ZIP ที่สร้าง",
-                    visible=False
-                )
-                
-                result_message = gr.Markdown("")
-        
-        # Event handlers
-        pdf_file.change(
-            fn=analyze_pdf_info,
-            inputs=[pdf_file],
-            outputs=[pdf_info]
-        )
-        
-        csv_file.change(
-            fn=analyze_csv_info,
-            inputs=[csv_file],
-            outputs=[csv_info]
-        )
-        
-        process_btn.click(
-            fn=process_pdf_csv,
-            inputs=[
-                pdf_file, 
-                csv_file, 
-                filename_column, 
-                file_prefix, 
-                use_form_fields
-            ],
-            outputs=[result_file, result_message]
-        ).then(
-            fn=lambda x: gr.update(visible=x is not None),
-            inputs=[result_file],
-            outputs=[result_file]
-        )
-    
-    return app
-
-# รันแอป
-if __name__ == "__main__":
-    app = create_interface()
-    app.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True,  # สร้าง public URL
-        debug=True
-    )
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        gr.Markdown("### 📁 อัพโหลดไฟล์")
+                        
+                        pdf_file = gr.File(
+                            label="PDF Template",
+                            file_types=[".pdf"],
+                            type="filepath"
+                        )
+                        
+                        csv_file = gr.File(
+                            label="CSV Data", 
+                            file_types=[".csv"],
+                            type="filepath"
+                        )
+                        
+                        gr.Markdown("### ⚙️ ตั้งค่