17370845950

ML Kit 翻译:获取已下载语言模型的列表

本教程详细介绍了如何使用 google ml kit 的 `modelmanager` 获取已下载的翻译语言模型列表。通过 `getdownloadedmodels` 方法获取 `translateremotemodel` 集合后,可进一步迭代并使用 `getlanguage()` 方法提取每个模型的 bcp-47 语言标签,从而清晰地获取已下载语言的名称。

1. 理解 ML Kit 翻译模型管理

Google ML Kit 提供了强大的翻译功能,支持离线翻译,这需要用户预先下载所需的语言模型。ModelManager 是 ML Kit 中用于管理这些模型的核心组件,它允许开发者查询、下载和删除语言模型。当我们需要了解设备上已下载了哪些语言模型时,ModelManager 提供了一个接口来获取这些信息。

2. 获取已下载的语言模型集合

要获取设备上所有已下载的翻译语言模型,我们需要使用 ModelManager 的 getDownloadedModels() 方法。此方法接受一个模型类作为参数,并返回一个 Set 集合,其中包含该类型的所有已下载模型。对于翻译功能,我们关注的是 TranslateRemoteModel。

以下是获取已下载 TranslateRemoteModel 集合的基本代码:

import com.google.mlkit.common.model.RemoteModelManager;
import com.google.mlkit.nl.translate.TranslateRemoteModel;
import android.util.Log;
import java.util.Set;

public class LanguageModelManager {

    private static final String TAG = "DownloadedLanguages";

    public void getDownloadedTranslationLanguages() {
        RemoteModelManager modelManager = RemoteModelManager.getInstance();

        modelManager.getDownloadedModels(TranslateRemoteModel.class)
                .addOnSuccessListener(translateRemoteModels -> {
                    // 成功获取到已下载模型的集合
                    Log.d(TAG, "onSuccess: 已下载的翻译模型集合: " + translateRemoteModels.toString());
                    // 此时 translateRemoteModels 是 Set 类型
                    // 接下来我们将从这些对象中提取语言名称
                })
                .addOnFailureListener(e -> {
                    // 获取失败
                    Log.e(TAG, "onFailure: 获取已下载模型失败", e);
                });
    }
}

执行上述代码后,onSuccess 回调会接收到一个 Set 集合。TranslateRemoteModel 对象本身包含了模型的元数据,但其 toString() 方法可能不会直接显示易读的语言名称,而是类似对象引用的字符串表示。

3. 从模型对象中提取语言名称

为了获取具体的语言名称(例如 "en" 代表英语,"es" 代表西班牙语),我们需要遍历 Set 集合中的每个 TranslateRemoteModel 对象,并调用其 getLanguage() 方法。getLanguage() 方法会返回一个符合 BCP-47 标准的语言标签字符串。

在 onSuccessListener 内部,我们可以添加以下逻辑来提取语言名称:

// ... 在 onSuccessListener 内部 ...
.addOnSuccessListener(translateRemoteModels -> {
    Log.d(TAG, "onSuccess: 已下载的翻译模型集合: " + translateRemoteModels.toString());

    if (translateRemoteModels.isEmpty()) {
        Log.d(TAG, "onSuccess: 当前没有已下载的翻译语言模型。");
        return;
    }

    Log.d(TAG, "onSuccess: 已下载的语言名称列表:");
    for (TranslateRemoteModel trm : translateRemoteModels) {
        String languageTag = trm.getLanguage(); // 获取 BCP-47 语言标签
        Log.d(TAG, "  - 语言标签: " + languageTag);
        // 您可以将这些语言标签存储在一个列表中,或用于UI显示
    }
})
// ...

4. 完整代码示例

结合上述步骤,以下是一个完整的示例,展示如何获取并打印所有已下载翻译语言的名称:

import androidx.annotation.NonNull;
import com.google.android.gms.tasks.OnFailureListener;
import com.google.android.gms.tasks.OnSuccessListener;
import com.google.mlkit.common.model.RemoteModelManager;
import com.google.mlkit.nl.translate.TranslateRemoteModel;
import android.util.Log;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;

public class MLKitTranslationLanguageHelper {

    private static final String TAG = "MLKitLangHelper";

    /**
     * 获取所有已下载的 ML Kit 翻译语言模型,并返回其 BCP-47 语言标签列表。
     *
     * @param listener 用于接收结果的回调接口。
     */
    public void getDownloadedTranslationLanguageTags(LanguageTagsListener listener) {
        RemoteModelManager modelManager = RemoteModelManager.getInstance();

        modelManager.getDownloadedModels(TranslateRemoteModel.class)
                .addOnSuccessListener(new OnSuccessListener>() {
                    @Override
                    public void onSuccess(Set translateRemoteModels) {
                        List downloadedLanguages = new ArrayList<>();
                        if (translateRemoteModels.isEmpty()) {
                            Log.d(TAG, "当前没有已下载的翻译语言模型。");
                        } else {
                            Log.d(TAG, "已下载的翻译语言模型:");
                            for (TranslateRemoteModel trm : translateRemoteModels) {
                                String languageTag = trm.getLanguage();
                                downloadedLanguages.add(languageTag);
                                Log.d(TAG, "  - " + languageTag);
                            }
                        }
                        if (listener != null) {
                            listener.onSuccess(downloadedLanguages);
                        }
                    }
                })
                .addOnFailureListener(new OnFailureListener() {
                    @Override
                    public void onFailure(@NonNull Exception e) {
                        Log.e(TAG, "获取已下载翻译模型失败", e);
                        if (listener != null) {
                            listener.onFailure(e);
                        }
                    }
                });
    }

    /**
     * 回调接口,用于处理获取语言标签的结果。
     */
    public interface LanguageTagsListener {
        void onSuccess(List languageTags);
        void onFailure(Exception e);
    }

    // 示例如何使用这个帮助类
    public static void main(String[] args) { // 注意:此main方法仅为示例,在Android中应在Activity/Fragment中使用
        new MLKitTranslationLanguageHelper().getDownloadedTranslationLanguageTags(new LanguageTagsListener() {
            @Override
            public void onSuccess(List languageTags) {
                Log.i(TAG, "成功获取到已下载语言列表: " + languageTags);
            }

            @Override
            public void onFailure(Exception e) {
                Log.e(TAG, "获取已下载语言列表失败", e);
            }
        });
    }
}

5. 注意事项

  • 异步操作: getDownloadedModels 是一个异步操作,结果通过 OnSuccessListener 和 OnFailureListener 回调返回。这意味着您不能立即在调用后获取结果,而必须在回调中处理。
  • 错误处理: 务必实现 OnFailureListener 来处理可能发生的错误,例如设备存储不足、网络问题或 ML Kit 内部错误。
  • 语言标签格式: getLanguage() 方法返回的是 BCP-47 语言标签,例如 "en" (英语), "es" (西班牙语), "zh" (中文)。这些标签是国际标准,在 ML Kit 中用于标识语言。
  • 空集合: 如果设备上没有下载任何翻译语言模型,onSuccess 回调将返回一个空的 Set。在处理结果时应考虑这种情况。
  • UI 更新: 如果您计划在用户界面中显示这些语言名称,请确保在主线程(UI 线程)上执行任何 UI 更新操作。

6. 总结

通过利用 ML Kit 的 RemoteModelManager 和 TranslateRemoteModel 类的 getLanguage() 方法,开发者可以轻松地获取设备上所有已下载翻译语言模型的列表。这种方法简洁高效,是管理 ML Kit 离线翻译功能的重要一环,确保应用程序能够准确地了解和利用可用的语言资源。