Could not import project 'FirebaseUI-Android'

I tried importing via import project wizard, didn't work. This project is using Gradle Kotlin DSL. I am using Android Studio 3.1.4.

I tried running gradle idea and got this message

FAILURE: Build failed with an exception.

* Where:
Build file '/Users/user/Projects/firebaseauthui/FirebaseUI-Android/build.gradle.kts' line: 400

* What went wrong:
SDK location not found. Define location with sdk.dir in the local.properties file or with an ANDROID_HOME environment variable.

* Try:
Run with --stacktrace option to get the stack trace. Run with --info or --debug option to get more log output. Run with --scan to get full insights.

What is the recommended way to import this project as they have migrated to Kotlin DSL?

from Could not import project 'FirebaseUI-Android'