How to Implement TabLayout with Icon in Android?

package org.geeksforgeeks.demo;

import android.os.Bundle;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.TextView;
import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.fragment.app.Fragment;

public class MainFragment extends Fragment {
    private TextView textView;

    @Nullable
    @Override
    public View onCreateView(@NonNull LayoutInflater inflater, @Nullable ViewGroup container, @Nullable Bundle savedInstanceState) {
        // Inflate the layout for this fragment
        View view = inflater.inflate(R.layout.fragment_main, container, false);

        // Assign variable
        textView = view.findViewById(R.id.text_view);

        // Get title from arguments (if available)
        Bundle args = getArguments();
        if (args != null) {
            String sTitle = args.getString("title");
            // Set title on TextView
            textView.setText(sTitle);
        }

        // Return the view
        return view;
    }
}

package org.geeksforgeeks.demo;

import android.os.Bundle;
import androidx.annotation.NonNull;
import androidx.fragment.app.Fragment;
import androidx.fragment.app.FragmentActivity;
import androidx.viewpager2.adapter.FragmentStateAdapter;
import java.util.List;

public class ViewPagerAdapter extends FragmentStateAdapter {
    private final List<String> titles;

    // Constructor
    public ViewPagerAdapter(@NonNull FragmentActivity activity, List<String> titles) {
        super(activity);
        this.titles = titles;
    }

    // Returns the number of items (fragments) in ViewPager
    @Override
    public int getItemCount() {
        return titles.size();
    }

    // Creates and returns a fragment for a given position
    @NonNull
    @Override
    public Fragment createFragment(int position) {
        MainFragment fragment = new MainFragment();
        Bundle args = new Bundle();
        args.putString("title", titles.get(position));
        fragment.setArguments(args);
        return fragment;
    }
}

package org.geeksforgeeks.demo;

import android.os.Bundle;
import androidx.appcompat.app.AppCompatActivity;
import androidx.core.content.ContextCompat;
import androidx.viewpager2.widget.ViewPager2;
import com.google.android.material.tabs.TabLayout;
import com.google.android.material.tabs.TabLayoutMediator;
import java.util.Arrays;
import java.util.List;

public class MainActivity extends AppCompatActivity {

    private TabLayout tabLayout;
    private ViewPager2 viewPager;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        // Initialize views
        tabLayout = findViewById(R.id.tab_layout);
        viewPager = findViewById(R.id.view_pager);

        // List of tab titles & icons
        List<String> tabTitles = Arrays.asList("Home", "Profile", "Settings");
        int[] tabIcons = {R.drawable.home, R.drawable.person, R.drawable.settings};

        // Set up ViewPager2 with adapter
        viewPager.setAdapter(new ViewPagerAdapter(this, tabTitles));

        // Connect TabLayout with ViewPager2 using TabLayoutMediator
        new TabLayoutMediator(tabLayout, viewPager, (tab, position) -> {
            tab.setText(tabTitles.get(position));
            tab.setIcon(ContextCompat.getDrawable(this, tabIcons[position]));
        }).attach();
    }
}