Align LiteRT integration with AI Edge Gallery setup

Author: Android-PowerUser

app/build.gradle.kts

@@ -111,7 +111,7 @@ dependencies {
     // MediaPipe GenAI for offline inference (LLM)
     implementation("com.google.mediapipe:tasks-genai:0.10.32")
     // LiteRT-LM for newer offline .litertlm models (e.g. Gemma 4 E4B it)
-    implementation("com.google.ai.edge.litertlm:litertlm-android:0.0.0-alpha06")
+    implementation("com.google.ai.edge.litertlm:litertlm-android:0.10.0")
 
     // Camera Core to potentially fix missing JNI lib issue
     implementation("androidx.camera:camera-core:1.4.0")

app/src/main/kotlin/com/google/ai/sample/feature/multimodal/PhotoReasoningViewModel.kt

@@ -86,7 +86,6 @@ class PhotoReasoningViewModel(
 
     private var llmInference: LlmInference? = null
     private var liteRtEngine: Engine? = null
-    private var liteRtNativeLoaded = false
     private val TAG = "PhotoReasoningViewModel"
 
     // WebRTC & Signaling
@@ -344,11 +343,10 @@ class PhotoReasoningViewModel(
                             "abis=${Build.SUPPORTED_ABIS?.joinToString() ?: "unknown"}, " +
                             "modelPath=${modelFile.absolutePath}, modelSizeBytes=${modelFile.length()}"
                     )
-                    ensureLiteRtNativeLoaded()
                     if (liteRtEngine == null) {
-                        val liteRtBackend = if (backend == InferenceBackend.GPU) Backend.GPU else Backend.CPU
-                        val visionBackend = Backend.CPU
-                        val audioBackend = Backend.CPU
+                        val liteRtBackend = if (backend == InferenceBackend.GPU) Backend.GPU() else Backend.CPU()
+                        val visionBackend = if (currentModel.supportsScreenshot) Backend.GPU() else null
+                        val audioBackend = null
                         val engineConfig = EngineConfig(
                             modelPath = modelFile.absolutePath,
                             backend = liteRtBackend,
@@ -413,14 +411,6 @@ class PhotoReasoningViewModel(
         }
     }
 
-    private fun ensureLiteRtNativeLoaded() {
-        if (liteRtNativeLoaded) return
-
-        System.loadLibrary("litertlm_jni")
-
-        liteRtNativeLoaded = true
-    }
-
     private fun isLiteRtAbiSupported(): Boolean {
         val supportedAbis = Build.SUPPORTED_ABIS?.toSet().orEmpty()
         return supportedAbis.contains("arm64-v8a") || supportedAbis.contains("x86_64")