import ai.openclaw.app.chat.ChatMessage

import ai.openclaw.app.chat.ChatMessageContent

import ai.openclaw.app.chat.ChatPendingToolCall

import ai.openclaw.app.chat.OutgoingAttachment

import ai.openclaw.app.ui.design.ClawListItem

import ai.openclaw.app.ui.design.ClawPanel

import ai.openclaw.app.ui.design.ClawStatus

import ai.openclaw.app.ui.design.ClawStatusPill

import ai.openclaw.app.ui.design.ClawTheme

import androidx.activity.compose.rememberLauncherForActivityResult

import androidx.activity.result.contract.ActivityResultContracts

import androidx.compose.foundation.BorderStroke

import androidx.compose.foundation.background

import androidx.compose.foundation.horizontalScroll

import androidx.compose.foundation.layout.Arrangement

import androidx.compose.foundation.layout.Box

import androidx.compose.foundation.layout.Column

import androidx.compose.foundation.lazy.LazyColumn

import androidx.compose.foundation.lazy.items

import androidx.compose.foundation.lazy.rememberLazyListState

import androidx.compose.foundation.rememberScrollState

import androidx.compose.foundation.shape.CircleShape

import androidx.compose.foundation.shape.RoundedCornerShape

import androidx.compose.foundation.text.BasicTextField

import androidx.compose.material.icons.Icons

import androidx.compose.material.icons.automirrored.filled.ArrowBack

import androidx.compose.material.icons.automirrored.filled.Send

import androidx.compose.material.icons.filled.AttachFile

import androidx.compose.material.icons.filled.Close

import androidx.compose.material.icons.filled.Mic

import androidx.compose.material.icons.filled.Refresh

import androidx.compose.material3.HorizontalDivider

import androidx.compose.runtime.LaunchedEffect

import androidx.compose.runtime.collectAsState

import androidx.compose.runtime.getValue

import androidx.compose.runtime.mutableStateListOf

import androidx.compose.runtime.mutableStateOf

import androidx.compose.runtime.remember

import androidx.compose.runtime.rememberCoroutineScope

import androidx.compose.ui.Modifier

import androidx.compose.ui.graphics.Color

import androidx.compose.ui.graphics.SolidColor

import androidx.compose.ui.platform.LocalContext

import androidx.compose.ui.text.font.FontWeight

import androidx.compose.ui.text.style.TextAlign

import androidx.compose.ui.text.style.TextOverflow

import androidx.compose.ui.unit.dp

import androidx.compose.ui.unit.sp

import kotlinx.coroutines.Dispatchers

import kotlinx.coroutines.launch

import kotlinx.coroutines.withContext

import java.text.DateFormat

import java.util.Date

import java.util.Locale

val sessions by viewModel.chatSessions.collectAsState()

val chatDraft by viewModel.chatDraft.collectAsState()

val pendingAssistantAutoSend by viewModel.pendingAssistantAutoSend.collectAsState()

val context = LocalContext.current

val resolver = context.contentResolver

val scope = rememberCoroutineScope()

val attachments = remember { mutableStateListOf<PendingImageAttachment>() }

val pickImages =

rememberLauncherForActivityResult(ActivityResultContracts.GetMultipleContents()) { uris ->

if (uris.isNullOrEmpty()) return@rememberLauncherForActivityResult

scope.launch(Dispatchers.IO) {

val next =

uris.take(8).mapNotNull { uri ->

try {

loadSizedImageAttachment(resolver, uri)

} catch (_: Throwable) {

null

}

}

withContext(Dispatchers.Main) {

attachments.addAll(next)

}

}

}

