bergamot-translator/app/bergamot-translator-app-bytearray.cpp

/*
 * main.cpp
 *
 * An example application to demonstrate the use of Bergamot translator.
 *
 */

#include <iostream>

#include "TranslationModel.h"
#include "translator/parser.h"
#include "translator/byte_array_util.h"

int main(int argc, char **argv) {

  // Create a configParser and load command line parameters into a YAML config
  // string.
  auto configParser = marian::bergamot::createConfigParser();
  auto options = configParser.parseOptions(argc, argv, true);
  std::string config = options->asYamlString();

  // Prepare model byte array
  marian::bergamot::AlignedMemory modelBytes = marian::bergamot::getModelMemoryFromConfig(options);

  // Route the config string to construct marian model through TranslationModel
  TranslationModel model(config, modelBytes.begin());

  TranslationRequest translationRequest;
  std::vector<std::string> texts;
  texts.emplace_back(
      "The Bergamot project will add and improve client-side machine "
      "translation in a web browser.  Unlike current cloud-based "
      "options, running directly on users’ machines empowers citizens to "
      "preserve their privacy and increases the uptake of language "
      "technologies in Europe in various sectors that require "
      "confidentiality.");
  texts.emplace_back(
      "Free software integrated with an open-source web "
      "browser, such as Mozilla Firefox, will enable bottom-up adoption "
      "by non-experts, resulting in cost savings for private and public "
      "sector users who would otherwise procure translation or operate "
      "monolingually.  Bergamot is a consortium coordinated by the "
      "University of Edinburgh with partners Charles University in "
      "Prague, the University of Sheffield, University of Tartu, and "
      "Mozilla.");

  auto results = model.translate(std::move(texts), translationRequest);

  // Resolve the future and get the actual result
  //std::vector<TranslationResult> results = futureResults.get();

  for (auto &result : results) {
    std::cout << "[original]: " << result.getOriginalText() << std::endl;
    std::cout << "[translated]: " << result.getTranslatedText() << std::endl;
    auto mappings = result.getSentenceMappings();
    for (auto &p : mappings) {
      std::string_view src = p.first;
      std::string_view tgt = p.second;

      std::cout << " [src Sentence]: " << src << std::endl;
      std::cout << " [tgt Sentence]: " << tgt << std::endl;
    }
    std::cout << std::endl;
  }

  return 0;
}