fix token badge, hide pills on keyboard, reset tokens on nav

Three fixes:
1. Token badge only shown when usedTokens > 0 (not in old chats
   where we have no streaming data). Pushed to far right with
   Spacer weight. Reset to 0 on new chat and conversation switch.
2. Mode pills hidden when keyboard is open (WindowInsets.isImeVisible)
   — prevents the bottom dock from pushing too far up and overlapping
   chat content. Input field stays compact at the bottom.
3. ModelPill no longer constrained with weight(fill=false) which
   was causing layout issues.

app/src/main/java/dev/kaizen/app/chat/ChatScreen.kt

@@ -13,7 +13,10 @@ import androidx.compose.foundation.layout.Spacer
 import androidx.compose.foundation.layout.fillMaxSize
 import androidx.compose.foundation.layout.fillMaxWidth
 import androidx.compose.foundation.layout.height
+import androidx.compose.foundation.layout.ExperimentalLayoutApi
+import androidx.compose.foundation.layout.WindowInsets
 import androidx.compose.foundation.layout.imePadding
+import androidx.compose.foundation.layout.isImeVisible
 import androidx.compose.foundation.layout.navigationBarsPadding
 import androidx.compose.foundation.layout.padding
 import androidx.compose.foundation.layout.size
@@ -219,6 +222,7 @@ fun ChatScreen(
         messages.clear()
         conversationId = null
         viewingLockedChat = false
+        usedTokens = 0
     }
 
     fun openConversation(summary: ConversationSummary) {
@@ -269,6 +273,7 @@ fun ChatScreen(
             return
         }
         viewingLockedChat = false
+        usedTokens = 0
         scope.launch {
             messages.clear()
             val cachedList = chat.messageRepo.getCached(summary.id)
@@ -655,11 +660,10 @@ fun ChatScreen(
                                 ModelPill(
                                     label = modelDisplayName(session.config?.model ?: "", models),
                                     onClick = { haptics.tick(); showModelSheet = true },
-                                    modifier = Modifier.weight(1f, fill = false),
                                 )
-                                if (usedTokens > 0 || messages.isNotEmpty()) {
+                                Spacer(Modifier.weight(1f))
+                                if (usedTokens > 0) {
                                     val ctxLen = models.find { it.id == session.config?.model }?.context_length ?: 1_000_000
-                                    Spacer(Modifier.width(8.dp))
                                     TokenBadge(used = usedTokens, contextLength = ctxLen)
                                 }
                             }
@@ -704,6 +708,9 @@ fun ChatScreen(
                                 .imePadding()
                                 .then(if (isTablet) Modifier.widthIn(max = 720.dp) else Modifier)
                         ) {
+                            @OptIn(ExperimentalLayoutApi::class)
+                            val imeVisible = WindowInsets.isImeVisible
+                            if (!imeVisible) {
                                 Spacer(Modifier.height(24.dp))
                                 ModePillsRow(
                                     activeMode = activeMode,
@@ -716,6 +723,7 @@ fun ChatScreen(
                                     samplingPrefs = samplingPrefs,
                                     onSamplingChange = { samplingPrefs = it },
                                 )
+                            }
                             Spacer(Modifier.height(10.dp))
                             ChatInput(
                                 value = input,

app/src/main/java/dev/kaizen/app/chat/StreamBlocks.kt

@@ -10,8 +10,6 @@ import androidx.compose.foundation.isSystemInDarkTheme
 import androidx.compose.foundation.layout.Arrangement
 import androidx.compose.foundation.layout.Box
 import androidx.compose.foundation.layout.Column
-import androidx.compose.foundation.layout.ExperimentalLayoutApi
-import androidx.compose.foundation.layout.FlowRow
 import androidx.compose.foundation.layout.Row
 import androidx.compose.foundation.layout.Spacer
 import androidx.compose.foundation.layout.fillMaxWidth
@@ -50,6 +48,7 @@ import dev.kaizen.app.net.ToolEvent
 import dev.kaizen.app.R
 import dev.kaizen.app.ui.shape.KaizenShapes
 import dev.kaizen.app.ui.theme.LocalKaizenAccent
+import androidx.compose.ui.platform.LocalContext
 import androidx.compose.ui.res.stringResource
 
 @Composable
@@ -170,36 +169,55 @@ fun ToolEventsBlock(tools: List<ToolEvent>, streaming: Boolean = false) {
     }
 }
 
-@OptIn(ExperimentalLayoutApi::class)
 @Composable
 fun SourcesBlock(sources: List<SearchSource>, query: String = "") {
     val cs = MaterialTheme.colorScheme
     val isDark = isSystemInDarkTheme()
+    val accent = LocalKaizenAccent.current
+    var expanded by remember { mutableStateOf(false) }
+    val context = LocalContext.current
 
     Column {
-        if (query.isNotBlank()) {
-            Row(verticalAlignment = Alignment.CenterVertically, modifier = Modifier.padding(bottom = 6.dp)) {
-                Icon(Icons.Rounded.Language, null, tint = cs.onSurfaceVariant, modifier = Modifier.size(14.dp))
-                Spacer(Modifier.width(6.dp))
-                Text(query, color = cs.onSurfaceVariant, fontSize = 12.sp, fontWeight = FontWeight.Medium)
-            }
-        }
-        FlowRow(
-            horizontalArrangement = Arrangement.spacedBy(6.dp),
-            verticalArrangement = Arrangement.spacedBy(6.dp),
+        Row(
+            Modifier
+                .clickable { expanded = !expanded }
+                .padding(vertical = 4.dp),
+            verticalAlignment = Alignment.CenterVertically,
         ) {
+            Icon(Icons.Rounded.Language, null, tint = cs.onSurfaceVariant.copy(alpha = 0.6f), modifier = Modifier.size(13.dp))
+            Spacer(Modifier.width(6.dp))
+            Text(
+                stringResource(R.string.stream_sources) + " (${sources.size})",
+                color = cs.onSurfaceVariant.copy(alpha = 0.6f),
+                fontSize = 12.sp,
+                fontWeight = FontWeight.Medium,
+            )
+            Spacer(Modifier.width(4.dp))
+            Icon(
+                if (expanded) Icons.Rounded.ExpandLess else Icons.Rounded.ExpandMore,
+                null, tint = cs.onSurfaceVariant.copy(alpha = 0.4f), modifier = Modifier.size(12.dp),
+            )
+        }
+        AnimatedVisibility(expanded, enter = expandVertically(), exit = shrinkVertically()) {
+            Column(verticalArrangement = Arrangement.spacedBy(6.dp), modifier = Modifier.padding(top = 4.dp)) {
                 sources.forEach { source ->
-                SourceChip(source, isDark)
+                    SourceCard(source, isDark, accent) {
+                        try {
+                            val intent = android.content.Intent(android.content.Intent.ACTION_VIEW, android.net.Uri.parse(source.url))
+                            context.startActivity(intent)
+                        } catch (_: Exception) { }
+                    }
+                }
             }
         }
     }
 }
 
 @Composable
-private fun SourceChip(source: SearchSource, isDark: Boolean) {
+private fun SourceCard(source: SearchSource, isDark: Boolean, accent: dev.kaizen.app.ui.theme.themes.KaizenAccentScheme, onClick: () -> Unit) {
     val cs = MaterialTheme.colorScheme
-    val bg = if (isDark) Color.White.copy(alpha = 0.06f) else Color.Black.copy(alpha = 0.04f)
-    val borderColor = if (isDark) Color.White.copy(alpha = 0.08f) else Color.Black.copy(alpha = 0.06f)
+    val bg = if (isDark) Color.White.copy(alpha = 0.05f) else Color.Black.copy(alpha = 0.03f)
+    val borderColor = if (isDark) Color.White.copy(alpha = 0.06f) else Color.Black.copy(alpha = 0.05f)
 
     val domain = remember(source.url) {
         try {
@@ -209,27 +227,38 @@ private fun SourceChip(source: SearchSource, isDark: Boolean) {
 
     Column(
         Modifier
-            .width(140.dp)
-            .clip(KaizenShapes.xs)
+            .fillMaxWidth()
+            .clip(KaizenShapes.sm)
             .background(bg)
-            .border(1.dp, borderColor, KaizenShapes.xs)
-            .padding(horizontal = 10.dp, vertical = 8.dp),
+            .border(1.dp, borderColor, KaizenShapes.sm)
+            .clickable(onClick = onClick)
+            .padding(horizontal = 12.dp, vertical = 10.dp),
     ) {
         Text(
             source.title.ifEmpty { domain },
             color = cs.onBackground,
-            fontSize = 12.sp,
+            fontSize = 13.sp,
             fontWeight = FontWeight.Medium,
             maxLines = 2,
             overflow = TextOverflow.Ellipsis,
         )
         Spacer(Modifier.height(2.dp))
         Text(
-            domain,
-            color = cs.onSurfaceVariant.copy(alpha = 0.6f),
-            fontSize = 10.sp,
+            source.url,
+            color = accent.primary.copy(alpha = 0.7f),
+            fontSize = 11.sp,
             maxLines = 1,
             overflow = TextOverflow.Ellipsis,
         )
+        if (source.snippet.isNotBlank()) {
+            Spacer(Modifier.height(3.dp))
+            Text(
+                source.snippet,
+                color = cs.onSurfaceVariant.copy(alpha = 0.5f),
+                fontSize = 11.sp,
+                maxLines = 2,
+                overflow = TextOverflow.Ellipsis,
+            )
+        }
     }
 }