LaunchedEffect(Unit) {

viewModel.loadChat(mainSessionKey)

V2ChatComposer(

value = input,

onValueChange = { input = it },

attachments = attachments,

thinkingLevel = thinkingLevel,

healthOk = healthOk,

pendingRunCount = pendingRunCount,

onThinkingLevelChange = viewModel::setChatThinkingLevel,

onPickImages = { pickImages.launch("image/*") },

onRemoveAttachment = { id -> attachments.removeAll { it.id == id } },

onVoice = onVoice,

onAbort = viewModel::abortChat,

onSend = {

val message = input.trim()

if (message.isEmpty()) return@V2ChatComposer

if (message.isEmpty() && attachments.isEmpty()) return@V2ChatComposer

val outgoing =

attachments.map { attachment ->

OutgoingAttachment(

type = "image",

mimeType = attachment.mimeType,

fileName = attachment.fileName,

base64 = attachment.base64,

)

}

input = ""

attachments.clear()

scope.launch {

viewModel.sendChat(message = message, thinking = thinkingLevel, attachments = emptyList())

viewModel.sendChat(message = message, thinking = thinkingLevel, attachments = outgoing)

}

},

)

private fun V2ChatComposer(

value: String,

onValueChange: (String) -> Unit,

attachments: List<PendingImageAttachment>,

thinkingLevel: String,

healthOk: Boolean,

pendingRunCount: Int,

onThinkingLevelChange: (String) -> Unit,

onPickImages: () -> Unit,

onRemoveAttachment: (String) -> Unit,

onVoice: () -> Unit,

onAbort: () -> Unit,

onSend: () -> Unit,

) {

Column(modifier = Modifier.fillMaxWidth().imePadding(), verticalArrangement = Arrangement.spacedBy(4.dp)) {

if (attachments.isNotEmpty()) {

V2AttachmentStrip(attachments = attachments, onRemoveAttachment = onRemoveAttachment)

}

V2ChatContextMeter(thinkingLevel = thinkingLevel, onClick = { onThinkingLevelChange(nextThinkingValue(thinkingLevel)) })

Row(modifier = Modifier.fillMaxWidth(), verticalAlignment = Alignment.CenterVertically, horizontalArrangement = Arrangement.spacedBy(6.dp)) {

V2ChatInputPill(value = value, onValueChange = onValueChange, onVoice = onVoice, modifier = Modifier.weight(1f))

V2ChatInputPill(value = value, onValueChange = onValueChange, onPickImages = onPickImages, onVoice = onVoice, modifier = Modifier.weight(1f))

V2SendButton(

enabled = healthOk && pendingRunCount == 0 && value.trim().isNotEmpty(),

enabled = healthOk && pendingRunCount == 0 && (value.trim().isNotEmpty() || attachments.isNotEmpty()),

onClick = onSend,

)

}

private fun V2ChatInputPill(

value: String,

onValueChange: (String) -> Unit,

onPickImages: () -> Unit,

onVoice: () -> Unit,

modifier: Modifier = Modifier,

) {

verticalAlignment = Alignment.CenterVertically,

horizontalArrangement = Arrangement.spacedBy(7.dp),

) {

Icon(imageVector = Icons.Default.AttachFile, contentDescription = "Attach file", modifier = Modifier.size(13.dp), tint = ClawTheme.colors.text)

Surface(onClick = onPickImages, modifier = Modifier.size(ClawTheme.spacing.touchTarget), shape = CircleShape, color = ClawTheme.colors.surfaceRaised, contentColor = ClawTheme.colors.text) {

Box(contentAlignment = Alignment.Center) {

Icon(imageVector = Icons.Default.AttachFile, contentDescription = "Attach image", modifier = Modifier.size(16.dp))

}

}

Box(modifier = Modifier.weight(1f)) {

BasicTextField(

value = value,

}

}

@Composable

private fun V2AttachmentStrip(

attachments: List<PendingImageAttachment>,

onRemoveAttachment: (String) -> Unit,

) {

Row(modifier = Modifier.fillMaxWidth().horizontalScroll(rememberScrollState()), horizontalArrangement = Arrangement.spacedBy(6.dp)) {

attachments.forEach { attachment ->

V2AttachmentChip(fileName = attachment.fileName, onRemove = { onRemoveAttachment(attachment.id) })

}

}

}

@Composable

private fun V2AttachmentChip(

fileName: String,

onRemove: () -> Unit,

) {

Surface(

shape = RoundedCornerShape(ClawTheme.radii.pill),

color = ClawTheme.colors.surfaceRaised,

contentColor = ClawTheme.colors.text,

border = BorderStroke(1.dp, ClawTheme.colors.border),

) {

Row(

modifier = Modifier.padding(start = 9.dp, top = 5.dp, end = 5.dp, bottom = 5.dp),

verticalAlignment = Alignment.CenterVertically,

horizontalArrangement = Arrangement.spacedBy(6.dp),

) {

Text(text = fileName, style = ClawTheme.type.caption, color = ClawTheme.colors.textMuted, maxLines = 1, overflow = TextOverflow.Ellipsis)

Surface(onClick = onRemove, modifier = Modifier.size(22.dp), shape = CircleShape, color = ClawTheme.colors.canvas, contentColor = ClawTheme.colors.text) {

Box(contentAlignment = Alignment.Center) {

Icon(imageVector = Icons.Default.Close, contentDescription = "Remove attachment", modifier = Modifier.size(13.dp))

}

}

}

}

}

private fun currentSessionTitle(

sessionKey: String,

sessions: List<ai.openclaw.app.chat.ChatSessionEntry>,