Traduzir textos com o Kit de ML no iOS

É possível usar o kit de aprendizado de máquina para traduzir texto entre idiomas. Atualmente, o Kit de ML oferece suporte para a tradução entre 59 idiomas.

Antes de começar

1. Se você ainda não adicionou o Firebase ao seu app, siga as etapas no guia de iniciação.
2. Inclua as bibliotecas do kit de ML no seu Podfile:

   pod 'Firebase/MLNLTranslate', '6.25.0'
   

   Depois de instalar ou atualizar os pods do seu projeto, abra o projeto do Xcode usando o .xcworkspace.
3. Importe o Firebase para seu app:

   Swift

   import Firebase

   Objective-C

   @import Firebase;

Traduzir uma string de texto

Para traduzir uma string entre dois idiomas, siga estas etapas:

1. Crie um objeto Translator, configurando-o com os idiomas de origem e de destino:

   Swift

   // Create an English-German translator:
   let options = TranslatorOptions(sourceLanguage: .en, targetLanguage: .de)
   let englishGermanTranslator = NaturalLanguage.naturalLanguage().translator(options: options)
   

   Objective-C

   // Create an English-German translator:
   FIRTranslatorOptions *options =
       [[FIRTranslatorOptions alloc] initWithSourceLanguage:FIRTranslateLanguageEN
                                             targetLanguage:FIRTranslateLanguageDE];
   FIRTranslator *englishGermanTranslator =
       [[FIRNaturalLanguage naturalLanguage] translatorWithOptions:options];
   

   Se você não souber o idioma do texto de entrada, use a API de identificação de idioma primeiro. Porém, certifique-se de não manter muitos modelos de idioma no dispositivo ao mesmo tempo.

2. Verifique se o modelo de tradução necessário foi salvo no dispositivo. Não chame translate(_:completion:) até verificar se o modelo está disponível.

   Swift

   let conditions = ModelDownloadConditions(
       allowsCellularAccess: false,
       allowsBackgroundDownloading: true
   )
   englishGermanTranslator.downloadModelIfNeeded(with: conditions) { error in
       guard error == nil else { return }
   
       // Model downloaded successfully. Okay to start translating.
   }
   

   Objective-C

   FIRModelDownloadConditions *conditions =
       [[FIRModelDownloadConditions alloc] initWithAllowsCellularAccess:NO
                                            allowsBackgroundDownloading:YES];
   [englishGermanTranslator downloadModelIfNeededWithConditions:conditions
                                                     completion:^(NSError *_Nullable error) {
     if (error != nil) {
       return;
     }
     // Model downloaded successfully. Okay to start translating.
   }];
   

   Os modelos de idiomas têm cerca de 30 MB, portanto, faça o download deles apenas quando necessário e usando Wi-Fi, a menos que o usuário tenha especificado o contrário. Além disso, exclua modelos desnecessários. Consulte Gerenciar explicitamente modelos de tradução.

3. Depois de confirmar o download do modelo, transmita uma string de texto no idioma de origem para translate(_:completion:):

   Swift

   englishGermanTranslator.translate(text) { translatedText, error in
       guard error == nil, let translatedText = translatedText else { return }
   
       // Translation succeeded.
   }
   

   Objective-C

   [englishGermanTranslator translateText:text
                               completion:^(NSString *_Nullable translatedText,
                                            NSError *_Nullable error) {
     if (error != nil || translatedText == nil) {
       return;
     }
   
     // Translation succeeded.
   }];
   

   O Kit de ML traduz o texto para o idioma de destino configurado e transmite o texto traduzido para o gerenciador de conclusão.

Gerenciar explicitamente modelos de tradução

Quando a API de tradução é usada conforme descrito acima, o Kit de ML faz o download automático de modelos de tradução específicos do idioma para o dispositivo, conforme necessário. Também é possível gerenciar explicitamente os modelos de tradução que você quer disponibilizar no dispositivo usando a API de gerenciamento de modelos de tradução do Kit de ML. Isso pode ser útil se você quiser fazer o download de modelos antecipadamente ou excluir modelos desnecessários do dispositivo.

