remittance_pdf_processing_cli.py

# File: remittance_pdf_processing_cli.py

import argparse
from remittance_pdf_processor import process_pdf, extract_text_from_pdf
from ccmt_verifier_aus import verify_invoice_numbers
from remittance_pdf_processing_types import Candidate, ProcessedPDFResult

def main():
 parser = argparse.ArgumentParser(description="Extract text and invoice numbers from a PDF file.")
 parser.add_argument("pdf_file", help="Path to the PDF file")
 parser.add_argument("-o", "--output", help="Output file path for extracted text (optional)")
 parser.add_argument("-v", "--verbose", action="store_true", help="Print extracted text from PDF")
 parser.add_argument("-f", "--force-image", action="store_true", help="Force processing PDF as image")
 parser.add_argument("--verify", action="store_true", help="Enable invoice number verification")
 parser.add_argument("--multi-hop", action="store_true", help="Force multi-hop processing")

 args = parser.parse_args()

 extracted_text = extract_text_from_pdf(args.pdf_file, wrap_pages=True)
 
 # Determine the invoice verifier function
 invoice_verifier = verify_invoice_numbers if args.verify else None

 extracted_result: ProcessedPDFResult = process_pdf(args.pdf_file, 
 force_image_processing=args.force_image,
 force_multi_hop=args.multi_hop,
 invoice_verifier=invoice_verifier)

 verified_candidate, unverified_candidate = extracted_result

 if args.verbose:
 print("Extracted text from PDF:")
 print("-" * 40)
 print(extracted_text)
 print("-" * 40)
 print()

 print("Extracted invoice numbers and payment amounts:")
 print(f"Multi-hop processing: {'Enabled' if args.multi_hop else 'Disabled'}")
 print(f"Verification: {'Enabled' if args.verify else 'Disabled'}")
 print("-" * 40)
 
 print("Verified Results:")
 if len(verified_candidate[0]) == 1 and len(verified_candidate[1]) == 1:
 print(f" Verified Invoice Numbers: {', '.join(verified_candidate[0][0])}")
 print(f" Verified Amount: {verified_candidate[1][0]}")
 elif len(verified_candidate[0]) == 1 and len(verified_candidate[1]) == 0:
 print(f" Verified Invoice Numbers: {', '.join(verified_candidate[0][0])}")
 print(" Amount doesn't match. This might be a partial payment.")
 elif len(verified_candidate[0]) >= 2:
 print(" Verified Invoice Numbers:")
 for i, invoice_numbers in enumerate(verified_candidate[0], 1):
 print(f" List {i}: {', '.join(invoice_numbers)}")
 else:
 print(" No verified results.")
 
 print("\nUnverified Invoice Numbers:")
 for i, invoice_numbers in enumerate(unverified_candidate[0], 1):
 print(f" Candidate {i}: {', '.join(invoice_numbers)}")
 
 print("\nUnverified Payment Amounts:")
 for i, amount in enumerate(unverified_candidate[1], 1):
 print(f" Candidate {i}: {amount}")
 
 print("-" * 40)

 if args.output:
 with open(args.output, 'w', encoding='utf-8') as f:
 f.write("Extracted text:\n")
 f.write(extracted_text)
 f.write("\n\nExtracted invoice numbers and payment amounts:\n")
 f.write(f"Multi-hop processing: {'Enabled' if args.multi_hop else 'Disabled'}\n")
 f.write(f"Verification: {'Enabled' if args.verify else 'Disabled'}\n")
 f.write("Verified Results:\n")
 if len(verified_candidate[0]) == 1 and len(verified_candidate[1]) == 1:
 f.write(f" Verified Invoice Numbers: {', '.join(verified_candidate[0][0])}\n")
 f.write(f" Verified Amount: {verified_candidate[1][0]}\n")
 elif len(verified_candidate[0]) == 1 and len(verified_candidate[1]) == 0:
 f.write(f" Verified Invoice Numbers: {', '.join(verified_candidate[0][0])}\n")
 f.write(" Amount doesn't match. This might be a partial payment.\n")
 elif len(verified_candidate[0]) >= 2:
 f.write(" Verified Invoice Numbers:\n")
 for i, invoice_numbers in enumerate(verified_candidate[0], 1):
 f.write(f" List {i}: {', '.join(invoice_numbers)}\n")
 else:
 f.write(" No verified results.\n")
 f.write("\nUnverified Invoice Numbers:\n")
 for i, invoice_numbers in enumerate(unverified_candidate[0], 1):
 f.write(f" Candidate {i}: {', '.join(invoice_numbers)}\n")
 f.write("\nUnverified Payment Amounts:\n")
 for i, amount in enumerate(unverified_candidate[1], 1):
 f.write(f" Candidate {i}: {amount}\n")
 print(f"Extracted text and invoice numbers written to {args.output}")

if __name__ == "__main__":
 main()