Update app.py

app.py

@@ -125,29 +125,69 @@ def get_extraction_prompt(model_choice, txt):
         '  "invoice_header": {\n'
         '    "invoice_number": "string or null",\n'
         '    "invoice_date": "string or null",\n'
+        '    "order_number": "string or null",\n'
+        '    "sales_order_number": "string or null",\n'
+        '    "purchase_order_number": "string or null",\n'
+        '    "order_date": "string or null",\n'
         '    "supplier_name": "string or null",\n'
         '    "supplier_address": "string or null",\n'
+        '    "supplier_phone": "string or null",\n'
+        '    "supplier_email": "string or null",\n'
+        '    "supplier_tax_id": "string or null",\n'
         '    "customer_name": "string or null",\n'
         '    "customer_address": "string or null",\n'
-        '    "po_number": "string or null",\n'
+        '    "customer_phone": "string or null",\n'
+        '    "customer_email": "string or null",\n'
+        '    "customer_tax_id": "string or null",\n'
+        '    "ship_to_name": "string or null",\n'
+        '    "ship_to_address": "string or null",\n'
+        '    "bill_to_name": "string or null",\n'
+        '    "bill_to_address": "string or null",\n'
         '    "tax_id": "string or null",\n'
+        '    "tax_registration_number": "string or null",\n'
+        '    "vat_number": "string or null",\n'
+        '    "currency": "string or null",\n'
         '    "payment_terms": "string or null",\n'
+        '    "payment_method": "string or null",\n'
+        '    "payment_reference": "string or null",\n'
+        '    "bank_account_number": "string or null",\n'
+        '    "iban": "string or null",\n'
+        '    "swift_code": "string or null",\n'
         '    "total_before_tax": "string or null",\n'
         '    "tax_amount": "string or null",\n'
+        '    "tax_rate": "string or null",\n'
+        '    "shipping_charges": "string or null",\n'
+        '    "discount": "string or null",\n'
         '    "total_due": "string or null",\n'
-        '    "currency": "string or null",\n'
+        '    "amount_paid": "string or null",\n'
+        '    "balance_due": "string or null",\n'
         '    "due_date": "string or null",\n'
-        '    "any_other_metadata": "string or null"\n'
+        '    "invoice_status": "string or null",\n'
+        '    "reference_number": "string or null",\n'
+        '    "project_code": "string or null",\n'
+        '    "department": "string or null",\n'
+        '    "contact_person": "string or null",\n'
+        '    "notes": "string or null",\n'
+        '    "additional_info": "string or null"\n'
         '  },\n'
         '  "line_items": [\n'
         '    {\n'
         '      "item_number": "string or null",\n'
+        '      "line_number": "string or null",\n'
+        '      "product_code": "string or null",\n'
+        '      "sku": "string or null",\n'
         '      "description": "string or null",\n'
         '      "quantity": "string or null",\n'
+        '      "unit_of_measure": "string or null",\n'
         '      "unit_price": "string or null",\n'
-        '      "total_price": "string or null",\n'
+        '      "discount": "string or null",\n'
         '      "tax_rate": "string or null",\n'
-        '      "sku": "string or null",\n'
+        '      "tax_amount": "string or null",\n'
+        '      "total_price": "string or null",\n'
+        '      "delivery_date": "string or null",\n'
+        '      "gl_code": "string or null",\n'
+        '      "cost_center": "string or null",\n'
+        '      "project_code": "string or null",\n'
         '      "any_other_line_item_field": "string or null"\n'
         '    }\n'
         '  ]\n'
@@ -157,6 +197,7 @@ def get_extraction_prompt(model_choice, txt):
         f"{txt}"
     )
 
+
 def extract_invoice_info(model_choice, text):
     prompt = get_extraction_prompt(model_choice, text)
     raw = query_llm(model_choice, prompt)