lib/mindee/product/business_card/business_card_v1.rb


# frozen_string_literal: true

require_relative '../../parsing'
require_relative 'business_card_v1_document'
require_relative 'business_card_v1_page'

module Mindee
  module Product
    # Business Card module.
    module BusinessCard
      # Business Card API version 1 inference prediction.
      class BusinessCardV1 < Mindee::Parsing::Common::Inference
        @endpoint_name = 'business_card'
        @endpoint_version = '1'
        @has_async = true
        @has_sync = false

        # @param prediction [Hash]
        def initialize(prediction)
          super
          @prediction = BusinessCardV1Document.new(prediction['prediction'], nil)
          @pages = []
          prediction['pages'].each do |page|
            @pages.push(BusinessCardV1Page.new(page))
          end
        end

        class << self
          # Name of the endpoint for this product.
          # @return [String]
          attr_reader :endpoint_name
          # Version for this product.
          # @return [String]
          attr_reader :endpoint_version
          # Whether this product has access to an asynchronous endpoint.
          # @return [bool]
          attr_reader :has_async
          # Whether this product has access to synchronous endpoint.
          # @return [bool]
          attr_reader :has_sync
        end
      end
    end
  end
end