Spaces:

vachaspathi
/

invoice-pdf-makers

Sleeping

App Files Files Community

vachaspathi commited on Nov 21, 2025

Commit

333db8a

verified ·

1 Parent(s): 120af7e

Update app.py

Browse files

Files changed (1) hide show

app.py +125 -73

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import zipfile
 from jinja2 import Template
 from weasyprint import HTML
-# --- 1. Universal Document Template ---
 DOC_TEMPLATE = """
 <!DOCTYPE html>
 <html lang="en">
@@ -56,8 +56,8 @@ DOC_TEMPLATE = """
     <div class="addresses">
         <div class="address-box">
             {% if doc_type == 'expenses' %}
-                <strong>Payable To:</strong>
-                <div>{{ payee }}</div>
             {% else %}
                 <strong>Bill To:</strong>
                 {% if customer_details %}
@@ -68,11 +68,11 @@ DOC_TEMPLATE = """
                     {% if customer_details.billing_address %}
                         {% set addr = customer_details.billing_address %}
                         <div>{{ addr.address }}</div>
-                        <div>{{ addr.city }}, {{ addr.state }} {{ addr.zip }}</div>
                         <div>{{ addr.country }}</div>
                     {% endif %}
                 {% else %}
-                    <div>{{ customer_id }}</div>
                 {% endif %}
             {% endif %}
         </div>
@@ -81,7 +81,7 @@ DOC_TEMPLATE = """
             {% if doc_type == 'expenses' %}
                 <strong>Category:</strong> {{ category }}
             {% elif doc_type == 'delivery_challans' %}
-                <strong>Ref Invoice:</strong> {{ reference_invoice }}
             {% elif doc_type == 'recurring_invoices' %}
                 <strong>Frequency:</strong> {{ frequency }}
             {% endif %}
@@ -92,14 +92,9 @@ DOC_TEMPLATE = """
         <thead>
             <tr>
                 <th width="50%">Item & Description</th>
-                {% if doc_type == 'delivery_challans' %}
-                    <th>Packed In</th>
-                    <th class="text-right">Quantity</th>
-                {% else %}
-                    <th class="text-right">Qty</th>
-                    <th class="text-right">Rate</th>
-                    <th class="text-right">Amount</th>
-                {% endif %}
             </tr>
         </thead>
         <tbody>
@@ -111,14 +106,9 @@ DOC_TEMPLATE = """
                     <div class="item-desc">{{ item.description }}</div>
                     {% endif %}
                 </td>
-                {% if doc_type == 'delivery_challans' %}
-                    <td>{{ item.packed_in }}</td>
-                    <td class="text-right">{{ item.quantity }}</td>
-                {% else %}
-                    <td class="text-right">{{ item.quantity }}</td>
-                    <td class="text-right">{{ item.rate }}</td>
-                    <td class="text-right">{{ item.calculated_amount }}</td>
-                {% endif %}
             </tr>
             {% endfor %}
         </tbody>
@@ -139,16 +129,84 @@ DOC_TEMPLATE = """
         {% if notes %}
         <p><strong>Notes:</strong> {{ notes }}</p>
         {% endif %}
-        {% if description %}
-        <p><strong>Description:</strong> {{ description }}</p>
-        {% endif %}
-        <p>Generated via HuggingFace Spaces</p>
     </div>
 </body>
 </html>
 """
-# --- 2. Logic ---
 def get_contact_map(data):
     contacts = data.get("contacts", [])
@@ -160,20 +218,24 @@ def get_contact_map(data):
     return c_map
 def normalize_record(record, doc_type, contact_map):
-    # Defaults
-    title = doc_type.replace("_", " ").upper()
-    number = "DOC-000"
     if doc_type == "invoices":
         number = record.get("invoice_number", "INV-000")
     elif doc_type == "recurring_invoices":
         number = record.get("profile_name", "REC-000")
     elif doc_type == "delivery_challans":
         number = record.get("challan_number", "DC-000")
     elif doc_type == "expenses":
         number = record.get("expense_id", "EXP-000")
-    # Process Items
     items = record.get("line_items", [])
     sub_total = 0.0
