using KVPExtraction.DocumentConverterService;
using System;
using System.IO;
using System.ServiceModel;
namespace KVPExtraction
{
    class Program
    {
        // The URL where the Web Service is located. Amend host name if needed.
        static string SERVICE_URL = "https://localhost:41734/Muhimbi.DocumentConverter.WebService/";
        static void Main(string[] args)
        {
            DocumentConverterServiceClient client = null;
            try
            {
                // Determine the source file and read it into a byte array.
                string sourceFileName = null;
                if (args.Length == 0)
                {
                    // If nothing is specified then read the first PDF file from the current folder.
                    string[] sourceFiles = Directory.GetFiles(Directory.GetCurrentDirectory(), "*.pdf");
                    if (sourceFiles.Length > 0)
                        sourceFileName = sourceFiles[0];
                    else
                    {
                        Console.WriteLine("Please specify a document to extract key-value pairs from.");
                        Console.ReadKey();
                        return;
                    }
                }
                else
                    sourceFileName = args[0];
                string expectedKeys = null;
                if (args.Length > 1)
                {
                    // The second argument is the expected keys file. Read its content.
                    expectedKeys = File.ReadAllText(args[1]);
                }
                else
                {
                    // Uncomment the line below if you wish to use the following expected keys.
                    //expectedKeys = "[{\"expectedKey\":\"grand total\",\"synonyms\":[\"total\"]},{\"expectedKey\":\"invoice number\",\"synonyms\":[\"invoice no.\"]}]";
                }
                byte[] sourceFile = File.ReadAllBytes(sourceFileName);
                // Open the service and configure the bindings.
                client = OpenService(SERVICE_URL);
                // Set the absolute minimum open options.
                OpenOptions openOptions = new OpenOptions();
                openOptions.OriginalFileName = Path.GetFileName(sourceFileName);
                openOptions.FileExtension = Path.GetExtension(sourceFileName);
                // Set the parameters for extracting key-value pairs.
                KVPSettings kvpSettings = new KVPSettings()
                {
                    // Specify whether or not the internal image should be automatically rotated.
                    AutoRotate = BooleanEnum.True,
                    // Only include results with a confidence higher than the threshold (in %).
                    ConfidenceThreshold = 50,
                    // Specify virtual resolution for retrieving data. Higher values will make the process more accurate but slower.
                    DPI = 300,
                    // Specify the expected keys or leave empty to resolve all pairs.
                    // Please note that extra information isn't included when `ExpectedKeys` is used.
                    ExpectedKeys = expectedKeys,
                    // Specify whether or not the confidence value for the extracted key-value pair should be included in the result.
                    IncludeConfidence = BooleanEnum.True,
                    // Specify whether or not the bounding box information for the key should be included.
                    IncludeKeyBoundingBox = BooleanEnum.True,
                    // Specify whether or not the page number the pair was found on should be included.
                    IncludePageNumber = BooleanEnum.True,
                    // Specify whether or not the type of the value should be included.
                    IncludeType = BooleanEnum.True,
                    // Specify whether or not the bounding box of the value should be included.
                    IncludeValueBoundingBox = BooleanEnum.True,
                    // Specify the desired output format (XML, JSON, or CSV).
                    KVPFormat = KVPOutputFormat.CSV,
                    // Specify the language used for OCR.
                    OCRLanguage = "eng",
                    // Specify the range of pages to search on.
                    PageRange = null,
                    // Specify whether or not extra symbols should be trimmed from values.
                    TrimSymbols = BooleanEnum.False
                };
                Console.WriteLine("Extracting key-value pairs.");
                // Carry out the extraction.
                byte[] result = client.ExtractKeyValuePairs(sourceFile, openOptions, kvpSettings);
                if (result != null)
                {
                    string destinationFileName = Path.GetFileNameWithoutExtension(sourceFileName) + "." + kvpSettings.KVPFormat;
                    using (FileStream fs = File.Create(destinationFileName))
                    {
                        fs.Write(result, 0, result.Length);
                        fs.Close();
                    }
                    Console.WriteLine("Result saved into " + destinationFileName);
                }
                else
                {
                    Console.WriteLine("Nothing returned.");
                }
                Console.WriteLine("Finished.");
            }
            catch (FaultException<WebServiceFaultException> ex)
            {
                Console.WriteLine("FaultException occurred: ExceptionType: " +
                                 ex.Detail.ExceptionType.ToString());
            }
            catch (Exception ex)
            {
                Console.WriteLine(ex.ToString());
            }
            finally
            {
                CloseService(client);
            }
            Console.ReadKey();
        }
        /// <summary>
        /// Configure the bindings and endpoints and open the service using the specified address.
        /// </summary>
        /// <returns>An instance of the Web Service.</returns>
        public static DocumentConverterServiceClient OpenService(string address)
        {
            DocumentConverterServiceClient client = null;
            try
            {
                BasicHttpBinding binding = new BasicHttpBinding();
                // Use standard Windows Security.
                binding.Security.Mode = BasicHttpSecurityMode.TransportCredentialOnly;
                binding.Security.Transport.ClientCredentialType =
                                                                HttpClientCredentialType.Windows;
                // Increase the client Timeout to deal with (very) long running requests.
                binding.SendTimeout = TimeSpan.FromMinutes(120);
                binding.ReceiveTimeout = TimeSpan.FromMinutes(120);
                // Set the maximum document size to 50MB.
                binding.MaxReceivedMessageSize = 50 * 1024 * 1024;
                binding.ReaderQuotas.MaxArrayLength = 50 * 1024 * 1024;
                binding.ReaderQuotas.MaxStringContentLength = 50 * 1024 * 1024;
                // Specify an identity (any identity) to get it past .net3.5 sp1.
                EndpointIdentity epi = EndpointIdentity.CreateUpnIdentity("unknown");
                EndpointAddress epa = new EndpointAddress(new Uri(address), epi);
                client = new DocumentConverterServiceClient(binding, epa);
                client.Open();
                return client;
            }
            catch (Exception)
            {
                CloseService(client);
                throw;
            }
        }
        /// <summary>
        /// Check if the client is open and then close it.
        /// </summary>
        /// <param name="client">The client to close.</param>
        public static void CloseService(DocumentConverterServiceClient client)
        {
            if (client != null && client.State == CommunicationState.Opened)
                client.Close();
        }
    }
}

[
  {
    "expectedKey": "grand total",
    "synonyms": ["total"]
  },
  {
    "expectedKey": "invoice number",
    "synonyms": ["invoice no."]
  }
]