How to Use Firebase ML Kit Smart Replies in Android?

package org.geeksforgeeks.demo;

import android.os.Bundle;
import android.view.View;
import android.widget.Toast;
import androidx.appcompat.app.AppCompatActivity;
import androidx.recyclerview.widget.LinearLayoutManager;
import com.google.firebase.ml.naturallanguage.FirebaseNaturalLanguage;
import com.google.firebase.ml.naturallanguage.smartreply.FirebaseTextMessage;
import com.google.firebase.ml.naturallanguage.smartreply.SmartReply;
import com.google.firebase.ml.naturallanguage.smartreply.SmartReplySuggestionResult;
import org.geeksforgeeks.demo.databinding.ActivityMainBinding;
import java.util.ArrayList;
import java.util.List;

public class MainActivity extends AppCompatActivity {

    private List<FirebaseTextMessage> messageList;
    private ArrayList<MessageModel> messageModelArrayList;
    private Adapter adapter;

    private ActivityMainBinding binding;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        binding = ActivityMainBinding.inflate(getLayoutInflater());
        setContentView(binding.getRoot());

        messageList = new ArrayList<>();
        messageModelArrayList = new ArrayList<>();
        adapter = new Adapter(this, messageModelArrayList);

        LinearLayoutManager manager = new LinearLayoutManager(this);
        binding.recyclerView.setLayoutManager(manager);
        binding.recyclerView.setAdapter(adapter);

        binding.sendButton.setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View v) {
                String message = binding.messageEditText.getText().toString();
                if (message.isEmpty()) {
                    Toast.makeText(MainActivity.this, "Please enter your message..", Toast.LENGTH_SHORT).show();
                    return;
                }
                sendMessage(message);
                binding.messageEditText.setText("");
            }
        });
    }

    private void sendMessage(String userMessage) {
        
        // Create a FirebaseTextMessage
        FirebaseTextMessage message = FirebaseTextMessage.createForRemoteUser(
                userMessage, System.currentTimeMillis(), "uid");

        messageList.add(message);
        
        // 0 -> user message
        messageModelArrayList.add(new MessageModel(userMessage, 0)); 
        
        adapter.notifyDataSetChanged();

        // Firebase Smart Reply
        SmartReply smartReply = FirebaseNaturalLanguage.getInstance().getSmartReply();

        smartReply.suggestReplies(messageList)
                .addOnSuccessListener(smartReplySuggestionResult -> {
                    if (smartReplySuggestionResult.getStatus() == SmartReplySuggestionResult.STATUS_NOT_SUPPORTED_LANGUAGE) {
                        Toast.makeText(MainActivity.this, "Language not supported..", Toast.LENGTH_SHORT).show();
                    } else if (smartReplySuggestionResult.getStatus() == SmartReplySuggestionResult.STATUS_SUCCESS) {
                        if (!smartReplySuggestionResult.getSuggestions().isEmpty()) {
                            messageModelArrayList.add(new MessageModel(smartReplySuggestionResult.getSuggestions().get(0).getText(), 1));
                            adapter.notifyDataSetChanged();
                        }
                    }
                })
                .addOnFailureListener(e -> 
                    Toast.makeText(MainActivity.this, "Fail to get data: " + e.getMessage(), Toast.LENGTH_SHORT).show());
    }
}

package org.geeksforgeeks.demo;

public class MessageModel {
    private String message;
    private int type;

    public MessageModel(String message, int type) {
        this.message = message;
        this.type = type;
    }

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }

    public int getType() {
        return type;
    }

    public void setType(int type) {
        this.type = type;
    }
}

package org.geeksforgeeks.demo;

import android.content.Context;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.TextView;
import androidx.annotation.NonNull;
import androidx.cardview.widget.CardView;
import androidx.core.content.ContextCompat;
import androidx.recyclerview.widget.RecyclerView;
import java.util.ArrayList;

public class Adapter extends RecyclerView.Adapter<Adapter.ViewHolder> {
    private final Context context;
    private final ArrayList<MessageModel> messageModelArrayList;

    public Adapter(Context context, ArrayList<MessageModel> messageModelArrayList) {
        this.context = context;
        this.messageModelArrayList = messageModelArrayList;
    }

    public static class ViewHolder extends RecyclerView.ViewHolder {
        TextView msgTV;
        CardView msgCV;

        public ViewHolder(@NonNull View itemView) {
            super(itemView);
            msgTV = itemView.findViewById(R.id.idTVMessage);
            msgCV = itemView.findViewById(R.id.idCVMessage);
        }
    }

    @NonNull
    @Override
    public ViewHolder onCreateViewHolder(@NonNull ViewGroup parent, int viewType) {
        View view = LayoutInflater.from(parent.getContext()).inflate(R.layout.msg_rv_item, parent, false);
        return new ViewHolder(view);
    }

    @Override
    public void onBindViewHolder(@NonNull ViewHolder holder, int position) {
        MessageModel modal = messageModelArrayList.get(position);

        holder.msgTV.setText(modal.getMessage());

        if (modal.getType() == 1) {
            holder.msgCV.setCardBackgroundColor(ContextCompat.getColor(context, R.color.colorPrimaryDark));
            holder.msgTV.setTextColor(ContextCompat.getColor(context, R.color.white));
        }
    }

    @Override
    public int getItemCount() {
        return messageModelArrayList.size();
    }
}