@@ -181,11 +243,12 @@ def normalize_record(record, doc_type, contact_map):
         try: qty = float(item.get("quantity", 1))
         except: qty = 0
         try: rate = float(item.get("rate", item.get("amount", 0)))
         except: rate = 0
         amount = item.get("amount")
-        if amount is None:
             amount = qty * rate
         else:
             try: amount = float(amount)
@@ -194,10 +257,9 @@ def normalize_record(record, doc_type, contact_map):
         item["quantity"] = qty
         item["rate"] = f"{rate:,.2f}"
         item["calculated_amount"] = f"{amount:,.2f}"
-        item["packed_in"] = item.get("packed_in", "-") # For challans
         sub_total += amount
     cust_id = record.get("customer_id")
     return {
@@ -209,25 +271,24 @@ def normalize_record(record, doc_type, contact_map):
         "status": record.get("status"),
         "customer_id": cust_id,
         "customer_details": contact_map.get(cust_id, {}),
-        "payee": record.get("payee"),
         "category": record.get("category"),
         "description": record.get("description"),
-        "reference_invoice": record.get("reference_invoice", ""),
         "frequency": record.get("frequency"),
         "line_items": items,
         "calculated_total": f"{sub_total:,.2f}",
         "notes": record.get("notes", "")
     }
 def render_pdf(record):
     template = Template(DOC_TEMPLATE)
     html_content = template.render(**record)
-    # Create Filename
     safe_name = str(record['doc_number']).replace(" ", "_").replace("/", "-").replace("\\", "-")
     filename = f"{safe_name}.pdf"
-    # WRITE PDF - IF THIS FAILS, GRADIO WILL SHOW THE REAL ERROR
     HTML(string=html_content).write_pdf(filename)
     return filename
@@ -235,64 +296,55 @@ def process_file(file_obj):
     if file_obj is None:
         return None
-    # 1. Load JSON
     try:
         with open(file_obj.name, 'r', encoding='utf-8') as f:
-            raw_data = json.load(f)
     except Exception as e:
         raise gr.Error(f"JSON Load Error: {e}")
-    # 2. Map Contacts
     contact_map = get_contact_map(raw_data)
     generated_files = []
     keys_to_process = ["invoices", "recurring_invoices", "delivery_challans", "expenses"]
-    # Debug: Print found keys to logs
-    print(f"DEBUG: JSON Keys found: {list(raw_data.keys())}")
-    # 3. Process Records
-    found_records_count = 0
     for key in keys_to_process:
         records = raw_data.get(key, [])
         if not isinstance(records, list):
             continue
         for rec in records:
-            found_records_count += 1
-            # Normalize
-            norm_rec = normalize_record(rec, key, contact_map)
-            # Render (No Try/Except here, let it fail so we see why)
-            pdf_path = render_pdf(norm_rec)
-            generated_files.append(pdf_path)
-    if found_records_count == 0:
-        raise gr.Error(f"Structure Error: JSON parsed, but no arrays found for keys: {keys_to_process}. Check your JSON keys.")
     if not generated_files:
-        raise gr.Error("Unknown Error: Records found but no PDFs generated.")
-    # 4. Zip or Return
-    if len(generated_files) == 1:
-        return generated_files[0]
-    else:
-        zip_name = "documents.zip"
-        with zipfile.ZipFile(zip_name, 'w') as zf:
-            for f in generated_files:
-                zf.write(f)
-                # Optional: keep files for debugging or remove them
-                os.remove(f)
-        return zip_name
-# --- 3. Interface ---
 iface = gr.Interface(
     fn=process_file,
-    inputs=gr.File(label="Upload JSON", file_types=[".json"]),
     outputs=gr.File(label="Download Result"),
-    title="Universal Document Generator",
-    description="Upload a JSON file. Supported keys: invoices, recurring_invoices, delivery_challans, expenses.",
     allow_flagging="never"
 )

 from jinja2 import Template
 from weasyprint import HTML
