Android ML Kit 翻译：如何检测已下载的语言模型状态

import com.google.mlkit.nl.translate.TranslateLanguage;
import com.google.mlkit.nl.translate.Translation;
import com.google.mlkit.nl.translate.Translator;
import com.google.mlkit.nl.translate.TranslatorOptions;
import com.google.mlkit.nl.translate.DownloadConditions;

import android.util.Log;
import androidx.annotation.NonNull;
import com.google.android.gms.tasks.OnFailureListener;
import com.google.android.gms.tasks.OnSuccessListener;

public class TranslationModelChecker {

    private static final String TAG = "TranslationModelChecker";

    /**
     * 检查特定语言模型（例如，从英语到德语）的下载状态。
     *
     * @param sourceLanguage 源语言（例如：TranslateLanguage.ENGLISH）
     * @param targetLanguage 目标语言（例如：TranslateLanguage.GERMAN）
     * @param onModelStatusListener 用于回调模型状态的监听器
     */
    public void checkTranslationModelStatus(
            @TranslateLanguage.LanguageType String sourceLanguage,
            @TranslateLanguage.LanguageType String targetLanguage,
            final OnModelStatusListener onModelStatusListener) {

        // 1. 创建 Translator 实例
        TranslatorOptions options = new TranslatorOptions.Builder()
                .setSourceLanguage(sourceLanguage)
                .setTargetLanguage(targetLanguage)
                .build();
        final Translator translator = Translation.getClient(options);

        // 2. 定义下载条件 (例如，只在 Wi-Fi 下下载)
        DownloadConditions conditions = new DownloadConditions.Builder()
                .requireWifi() // 要求 Wi-Fi 网络
                // .requireCharging() // 如果需要，也可以要求充电状态
                .build();

        // 3. 调用 downloadModelIfNeeded() 并添加监听器
        translator.downloadModelIfNeeded(conditions)
                .addOnSuccessListener(new OnSuccessListener<Void>() {
                    @Override
                    public void onSuccess(Void v) {
                        // 模型已下载或已存在。
                        // 此时可以开始翻译，或者更新UI表示模型可用。
                        Log.d(TAG, "Model for " + sourceLanguage + " to " + targetLanguage + " is ready.");
                        if (onModelStatusListener != null) {
                            onModelStatusListener.onModelReady();
                        }
                        // 翻译器在使用完毕后应关闭
                        translator.close();
                    }
                })
                .addOnFailureListener(new OnFailureListener() {
                    @Override
                    public void onFailure(@NonNull Exception e) {
                        // 模型无法下载或发生其他内部错误。
                        // 此时应处理错误，例如显示错误信息，或隐藏翻译UI。
                        Log.e(TAG, "Failed to download or check model for " + sourceLanguage + " to " + targetLanguage + ": " + e.getMessage(), e);
                        if (onModelStatusListener != null) {
                            onModelStatusListener.onModelDownloadFailed(e);
                        }
                        // 翻译器在使用完毕后应关闭
                        translator.close();
                    }
                });
    }

    /**
     * 自定义监听器接口，用于回调模型下载状态。
     */
    public interface OnModelStatusListener {
        void onModelReady();
        void onModelDownloadFailed(Exception e);
    }

    // 示例用法
    public static void main(String[] args) {
        // 假设在一个 Android Activity 或 Fragment 中调用
        // TranslationModelChecker checker = new TranslationModelChecker();
        // checker.checkTranslationModelStatus(
        //         TranslateLanguage.ENGLISH,
        //         TranslateLanguage.GERMAN,
        //         new OnModelStatusListener() {
        //             @Override
        //             public void onModelReady() {
        //                 // 模型已就绪，显示翻译UI或离线图标
        //                 System.out.println("Model is ready! You can now translate.");
        //             }
        //
        //             @Override
        //             public void onModelDownloadFailed(Exception e) {
        //                 // 模型下载失败，隐藏翻译UI或显示错误信息
        //                 System.err.println("Model download failed: " + e.getMessage());
        //             }
        //         });
    }
}