Add permission runtime

Author: lethargicpanda

ai-catalog/samples/gemini-live-todo/build.gradle.kts

@@ -62,6 +62,7 @@ dependencies {
     implementation(libs.hilt.android)
     ksp(libs.hilt.compiler)
     implementation(libs.hilt.navigation.compose)
+    implementation(libs.androidx.activity.compose)
     implementation(libs.androidx.material3.android)
     implementation(libs.kotlinx.serialization.json)
     testImplementation(libs.junit)

ai-catalog/samples/gemini-live-todo/src/main/java/com/android/ai/samples/geminilivetodo/ui/TodoScreen.kt

@@ -16,6 +16,8 @@
 package com.android.ai.samples.geminilivetodo.ui
 
 import android.Manifest
+import android.app.Activity
+import androidx.activity.compose.LocalActivity
 import androidx.annotation.RequiresPermission
 import androidx.compose.animation.Animatable
 import androidx.compose.animation.animateColor
@@ -25,6 +27,7 @@ import androidx.compose.animation.core.infiniteRepeatable
 import androidx.compose.animation.core.rememberInfiniteTransition
 import androidx.compose.animation.core.tween
 import androidx.compose.foundation.background
+import androidx.compose.foundation.layout.Box
 import androidx.compose.foundation.layout.Column
 import androidx.compose.foundation.layout.Row
 import androidx.compose.foundation.layout.Spacer
@@ -43,6 +46,7 @@ import androidx.compose.material.icons.filled.MicOff
 import androidx.compose.material3.AlertDialog
 import androidx.compose.material3.Button
 import androidx.compose.material3.Checkbox
+import androidx.compose.material3.CircularProgressIndicator
 import androidx.compose.material3.ExperimentalMaterial3Api
 import androidx.compose.material3.FabPosition
 import androidx.compose.material3.FloatingActionButton
@@ -64,6 +68,7 @@ import androidx.compose.runtime.setValue
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
 import androidx.compose.ui.graphics.Color
+import androidx.compose.ui.platform.LocalContext
 import androidx.compose.ui.res.stringResource
 import androidx.compose.ui.text.TextStyle
 import androidx.compose.ui.text.style.TextDecoration
@@ -86,16 +91,18 @@ fun TodoScreen(viewModel: TodoScreenViewModel = hiltViewModel()) {
     val uiState by viewModel.uiState.collectAsStateWithLifecycle()
     var text by remember { mutableStateOf("") }
 
+    val activity = LocalActivity.current as Activity
+
     LaunchedEffect(Unit) {
-        viewModel.initializeGeminiLive()
+        viewModel.initializeGeminiLive(activity)
     }
 
     Scaffold(
         topBar = {
             TopAppBar(
                 colors = TopAppBarDefaults.topAppBarColors(
                     containerColor = MaterialTheme.colorScheme.primaryContainer,
-                    titleContentColor = MaterialTheme.colorScheme.onPrimary,
+                    titleContentColor = MaterialTheme.colorScheme.primary,
                 ),
                 title = { Text(stringResource(R.string.gemini_live_title)) },
             )
@@ -127,7 +134,13 @@ fun TodoScreen(viewModel: TodoScreenViewModel = hiltViewModel()) {
 
             when (uiState) {
                 is TodoScreenUiState.Initial -> {
-                    // Show a loading indicator or initial state
+                    Box(
+                        modifier = Modifier
+                            .fillMaxSize(),
+                        contentAlignment = Alignment.Center,
+                    ) {
+                        CircularProgressIndicator()
+                    }
                 }
                 is TodoScreenUiState.Success -> {
                     val todos = (uiState as TodoScreenUiState.Success).todos
@@ -163,7 +176,11 @@ fun TodoScreen(viewModel: TodoScreenViewModel = hiltViewModel()) {
 }
 
 @Composable
-fun TodoInput(text: String, onTextChange: (String) -> Unit, onAddClick: () -> Unit) {
+fun TodoInput(
+    text: String,
+    onTextChange: (String) -> Unit,
+    onAddClick: () -> Unit,
+) {
     Row(
         modifier = Modifier
             .fillMaxWidth()
@@ -188,7 +205,10 @@ fun TodoInput(text: String, onTextChange: (String) -> Unit, onAddClick: () -> Un
 }
 
 @Composable
-fun MicButton(uiState: TodoScreenUiState, onToggle: () -> Unit) {
+fun MicButton(
+    uiState: TodoScreenUiState,
+    onToggle: () -> Unit,
+) {
     if (uiState is TodoScreenUiState.Success) {
         val successState = uiState as TodoScreenUiState.Success
         val micIcon = when {
@@ -237,7 +257,12 @@ fun MicButton(uiState: TodoScreenUiState, onToggle: () -> Unit) {
 }
 
 @Composable
-fun TodoItem(modifier: Modifier, task: Todo, onToggle: () -> Unit, onDelete: () -> Unit) {
+fun TodoItem(
+    modifier: Modifier,
+    task: Todo,
+    onToggle: () -> Unit,
+    onDelete: () -> Unit,
+) {
     val defaultBackgroundColor = Color.Transparent
     val backgroundColor = remember { Animatable(defaultBackgroundColor) }
 

ai-catalog/samples/gemini-live-todo/src/main/java/com/android/ai/samples/geminilivetodo/ui/TodoScreenViewModel.kt

@@ -15,8 +15,14 @@
  */
 package com.android.ai.samples.geminilivetodo.ui
 
+import android.Manifest
+import android.app.Activity
+import android.content.Context
+import android.content.pm.PackageManager
 import android.util.Log
 import androidx.annotation.RequiresPermission
+import androidx.core.app.ActivityCompat
+import androidx.core.content.ContextCompat
 import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
 import com.android.ai.samples.geminilivetodo.data.TodoRepository
@@ -90,7 +96,7 @@ class TodoScreenViewModel @Inject constructor(private val todoRepository: TodoRe
             if (currentState !is TodoScreenUiState.Success) return@launch
 
             session?.let {
-                if (currentState.isLiveSessionRunning) {
+                if (!currentState.isLiveSessionRunning) {
                     it.startAudioConversation(::handleFunctionCall)
                     _uiState.update {
                         if (it is TodoScreenUiState.Success) {
@@ -113,7 +119,8 @@ class TodoScreenViewModel @Inject constructor(private val todoRepository: TodoRe
         }
     }
 
-    fun initializeGeminiLive() {
+    fun initializeGeminiLive(activity: Activity) {
+        requestAudioPermissionIfNeeded(activity)
         viewModelScope.launch {
             Log.d(TAG, "Start Gemini Live initialization")
             val liveGenerationConfig = liveGenerationConfig {
@@ -181,10 +188,10 @@ class TodoScreenViewModel @Inject constructor(private val todoRepository: TodoRe
             } catch (e: Exception) {
                 Log.e(TAG, "Error connecting to the model", e)
                 _uiState.update {
-                    TodoScreenUiState.Error(
-                        isLiveSessionReady = false,
-                        isLiveSessionRunning = false,
-                    )
+                        TodoScreenUiState.Error(
+                            isLiveSessionReady = false,
+                            isLiveSessionRunning = false,
+                        )
                 }
             }
 
@@ -251,4 +258,14 @@ class TodoScreenViewModel @Inject constructor(private val todoRepository: TodoRe
             }
         }
     }
+
+    fun requestAudioPermissionIfNeeded(activity: Activity){
+        if (ContextCompat.checkSelfPermission(
+                activity,
+                Manifest.permission.RECORD_AUDIO
+            ) != PackageManager.PERMISSION_GRANTED
+        ) {
+            ActivityCompat.requestPermissions(activity, arrayOf(Manifest.permission.RECORD_AUDIO), 1)
+        }
+    }
 }