+# --- 1. PDF Template ---
 DOC_TEMPLATE = """
 <!DOCTYPE html>
 <html lang="en">
     <div class="addresses">
         <div class="address-box">
             {% if doc_type == 'expenses' %}
+                <strong>Payable To / Vendor:</strong>
+                <div>{{ payee or 'N/A' }}</div>
             {% else %}
                 <strong>Bill To:</strong>
                 {% if customer_details %}
                     {% if customer_details.billing_address %}
                         {% set addr = customer_details.billing_address %}
                         <div>{{ addr.address }}</div>
+                        <div>{{ addr.city }} {{ addr.zip }}</div>
                         <div>{{ addr.country }}</div>
                     {% endif %}
                 {% else %}
+                    <div>ID: {{ customer_id }}</div>
                 {% endif %}
             {% endif %}
         </div>
             {% if doc_type == 'expenses' %}
                 <strong>Category:</strong> {{ category }}
             {% elif doc_type == 'delivery_challans' %}
+                <strong>Details:</strong> Shipping Note
             {% elif doc_type == 'recurring_invoices' %}
                 <strong>Frequency:</strong> {{ frequency }}
             {% endif %}
         <thead>
             <tr>
                 <th width="50%">Item & Description</th>
+                <th class="text-right">Qty</th>
+                <th class="text-right">Rate</th>
+                <th class="text-right">Amount</th>
             </tr>
         </thead>
         <tbody>
                     <div class="item-desc">{{ item.description }}</div>
                     {% endif %}
                 </td>
+                <td class="text-right">{{ item.quantity }}</td>
+                <td class="text-right">{{ item.rate }}</td>
+                <td class="text-right">{{ item.calculated_amount }}</td>
             </tr>
             {% endfor %}
         </tbody>
         {% if notes %}
         <p><strong>Notes:</strong> {{ notes }}</p>
         {% endif %}
+        <p>Generated Document</p>
     </div>
 </body>
 </html>
 """
+# --- 2. Data Cleaning & Mapping Logic ---
+def clean_input_data(raw_data):
+    """
+    Fixes the input data:
+    1. Extracts inner JSON if it's wrapped in Pandoc/Markdown blocks.
+    2. Renames specific fields (customer_ref -> customer_id) to match the template.
+    """
+    # 1. Handle Pandoc/Markdown wrapper (The "Structure Error" fix)
+    if "blocks" in raw_data and isinstance(raw_data["blocks"], list):
+        print("DEBUG: Detected Pandoc format. Extracting inner JSON...")
+        try:
+            # Iterate through blocks to find the CodeBlock containing the JSON string
+            for block in raw_data["blocks"]:
+                if block.get("t") == "CodeBlock":
+                    # The content is usually in block['c'][1]
+                    inner_json_str = block["c"][1]
+                    return clean_input_data(json.loads(inner_json_str))
+        except Exception as e:
+            print(f"DEBUG: Failed to extract inner JSON: {e}")
+            # If extraction fails, return raw_data and hope for the best
+            pass
+    # 2. Standardize Keys (The "Data Missing" fix)
+    # We create a new clean dictionary mapping standard keys to the user's keys
+    cleaned = {}
+    # Map Clients -> Contacts
+    if "clients" in raw_data:
+        cleaned["contacts"] = []
+        for c in raw_data["clients"]:
+            # Map 'contact' object to top level if nested
+            contact_info = c.get("contact", c)
+            # Ensure client_id maps to contact_id
+            contact_info["contact_id"] = c.get("client_id", c.get("contact_id"))
+            cleaned["contacts"].append(contact_info)
+    elif "contacts" in raw_data:
+        cleaned["contacts"] = raw_data["contacts"]
+    # Process Transactions
+    # Map (User's Key) -> (Standard Key)
+    mappings = [
+        ("invoices", "invoices"),
+        ("recurring_invoices", "recurring_invoices"),
+        ("delivery_challans", "delivery_challans"),
+        ("expenses", "expenses")
+    ]
+    for user_key, standard_key in mappings:
+        if user_key in raw_data:
+            cleaned[standard_key] = []
+            for item in raw_data[user_key]:
+                # Fix Reference IDs
+                if "customer_ref" in item:
+                    item["customer_id"] = item["customer_ref"]
+                if "client_ref" in item:
+                    item["customer_id"] = item["client_ref"]
+                # Fix Items List
+                if "items" in item:
+                    item["line_items"] = item["items"]
+                # Fix Recurring Invoice Name
+                if "name" in item and standard_key == "recurring_invoices":
+                    item["profile_name"] = item["name"]
+                if "recurrence_frequency" in item:
+                    item["frequency"] = item["recurrence_frequency"]
+                cleaned[standard_key].append(item)
+    return cleaned
 def get_contact_map(data):
     contacts = data.get("contacts", [])
     return c_map
 def normalize_record(record, doc_type, contact_map):
