Spaces:
Sleeping
Sleeping
| """ | |
| sample_docs.py — Generates realistic synthetic document images for instant 1-click testing in DocIQ. | |
| """ | |
| from PIL import Image, ImageDraw, ImageFont | |
| import io | |
| def _get_fonts(): | |
| """Try to obtain clean TrueType fonts, fall back to default.""" | |
| try: | |
| title_font = ImageFont.truetype("arial.ttf", 24) | |
| heading_font = ImageFont.truetype("arial.ttf", 18) | |
| body_font = ImageFont.truetype("arial.ttf", 15) | |
| mono_font = ImageFont.truetype("cour.ttf", 14) | |
| small_font = ImageFont.truetype("arial.ttf", 12) | |
| except Exception: | |
| title_font = ImageFont.load_default() | |
| heading_font = title_font | |
| body_font = title_font | |
| mono_font = title_font | |
| small_font = title_font | |
| return title_font, heading_font, body_font, mono_font, small_font | |
| def generate_invoice_sample() -> Image.Image: | |
| """Generate a clean business invoice document.""" | |
| width, height = 800, 1000 | |
| img = Image.new("RGB", (width, height), (255, 255, 255)) | |
| draw = ImageDraw.Draw(img) | |
| title_font, heading_font, body_font, mono_font, small_font = _get_fonts() | |
| # Header Banner | |
| draw.rectangle([0, 0, width, 90], fill=(30, 41, 59)) | |
| draw.text((40, 28), "ACME CLOUD SOLUTIONS INC.", fill=(255, 255, 255), font=title_font) | |
| draw.text((580, 32), "INVOICE", fill=(129, 140, 248), font=title_font) | |
| # Invoice Meta | |
| y = 120 | |
| draw.text((40, y), "Billed To:", fill=(100, 116, 139), font=small_font) | |
| draw.text((40, y + 20), "TechCorp International Ltd.", fill=(15, 23, 42), font=heading_font) | |
| draw.text((40, y + 46), "100 Innovation Way, Suite 400", fill=(71, 85, 105), font=body_font) | |
| draw.text((40, y + 70), "Email: billing@techcorp.com", fill=(71, 85, 105), font=body_font) | |
| draw.text((40, y + 94), "Phone: +1 (555) 234-5678", fill=(71, 85, 105), font=body_font) | |
| draw.text((540, y), "Invoice Details:", fill=(100, 116, 139), font=small_font) | |
| draw.text((540, y + 20), "Invoice #: INV-2026-8842", fill=(15, 23, 42), font=mono_font) | |
| draw.text((540, y + 46), "Date: 15/09/2026", fill=(71, 85, 105), font=body_font) | |
| draw.text((540, y + 70), "Due Date: 30/09/2026", fill=(71, 85, 105), font=body_font) | |
| draw.text((540, y + 94), "Status: PENDING", fill=(217, 119, 6), font=mono_font) | |
| # Table Header | |
| y = 280 | |
| draw.rectangle([40, y, width - 40, y + 40], fill=(241, 245, 249)) | |
| draw.text((55, y + 10), "Description", fill=(30, 41, 59), font=heading_font) | |
| draw.text((420, y + 10), "Qty", fill=(30, 41, 59), font=heading_font) | |
| draw.text((520, y + 10), "Unit Price", fill=(30, 41, 59), font=heading_font) | |
| draw.text((660, y + 10), "Amount", fill=(30, 41, 59), font=heading_font) | |
| # Table Rows | |
| items = [ | |
| ("Cloud Server Hosting - Cluster A", "2", "$450.00", "$900.00"), | |
| ("Database Backup & Storage (10TB)", "1", "$250.00", "$250.00"), | |
| ("API Gateway Bandwidth Overages", "1", "$150.00", "$150.00"), | |
| ("Security SSL & Compliance Pack", "1", "$199.00", "$199.00"), | |
| ] | |
| row_y = y + 55 | |
| for desc, qty, price, total in items: | |
| draw.text((55, row_y), desc, fill=(51, 65, 85), font=body_font) | |
| draw.text((430, row_y), qty, fill=(51, 65, 85), font=body_font) | |
| draw.text((520, row_y), price, fill=(51, 65, 85), font=body_font) | |
| draw.text((660, row_y), total, fill=(15, 23, 42), font=body_font) | |
| draw.line([(40, row_y + 35), (width - 40, row_y + 35)], fill=(226, 232, 240), width=1) | |
| row_y += 48 | |
| # Summary | |
| sum_y = row_y + 20 | |
| draw.text((520, sum_y), "Subtotal:", fill=(71, 85, 105), font=body_font) | |
| draw.text((660, sum_y), "$1,499.00", fill=(15, 23, 42), font=body_font) | |
| draw.text((520, sum_y + 30), "Tax (10%):", fill=(71, 85, 105), font=body_font) | |
| draw.text((660, sum_y + 30), "$149.90", fill=(15, 23, 42), font=body_font) | |
| draw.rectangle([500, sum_y + 65, width - 40, sum_y + 115], fill=(238, 242, 255)) | |
| draw.text((520, sum_y + 78), "Total Due:", fill=(67, 56, 202), font=heading_font) | |
| draw.text((650, sum_y + 78), "$1,648.90", fill=(67, 56, 202), font=heading_font) | |
| # Footer note | |
| draw.text((40, 850), "Payment Terms: Wire transfer payable to ACME Cloud Solutions within 15 days.", fill=(100, 116, 139), font=small_font) | |
| draw.text((40, 875), "Support inquiries: support@acmecloud.io | Website: https://acmecloud.io", fill=(100, 116, 139), font=small_font) | |
| draw.text((40, 920), "CONFIDENTIAL DOCUMENT - FOR RECIPIENT USE ONLY", fill=(148, 163, 184), font=small_font) | |
| return img | |
| def generate_receipt_sample() -> Image.Image: | |
| """Generate a clean retail store receipt with PII and pricing.""" | |
| width, height = 500, 750 | |
| img = Image.new("RGB", (width, height), (252, 252, 252)) | |
| draw = ImageDraw.Draw(img) | |
| title_font, heading_font, body_font, mono_font, small_font = _get_fonts() | |
| y = 30 | |
| draw.text((160, y), "METRO MART", fill=(20, 20, 20), font=title_font) | |
| draw.text((140, y + 35), "Store #402 - 5th Avenue", fill=(80, 80, 80), font=body_font) | |
| draw.text((170, y + 60), "Tel: 555-019-2834", fill=(80, 80, 80), font=small_font) | |
| draw.line([(30, y + 85), (width - 30, y + 85)], fill=(180, 180, 180), width=1) | |
| y = 130 | |
| draw.text((30, y), "Date: 22/09/2026", fill=(60, 60, 60), font=small_font) | |
| draw.text((300, y), "Time: 14:32:05", fill=(60, 60, 60), font=small_font) | |
| draw.text((30, y + 22), "Cashier: David M.", fill=(60, 60, 60), font=small_font) | |
| draw.text((300, y + 22), "Receipt: #89211", fill=(60, 60, 60), font=small_font) | |
| y = 180 | |
| draw.line([(30, y), (width - 30, y)], fill=(180, 180, 180), width=1) | |
| y += 10 | |
| draw.text((30, y), "ITEM", fill=(40, 40, 40), font=heading_font) | |
| draw.text((380, y), "PRICE", fill=(40, 40, 40), font=heading_font) | |
| y += 30 | |
| items = [ | |
| ("Organic Whole Milk 1L", "$3.49"), | |
| ("Artisan Sourdough Loaf", "$4.99"), | |
| ("Dark Roast Ground Coffee 250g", "$8.99"), | |
| ("Greek Yogurt Honey 500g", "$4.29"), | |
| ("Fresh Valencia Oranges 1kg", "$5.50"), | |
| ("Avocados Hass (Pack of 3)", "$3.99"), | |
| ] | |
| for item, price in items: | |
| draw.text((30, y), item, fill=(50, 50, 50), font=body_font) | |
| draw.text((390, y), price, fill=(20, 20, 20), font=mono_font) | |
| y += 28 | |
| y += 15 | |
| draw.line([(30, y), (width - 30, y)], fill=(180, 180, 180), width=1) | |
| y += 15 | |
| draw.text((250, y), "SUBTOTAL:", fill=(80, 80, 80), font=body_font) | |
| draw.text((380, y), "$31.25", fill=(20, 20, 20), font=mono_font) | |
| y += 26 | |
| draw.text((250, y), "TAX (8%):", fill=(80, 80, 80), font=body_font) | |
| draw.text((380, y), "$2.50", fill=(20, 20, 20), font=mono_font) | |
| y += 30 | |
| draw.text((220, y), "TOTAL AMOUNT:", fill=(15, 23, 42), font=heading_font) | |
| draw.text((370, y), "$33.75", fill=(15, 23, 42), font=heading_font) | |
| y += 50 | |
| draw.line([(30, y), (width - 30, y)], fill=(200, 200, 200), width=1) | |
| y += 15 | |
| draw.text((30, y), "Card: Visa Ending in 4521", fill=(80, 80, 80), font=small_font) | |
| draw.text((30, y + 20), "Auth Code: 092813 - Approved", fill=(80, 80, 80), font=small_font) | |
| y += 50 | |
| draw.text((150, y), "*** THANK YOU FOR SHOPPING ***", fill=(100, 100, 100), font=small_font) | |
| draw.text((175, y + 20), "Visit us at metromart.com", fill=(120, 120, 120), font=small_font) | |
| return img | |
| SAMPLE_DOCUMENTS = { | |
| "Commercial Invoice": generate_invoice_sample, | |
| "Retail Store Receipt": generate_receipt_sample, | |
| } | |