Update model list screen to the new design.

Also update ux for multi-image message

PiperOrigin-RevId: 791036851

Author: Google AI Edge Gallery

Android/src/app/src/main/java/com/google/ai/edge/gallery/data/Consts.kt

@@ -16,6 +16,8 @@
 
 package com.google.ai.edge.gallery.data
 
+import androidx.compose.ui.unit.dp
+
 // Keys used to send/receive data to Work.
 const val KEY_MODEL_URL = "KEY_MODEL_URL"
 const val KEY_MODEL_NAME = "KEY_MODEL_NAME"
@@ -53,3 +55,6 @@ const val MAX_AUDIO_CLIP_DURATION_SEC = 30
 
 // Audio-recording related consts.
 const val SAMPLE_RATE = 16000
+
+// The size the icon shown under each of the model names in the model list screen.
+val MODEL_INFO_ICON_SIZE = 18.dp

Android/src/app/src/main/java/com/google/ai/edge/gallery/data/Model.kt

@@ -93,6 +93,9 @@ data class Model(
   /** Whether the model is imported or not. */
   val imported: Boolean = false,
 
+  /* The task type ids that this model is best for. */
+  val bestForTaskTypes: List<String> = listOf(),
+
   // The following fields are managed by the app. Don't need to set manually.
   var normalizedName: String = "",
   var instance: Any? = null,

Android/src/app/src/main/java/com/google/ai/edge/gallery/data/ModelAllowlist.kt

@@ -40,6 +40,7 @@ data class AllowedModel(
   val llmSupportImage: Boolean? = null,
   val llmSupportAudio: Boolean? = null,
   val estimatedPeakMemoryInBytes: Long? = null,
+  val bestForTaskTypes: List<String>? = null,
 ) {
   fun toModel(): Model {
     // Construct HF download url.
@@ -98,6 +99,7 @@ data class AllowedModel(
       learnMoreUrl = "https://huggingface.co/${modelId}",
       llmSupportImage = llmSupportImage == true,
       llmSupportAudio = llmSupportAudio == true,
+      bestForTaskTypes = bestForTaskTypes ?: listOf(),
     )
   }
 

Android/src/app/src/main/java/com/google/ai/edge/gallery/data/Tasks.kt

@@ -112,7 +112,6 @@ val TASK_LLM_ASK_AUDIO =
     type = TaskType.LLM_ASK_AUDIO,
     icon = Icons.Outlined.Mic,
     models = mutableListOf(),
-    // TODO(do not submit)
     description =
       "Instantly transcribe and/or translate audio clips using on-device large language models",
     docUrl = "https://ai.google.dev/edge/mediapipe/solutions/genai/llm_inference/android",
@@ -142,5 +141,11 @@ fun processTasks() {
     for (model in task.models) {
       model.preProcess()
     }
+    // Move the model that is best for this task to the front.
+    val bestModel = task.models.find { it.bestForTaskTypes.contains(task.type.id) }
+    if (bestModel != null) {
+      task.models.remove(bestModel)
+      task.models.add(0, bestModel)
+    }
   }
 }

Android/src/app/src/main/java/com/google/ai/edge/gallery/ui/common/ClickableLink.kt

@@ -39,7 +39,7 @@ import com.google.ai.edge.gallery.firebaseAnalytics
 import com.google.ai.edge.gallery.ui.theme.customColors
 
 @Composable
-fun ClickableLink(url: String, linkText: String, icon: ImageVector) {
+fun ClickableLink(url: String, linkText: String, icon: ImageVector? = null) {
   val uriHandler = LocalUriHandler.current
   val annotatedText =
     AnnotatedString(
@@ -59,7 +59,9 @@ fun ClickableLink(url: String, linkText: String, icon: ImageVector) {
     )
 
   Row(verticalAlignment = Alignment.CenterVertically, horizontalArrangement = Arrangement.Center) {
-    Icon(icon, contentDescription = "", modifier = Modifier.size(16.dp))
+    if (icon != null) {
+      Icon(icon, contentDescription = "", modifier = Modifier.size(16.dp))
+    }
     Text(
       text = annotatedText,
       textAlign = TextAlign.Center,