+    # Determine Title and Number
     if doc_type == "invoices":
+        title = "INVOICE"
         number = record.get("invoice_number", "INV-000")
     elif doc_type == "recurring_invoices":
+        title = "RECURRING INVOICE"
         number = record.get("profile_name", "REC-000")
     elif doc_type == "delivery_challans":
+        title = "DELIVERY CHALLAN"
         number = record.get("challan_number", "DC-000")
     elif doc_type == "expenses":
+        title = "EXPENSE RECEIPT"
         number = record.get("expense_id", "EXP-000")
+    else:
+        title = doc_type.upper()
+        number = "DOC-000"
+    # Process Line Items
     items = record.get("line_items", [])
     sub_total = 0.0
         try: qty = float(item.get("quantity", 1))
         except: qty = 0
+        # Expenses often have 'amount' directly, others have 'rate'
         try: rate = float(item.get("rate", item.get("amount", 0)))
         except: rate = 0
         amount = item.get("amount")
+        if amount is None or doc_type != 'expenses':
             amount = qty * rate
         else:
             try: amount = float(amount)
         item["quantity"] = qty
         item["rate"] = f"{rate:,.2f}"
         item["calculated_amount"] = f"{amount:,.2f}"
         sub_total += amount
+    # Get Customer Details
     cust_id = record.get("customer_id")
     return {
         "status": record.get("status"),
         "customer_id": cust_id,
         "customer_details": contact_map.get(cust_id, {}),
+        "payee": record.get("payee", record.get("category")), # Fallback for expenses
         "category": record.get("category"),
         "description": record.get("description"),
         "frequency": record.get("frequency"),
         "line_items": items,
         "calculated_total": f"{sub_total:,.2f}",
         "notes": record.get("notes", "")
     }
+# --- 3. Main Process ---
 def render_pdf(record):
     template = Template(DOC_TEMPLATE)
     html_content = template.render(**record)
     safe_name = str(record['doc_number']).replace(" ", "_").replace("/", "-").replace("\\", "-")
     filename = f"{safe_name}.pdf"
     HTML(string=html_content).write_pdf(filename)
     return filename
     if file_obj is None:
         return None
     try:
         with open(file_obj.name, 'r', encoding='utf-8') as f:
+            raw_input = json.load(f)
     except Exception as e:
         raise gr.Error(f"JSON Load Error: {e}")
+    # CLEAN THE DATA
+    raw_data = clean_input_data(raw_input)
+    # Map Contacts
     contact_map = get_contact_map(raw_data)
     generated_files = []
     keys_to_process = ["invoices", "recurring_invoices", "delivery_challans", "expenses"]
     for key in keys_to_process:
         records = raw_data.get(key, [])
         if not isinstance(records, list):
             continue
         for rec in records:
+            try:
+                norm_rec = normalize_record(rec, key, contact_map)
+                pdf_path = render_pdf(norm_rec)
+                generated_files.append(pdf_path)
+            except Exception as e:
+                print(f"Error processing {key} record: {e}")
     if not generated_files:
+        # Check debug info
+        found_keys = list(raw_data.keys())
+        raise gr.Error(f"No PDFs generated. Found keys after cleaning: {found_keys}")
+    # Zip Result
+    zip_name = "documents.zip"
+    with zipfile.ZipFile(zip_name, 'w') as zf:
+        for f in generated_files:
+            zf.write(f)
+            os.remove(f)
+    return zip_name
+# --- 4. Interface ---
 iface = gr.Interface(
     fn=process_file,
+    inputs=gr.File(label="Upload JSON (Zoho/Pandoc format supported)", file_types=[".json"]),
     outputs=gr.File(label="Download Result"),
+    title="Universal Invoice & Expense Generator",
+    description="Upload your JSON file. This tool automatically detects Invoice, Recurring Invoice, Challan, and Expense data, cleans the format, and generates PDFs.",
     allow_flagging="never"
 )