Para receber os modelos de tradução armazenados no dispositivo:

let localModels = ModelManager.modelManager().downloadedTranslateModels

NSSet<FIRTranslateRemoteModel *> *localModels =
    [FIRModelManager modelManager].downloadedTranslateModels;

Para excluir um modelo:

// Delete the German model if it's on the device.
let deModel = TranslateRemoteModel.translateRemoteModel(language: .de)
ModelManager.modelManager().deleteDownloadedModel(deModel) { error in
    guard error == nil else { return }
    // Model deleted.
}

// Delete the German model if it's on the device.
FIRTranslateRemoteModel *deModel =
    [FIRTranslateRemoteModel translateRemoteModelWithLanguage:FIRTranslateLanguageDE];
[[FIRModelManager modelManager] deleteDownloadedModel:deModel
                                           completion:^(NSError * _Nullable error) {
                                               if (error != nil) {
                                                   return;
                                               }
                                               // Model deleted.
                                           }];

Para fazer o download de um modelo:

// Download the French model.
let frModel = TranslateRemoteModel.translateRemoteModel(language: .fr)

// Keep a reference to the download progress so you can check that the model
// is available before you use it.
progress = ModelManager.modelManager().download(
    frModel,
    conditions: ModelDownloadConditions(
        allowsCellularAccess: false,
        allowsBackgroundDownloading: true
    )
)

NotificationCenter.default.addObserver(
    forName: .firebaseMLModelDownloadDidSucceed,
    object: nil,
    queue: nil
) { [weak self] notification in
    guard let strongSelf = self,
        let userInfo = notification.userInfo,
        let model = userInfo[ModelDownloadUserInfoKey.remoteModel.rawValue]
            as? TranslateRemoteModel,
        model == frModel
        else { return }
    // The model was downloaded and is available on the device
}

NotificationCenter.default.addObserver(
    forName: .firebaseMLModelDownloadDidFail,
    object: nil,
    queue: nil
) { [weak self] notification in
    guard let strongSelf = self,
        let userInfo = notification.userInfo,
        let model = userInfo[ModelDownloadUserInfoKey.remoteModel.rawValue]
            as? TranslateRemoteModel
        else { return }
    let error = userInfo[ModelDownloadUserInfoKey.error.rawValue]
    // ...
}

// Download the French model.
FIRModelDownloadConditions *conditions =
    [[FIRModelDownloadConditions alloc] initWithAllowsCellularAccess:NO
                                         allowsBackgroundDownloading:YES];
FIRTranslateRemoteModel *frModel =
    [FIRTranslateRemoteModel translateRemoteModelWithLanguage:FIRTranslateLanguageFR];

// Keep a reference to the download progress so you can check that the model
// is available before you use it.
self.downloadProgress = [[FIRModelManager modelManager] downloadModel:frModel
                                                           conditions:conditions];

__block MyViewController *weakSelf = self;

[NSNotificationCenter.defaultCenter
 addObserverForName:FIRModelDownloadDidSucceedNotification
 object:nil
 queue:nil
 usingBlock:^(NSNotification * _Nonnull note) {
     if (weakSelf == nil | note.userInfo == nil) {
         return;
     }

     FIRTranslateRemoteModel *model = note.userInfo[FIRModelDownloadUserInfoKeyRemoteModel];
     if ([model isKindOfClass:[FIRTranslateRemoteModel class]]
         && model == frModel) {
         // The model was downloaded and is available on the device
     }
 }];

[NSNotificationCenter.defaultCenter
 addObserverForName:FIRModelDownloadDidFailNotification
 object:nil
 queue:nil
 usingBlock:^(NSNotification * _Nonnull note) {
     if (weakSelf == nil | note.userInfo == nil) {
         return;
     }

     NSError *error = note.userInfo[FIRModelDownloadUserInfoKeyError];
 }];