Update app.py

app.py

@@ -1,7 +1,7 @@
 # =====================================================
 # AI INSURANCE CLAIM GENERATOR
 # FINAL VERSION
-# Working Extraction Logic + Professional PDF + Email
+# Accurate Extraction + Professional Claim PDF + Email
 # Hugging Face Space Ready
 # =====================================================
 
@@ -14,12 +14,14 @@ import requests
 import os
 import io
 import base64
+from datetime import datetime
 
 from transformers import LayoutLMTokenizerFast, LayoutLMForTokenClassification
 
 # PDF LIBRARY
 from reportlab.lib.pagesizes import A4
 from reportlab.pdfgen import canvas
+from reportlab.lib.colors import HexColor
 
 # =====================================================
 # CONFIG
@@ -127,7 +129,7 @@ def extract_total(words):
     return "Not Found"
 
 # =====================================================
-# PROFESSIONAL PDF GENERATOR
+# PROFESSIONAL CLAIM PDF
 # =====================================================
 def create_pdf_base64(extracted):
 
@@ -136,28 +138,70 @@ def create_pdf_base64(extracted):
     p = canvas.Canvas(buffer, pagesize=A4)
     width, height = A4
 
-    y = height - 60
+    # Colors
+    blue = HexColor("#0B5ED7")
+    dark = HexColor("#222222")
+    gray = HexColor("#666666")
+    light = HexColor("#F5F7FA")
 
-    p.setFont("Helvetica-Bold", 18)
-    p.drawString(50, y, "AI Insurance Claim Report")
+    # Header Bar
+    p.setFillColor(blue)
+    p.rect(0, height - 80, width, 80, fill=1, stroke=0)
 
-    y -= 40
-    p.setFont("Helvetica", 12)
+    p.setFillColorRGB(1, 1, 1)
+    p.setFont("Helvetica-Bold", 22)
+    p.drawString(40, height - 50, "INSURANCE CLAIM FORM")
+
+    p.setFont("Helvetica", 10)
+    p.drawRightString(width - 40, height - 52, "AI Generated Report")
+
+    # Body Box
+    p.setFillColor(light)
+    p.roundRect(35, height - 430, width - 70, 290, 10, fill=1, stroke=0)
+
+    y = height - 130
+
+    p.setFillColor(dark)
+    p.setFont("Helvetica-Bold", 14)
+    p.drawString(50, y, "Claim Summary")
+
+    y -= 35
 
     rows = [
-        f"Provider Name : {extracted['company']}",
-        f"Bill Date     : {extracted['date']}",
-        f"Claim Amount  : {extracted['total']}",
-        f"Confidence    : {extracted['confidence']}",
-        f"Decision      : {extracted['decision']}"
+        ("Claim ID", f"CLM-{datetime.now().strftime('%Y%m%d%H%M%S')}"),
+        ("Provider Name", extracted["company"]),
+        ("Bill Date", extracted["date"]),
+        ("Claim Amount", f"₹ {extracted['total']}"),
+        ("Status", "Submitted"),
+        ("Generated On", datetime.now().strftime("%d-%m-%Y %H:%M"))
     ]
 
-    for row in rows:
-        p.drawString(50, y, row)
-        y -= 28
+    for label, value in rows:
+
+        p.setFillColor(gray)
+        p.setFont("Helvetica-Bold", 11)
+        p.drawString(55, y, label)
+
+        p.setFillColor(dark)
+        p.setFont("Helvetica", 11)
+        p.drawString(220, y, str(value))
+
+        y -= 32
+
+    # Footer Notes
+    p.setFillColor(gray)
+    p.setFont("Helvetica", 9)
+    p.drawString(
+        40,
+        60,
+        "This document was generated automatically by AI Insurance Claim Generator."
+    )
 
-    y -= 20
-    p.drawString(50, y, "Generated by AI Insurance Claim Generator")
+    p.drawRightString(
+        width - 40,
+        60,
+        "Confidential"
+    )
 
     p.showPage()
     p.save()
@@ -296,7 +340,7 @@ def send_claim_email(to_email, extracted):
     <p><b>Date:</b> {extracted['date']}</p>
     <p><b>Amount:</b> ₹{extracted['total']}</p>
 
-    <p>Attached: AI Claim Report PDF</p>
+    <p>Attached: Professional Claim PDF</p>
     """
 
     try:
@@ -372,7 +416,7 @@ demo = gr.Interface(
     ],
 
     title="📄 AI Insurance Claim Generator",
-    description="Upload receipt → Extract fields → Generate PDF → Auto Email"
+    description="Upload receipt → Extract fields → Generate Claim PDF → Auto Email"
 )
 
 demo.launch()